搭建基础

Author: CppCXY

CodeFormatServer/src/LanguageService.cpp

@@ -10,6 +10,7 @@
 #include "CodeFormatServer/Service/CodeFormatService.h"
 #include "CodeFormatServer/Service/ModuleService.h"
 #include "Util/Url.h"
+#include "Util/format.h"
 
 using namespace std::placeholders;
 
@@ -359,6 +360,47 @@ std::shared_ptr<vscode::Serializable> LanguageService::OnExecuteCommand(
 	}
 	else if (param->command == "emmylua.import.me")
 	{
+		if (param->arguments.size() < 3)
+		{
+			return result;
+		}
+
+		std::string uri = param->arguments[0];
+		vscode::Range range;
+
+		range.Deserialize(param->arguments[1]);
+
+		std::string moduleName = param->arguments[2];
+		auto pos = moduleName.find_last_of('.');
+		std::string moduleDefineName;
+
+		if (pos == std::string_view::npos)
+		{
+			moduleDefineName = moduleName;
+		}
+		else
+		{
+			moduleDefineName = moduleName.substr(pos + 1);
+		}
+
+		std::string requireString = format("local {} = require(\"{}\")\n", moduleDefineName, moduleName);
+		auto parser = LanguageClient::GetInstance().GetFileParser(uri);
+
+		auto applyParams = std::make_shared<vscode::ApplyWorkspaceEditParams>();
+		auto it = applyParams->edit.changes.emplace(uri, std::vector<vscode::TextEdit>());
+		auto& change = it.first->second;
+
+		auto& edit = change.emplace_back();
+		LuaFormatRange formattedRange(static_cast<int>(range.start.line), static_cast<int>(range.end.line));
+
+		edit.newText = requireString;
+
+		edit.range = vscode::Range(
+			vscode::Position(formattedRange.StartLine, formattedRange.StartCharacter),
+			vscode::Position(formattedRange.EndLine + 1, formattedRange.EndCharacter)
+		);
+
+		LanguageClient::GetInstance().SendRequest("workspace/applyEdit", applyParams);
 	}
 	return result;
 }

CodeFormatServer/src/Service/ModuleService.cpp

@@ -228,10 +228,77 @@ std::vector<ModuleIndex::Module> ModuleService::GetImportModules(std::string_vie
 	auto& rangeModules = fIt->second;
 	auto equalRange = rangeModules.equal_range(range);
 	std::vector<ModuleIndex::Module> result;
-	for(auto it = equalRange.first; it != equalRange.second; ++it)
+	for (auto it = equalRange.first; it != equalRange.second; ++it)
 	{
 		result.emplace_back(it->second);
 	}
 
 	return result;
 }
+
+vscode::Range ModuleService::FindRequireRange(std::shared_ptr<LuaParser> parser)
+{
+	vscode::Range range;
+	auto chunkAst = parser->GetAst();
+	if (chunkAst->GetChildren().empty())
+	{
+		return range;
+	}
+
+	auto blockNode = chunkAst->GetChildren().front();
+
+	std::shared_ptr<LuaAstNode> lastNode = nullptr;
+
+	auto& children = blockNode->GetChildren();
+	for (auto statement : children)
+	{
+		auto type = statement->GetType();
+
+		switch (type)
+		{
+		case LuaAstNodeType::Comment:
+			{
+				break;
+			}
+		case LuaAstNodeType::LocalStatement:
+			{
+				std::shared_ptr<LuaAstNode> expressionList = statement->FindFirstOf(LuaAstNodeType::ExpressionList);
+				if (expressionList)
+				{
+					auto expression = expressionList->FindFirstOf(LuaAstNodeType::Expression);
+					if (expression)
+					{
+						auto callExpression = expression->FindFirstOf(LuaAstNodeType::CallExpression);
+						if (callExpression)
+						{
+							// ½«Ô¼¶¨×ö³É¹æ·¶
+							auto methodNameNode = callExpression->FindFirstOf(LuaAstNodeType::PrimaryExpression);
+							if (methodNameNode)
+							{
+								if (methodNameNode->GetText() == "require")
+								{
+									break;
+								}
+							}
+						}
+					}
+				}
+				goto endLoop;
+			}
+		default:
+			{
+				break;
+			}
+		}
+		lastNode = statement;
+	}
+endLoop:
+	if (lastNode)
+	{
+		vscode::Position insertPosition(parser->GetLine(lastNode->GetTextRange().EndOffset) + 1, 0);
+		range.start = insertPosition;
+		range.end = insertPosition;
+	}
+
+	return range;
+}