如何在当前文档中在 Visual Studio 中生成代码？

Question

如果这是很容易搜索的内容，我深表歉意，但我不确定正确的搜索词是什么。

我现在正在使用 Visual Studio，主要用于 Unity 开发。在创建新脚本时，我经常声明一些私有成员变量，如下所示：

private Rigidbody _rigidbody;
private SomeOtherComponent _myComponent;

为了初始化这些变量，我需要在一个名为“Start”的函数中添加一个 GetComponent 调用，如下所示：

void Start(){
    _rigidbody = GetComponent<Rigidbody>();
    _myComponent = GetComponent<SomeOtherComponent>();
}

现在，我找到了一个 Visual Studio sn-p 来快速键入“GetComponent”部分，这很好，但我正在寻找一种在 Start 函数中自动生成代码行的方法。在理想的世界里，我可以输入

private Rigidbody _rigidbody;

然后我可以按一些组合键，它会自动添加

_rigidbody = GetComponent<Rigidbody>();

到 Start 函数。

片段只能让我走到一半。我需要做什么来为 Visual Studio 创建这种类型的扩展？这是一件很难完成的事情吗？

Answer 1

你可以试试我的Visual Commander 扩展，它专为这种轻量级可扩展性而设计。它允许将文档编辑为文本或使用 Visual Studio 代码模型和 Roslyn。

例如，当插入符号位于变量声明中的 _rigidbody 或 _myComponent 上时，调用以下命令：

public class C : VisualCommanderExt.ICommand
{
    public void Run(EnvDTE80.DTE2 DTE, Microsoft.VisualStudio.Shell.Package package)
    {
        this.DTE = DTE;

        EnvDTE.CodeVariable v = FindCurrentVariable();
        if (v != null)
        {
            string initialization = v.Name + " = GetComponent<" + v.Type.CodeType.Name + ">();";
            AddLine(FindFunction("Start"), initialization);
        }
    }

    EnvDTE.CodeFunction FindFunction(string name)
    {
        EnvDTE.TextSelection ts = DTE.ActiveWindow.Selection as EnvDTE.TextSelection;
        if (ts == null)
            return null;
        EnvDTE.CodeClass codeClass = ts.ActivePoint.CodeElement[EnvDTE.vsCMElement.vsCMElementClass]
            as EnvDTE.CodeClass;
        if (codeClass == null)
            return null;
        foreach (EnvDTE.CodeElement elem in codeClass.Members)
        {
            if (elem.Kind == EnvDTE.vsCMElement.vsCMElementFunction && elem.Name == name)
                return elem as EnvDTE.CodeFunction;
        }
        return null;
    }

    EnvDTE.CodeVariable FindCurrentVariable()
    {
        EnvDTE.TextSelection ts = DTE.ActiveWindow.Selection as EnvDTE.TextSelection;
        if (ts == null)
            return null;
        return ts.ActivePoint.CodeElement[EnvDTE.vsCMElement.vsCMElementVariable]
            as EnvDTE.CodeVariable;
    }

    void AddLine(EnvDTE.CodeFunction f, string text)
    {
        EnvDTE.TextPoint tp = f.GetStartPoint(EnvDTE.vsCMPart.vsCMPartBody);
        EnvDTE.EditPoint p = tp.CreateEditPoint();
        p.Insert(text + System.Environment.NewLine);
        p.SmartFormat(tp);
    }

    EnvDTE80.DTE2 DTE;
}