You're breathtaking!

Source/Editor/Scripting/CodeEditors/VisualStudio/VisualStudioEditor.cpp

@@ -0,0 +1,207 @@
+// Copyright (c) 2012-2020 Wojciech Figat. All rights reserved.
+
+#if USE_VISUAL_STUDIO_DTE
+
+#include "VisualStudioEditor.h"
+#include "Engine/Platform/FileSystem.h"
+#include "Engine/Core/Log.h"
+#include "Editor/Editor.h"
+#include "Editor/Scripting/ScriptsBuilder.h"
+#include "Engine/Engine/Globals.h"
+#include "Engine/Platform/Win32/IncludeWindowsHeaders.h"
+#include "VisualStudioConnection.h"
+
+VisualStudioEditor::VisualStudioEditor(VisualStudioVersion version, const String& execPath, const String& CLSID)
+    : _version(version)
+    , _execPath(execPath)
+    , _CLSID(CLSID)
+{
+    switch (version)
+    {
+    case VisualStudioVersion::VS2008:
+        _type = CodeEditorTypes::VS2008;
+        break;
+    case VisualStudioVersion::VS2010:
+        _type = CodeEditorTypes::VS2010;
+        break;
+    case VisualStudioVersion::VS2012:
+        _type = CodeEditorTypes::VS2012;
+        break;
+    case VisualStudioVersion::VS2013:
+        _type = CodeEditorTypes::VS2013;
+        break;
+    case VisualStudioVersion::VS2015:
+        _type = CodeEditorTypes::VS2015;
+        break;
+    case VisualStudioVersion::VS2017:
+        _type = CodeEditorTypes::VS2017;
+        break;
+    case VisualStudioVersion::VS2019:
+        _type = CodeEditorTypes::VS2019;
+        break;
+    default: CRASH;
+        break;
+    }
+    _solutionPath = Globals::ProjectFolder / Editor::Project->Name + TEXT(".sln");
+}
+
+void VisualStudioEditor::FindEditors(Array<CodeEditor*>* output)
+{
+    String installDir;
+    String clsID;
+    String regVsRootNode;
+
+    // Select registry node for
+    if (Platform::Is64BitPlatform())
+        regVsRootNode = TEXT("SOFTWARE\\Wow6432Node\\Microsoft\\VisualStudio\\");
+    else
+        regVsRootNode = TEXT("SOFTWARE\\Microsoft\\VisualStudio\\");
+
+    VisualStudio::InstanceInfo infos[8];
+    const int32 count = VisualStudio::GetVisualStudioVersions(infos, ARRAY_COUNT(infos));
+    for (int32 i = 0; i < count; i++)
+    {
+        auto& info = infos[i];
+        VisualStudioVersion version;
+        switch (info.VersionMajor)
+        {
+        case 16:
+            version = VisualStudioVersion::VS2019;
+            break;
+        case 15:
+            version = VisualStudioVersion::VS2017;
+            break;
+        default:
+            break;
+        }
+
+        String executablePath(info.ExecutablePath);
+        if (!FileSystem::FileExists(executablePath))
+            continue;
+
+        // Create editor
+        clsID = info.CLSID;
+        auto editor = New<VisualStudioEditor>(version, executablePath, clsID);
+        output->Add(editor);
+    }
+
+    struct VersionData
+    {
+        VisualStudioVersion Version;
+        const Char* RegistryKey;
+    };
+
+    const VersionData versions[] =
+    {
+        // Order matters
+        { VisualStudioVersion::VS2015, TEXT("14.0"), },
+        { VisualStudioVersion::VS2013, TEXT("12.0"), },
+        { VisualStudioVersion::VS2012, TEXT("11.0"), },
+        { VisualStudioVersion::VS2012, TEXT("11.0"), },
+        { VisualStudioVersion::VS2008, TEXT("9.0") },
+    };
+
+    for (int32 i = 0; i < ARRAY_COUNT(versions); i++)
+    {
+        auto registryKey = regVsRootNode + versions[i].RegistryKey;
+
+        // Read install directory
+        if (Platform::ReadRegValue(HKEY_LOCAL_MACHINE, registryKey, TEXT("InstallDir"), &installDir) || installDir.IsEmpty())
+            continue;
+
+        // Ensure that file exists
+        String execPath = installDir + TEXT("devenv.exe");
+        if (!FileSystem::FileExists(execPath))
+            continue;
+
+        // Read version info id
+        clsID.Clear();
+        Platform::ReadRegValue(HKEY_LOCAL_MACHINE, registryKey, TEXT("ThisVersionDTECLSID"), &clsID);
+
+        // Create editor
+        auto editor = New<VisualStudioEditor>(versions[i].Version, execPath, clsID);
+        output->Add(editor);
+    }
+}
+
+CodeEditorTypes VisualStudioEditor::GetType() const
+{
+    return _type;
+}
+
+String VisualStudioEditor::GetName() const
+{
+    return String(ToString(_version));
+}
+
+void VisualStudioEditor::OpenFile(const String& path, int32 line)
+{
+    // Generate project files if solution is missing
+    if (!FileSystem::FileExists(_solutionPath))
+    {
+        ScriptsBuilder::GenerateProject();
+    }
+
+    // Open file
+    const VisualStudio::Connection connection(*_CLSID, *_solutionPath);
+    const auto result = connection.OpenFile(*path, line);
+    if (result.Failed())
+    {
+        LOG(Warning, "Cannot open file \'{0}\':{1}. {2}.", path, line, String(result.Message.c_str()));
+    }
+}
+
+void VisualStudioEditor::OpenSolution()
+{
+    // Generate project files if solution is missing
+    if (!FileSystem::FileExists(_solutionPath))
+    {
+        ScriptsBuilder::GenerateProject();
+    }
+
+    // Open solution
+    const VisualStudio::Connection connection(*_CLSID, *_solutionPath);
+    const auto result = connection.OpenSolution();
+    if (result.Failed())
+    {
+        LOG(Warning, "Cannot open solution. {0}", String(result.Message.c_str()));
+    }
+}
+
+void VisualStudioEditor::OnFileAdded(const String& path)
+{
+    // TODO: finish dynamic files adding to the project - for now just regenerate it
+    ScriptsBuilder::GenerateProject();
+    return;
+    if (!FileSystem::FileExists(_solutionPath))
+    {
+        return;
+    }
+
+    // Edit solution
+    const VisualStudio::Connection connection(*_CLSID, *_solutionPath);
+    if (connection.IsActive())
+    {
+        String tmp = path;
+        tmp.Replace('/', '\\');
+        String tmp2 = tmp.Substring(Globals::ProjectSourceFolder.Length() + 1);
+        const auto result = connection.AddFile(*tmp, *tmp2);
+        if (result.Failed())
+        {
+            LOG(Warning, "Cannot add file to project. {0}", String(result.Message.c_str()));
+        }
+    }
+}
+
+bool VisualStudioEditor::UseAsyncForOpen() const
+{
+    // Need to generate project files if missing first
+    if (!FileSystem::FileExists(_solutionPath))
+        return true;
+
+    // Open in async only when no solution opened
+    const VisualStudio::Connection connection(*_CLSID, *_solutionPath);
+    return !connection.IsActive();
+}
+
+#endif