简单支持typeformatting

Author: CppCXY

CodeFormatServer/src/LanguageService.cpp

@@ -26,6 +26,8 @@ bool LanguageService::Initialize()
 	_handles["textDocument/didClose"] = DynamicBind(OnClose, vscode::DidCloseTextDocumentParams);
 	_handles["updateEditorConfig"] = DynamicBind(OnEditorConfigUpdate, vscode::EditorConfigUpdateParams);
 	_handles["textDocument/rangeFormatting"] = DynamicBind(OnRangeFormatting, vscode::DocumentRangeFormattingParams);
+	_handles["textDocument/onTypeFormatting"] = DynamicBind(OnTypeFormatting, vscode::TextDocumentPositionParams);
+
 	return true;
 }
 
@@ -46,6 +48,15 @@ std::shared_ptr<vscode::InitializeResult> LanguageService::OnInitialize(std::sha
 
 	result->capabilities.documentFormattingProvider = true;
 	result->capabilities.documentRangeFormattingProvider = true;
+
+	vscode::DocumentOnTypeFormattingOptions typeOptions;
+
+	typeOptions.firstTriggerCharacter = ";";
+
+	typeOptions.moreTriggerCharacter = {"\"", "\'", ",", ":", "."};
+
+	result->capabilities.documentOnTypeFormattingProvider = typeOptions;
+
 	result->capabilities.textDocumentSync.change = vscode::TextDocumentSyncKind::Full;
 	result->capabilities.textDocumentSync.openClose = true;
 
@@ -184,3 +195,42 @@ std::shared_ptr<vscode::Serializable> LanguageService::OnRangeFormatting(
 	);
 	return result;
 }
+
+std::shared_ptr<vscode::Serializable> LanguageService::OnTypeFormatting(
+	std::shared_ptr<vscode::TextDocumentPositionParams> param)
+{
+	auto text = LanguageClient::GetInstance().GetFile(param->textDocument.uri);
+
+	auto result = std::make_shared<vscode::DocumentFormattingResult>();
+
+	if (text.empty())
+	{
+		result->hasError = true;
+		return result;
+	}
+
+	auto options = LanguageClient::GetInstance().GetOptions(param->textDocument.uri);
+
+	std::shared_ptr<LuaParser> parser = LuaParser::LoadFromBuffer(std::move(text));
+	parser->BuildAstWithComment();
+
+	// ºöÂÔ¼üÈë´íÎó
+	// if (parser->HasError())
+	// {
+	// 	result->hasError = true;
+	// 	return result;
+	// }
+
+	LuaFormatter formatter(parser, *options);
+	formatter.BuildFormattedElement();
+
+	auto& edit = result->edits.emplace_back();
+	LuaFormatRange formattedRange(param->position.line, param->position.line);
+
+	edit.newText = formatter.GetRangeFormattedText(formattedRange);
+	edit.range = vscode::Range(
+		vscode::Position(formattedRange.StartLine, 0),
+		vscode::Position(formattedRange.EndLine + 1, 0)
+	);
+	return result;
+}

CodeFormatServer/src/VSCode.cpp

@@ -43,6 +43,10 @@ std::shared_ptr<vscode::Serializable> vscode::MakeFromRequest(std::string_view m
 	{
 		return MakeRequestObject<DocumentRangeFormattingParams>(json);
 	}
+	else if (method == "textDocument/onTypeFormatting")
+	{
+		return MakeRequestObject<TextDocumentPositionParams>(json);
+	}
 
 	return nullptr;
 }
@@ -194,6 +198,8 @@ nlohmann::json vscode::ServerCapabilities::Serialize()
 	object["textDocumentSync"] = textDocumentSync.Serialize();
 	object["documentFormattingProvider"] = documentFormattingProvider;
 	object["documentRangeFormattingProvider"] = documentRangeFormattingProvider;
+	object["documentOnTypeFormattingProvider"] = documentOnTypeFormattingProvider.Serialize();
+
 	return object;
 }
 
@@ -337,3 +343,25 @@ void vscode::DocumentRangeFormattingParams::Deserialize(nlohmann::json json)
 	textDocument.Deserialize(json["textDocument"]);
 	range.Deserialize(json["range"]);
 }
+
+nlohmann::json vscode::DocumentOnTypeFormattingOptions::Serialize()
+{
+	auto object = nlohmann::json::object();
+
+	object["firstTriggerCharacter"] = firstTriggerCharacter;
+	auto array = nlohmann::json::array();
+
+	for (auto& c : moreTriggerCharacter)
+	{
+		array.push_back(c);
+	}
+
+	object["moreTriggerCharacter"] = array;
+	return object;
+}
+
+void vscode::TextDocumentPositionParams::Deserialize(nlohmann::json json)
+{
+	textDocument.Deserialize(json["textDocument"]);
+	position.Deserialize(json["position"]);
+}

include/CodeFormatServer/LanguageService.h

@@ -34,6 +34,8 @@ class LanguageService
 
 	std::shared_ptr<vscode::Serializable> OnRangeFormatting(std::shared_ptr<vscode::DocumentRangeFormattingParams> param);
 
+	std::shared_ptr<vscode::Serializable> OnTypeFormatting(std::shared_ptr<vscode::TextDocumentPositionParams> param);
+
 	std::map<std::string, MessageHandle, std::less<>> _handles;
 };
 

include/CodeFormatServer/VSCode.h

@@ -8,6 +8,7 @@
 
 namespace vscode
 {
+
 class Serializable
 {
 public:
@@ -127,12 +128,22 @@ class TextDocumentSyncOptions : public Serializable
 	nlohmann::json Serialize() override;
 };
 
+class DocumentOnTypeFormattingOptions : public Serializable
+{
+public:
+	std::string firstTriggerCharacter;
+	std::vector<std::string> moreTriggerCharacter;
+
+	nlohmann::json Serialize() override;
+};
+
 class ServerCapabilities : public Serializable
 {
 public:
 	TextDocumentSyncOptions textDocumentSync;
 	bool documentFormattingProvider = false;
 	bool documentRangeFormattingProvider = false;
+	DocumentOnTypeFormattingOptions documentOnTypeFormattingProvider;
 
 	nlohmann::json Serialize() override;
 };
@@ -268,4 +279,13 @@ class DocumentRangeFormattingParams: public Serializable
 	void Deserialize(nlohmann::json json) override;
 };
 
+class TextDocumentPositionParams: public Serializable
+{
+public:
+	TextDocument textDocument;
+	Position position;
+
+	void Deserialize(nlohmann::json json) override;
+};
+
 }