initial commit

Author: tris203

.gitignore

@@ -0,0 +1,2 @@
+main.js
+test.js

.luacheckrc

@@ -0,0 +1 @@
+globals = { "vim", "describe", "it", "before_each", "after_each", "assert", "async" }

.luarc.json

@@ -0,0 +1,15 @@
+{
+  "runtime": {
+    "version": "LuaJIT",
+    "pathStrict": true
+  },
+  "type": {
+    "checkTableShape": true
+  },
+    "diagnostics.globals": [
+        "describe",
+        "it",
+        "before_each",
+        "after_each"
+    ]
+}

.stylua.toml

@@ -0,0 +1,5 @@
+column_width = 120
+line_endings = "Unix"
+indent_type = "Spaces"
+indent_width = 4
+quote_style = "AutoPreferDouble"

lsp/copilot.lua

@@ -0,0 +1,53 @@
+---@type vim.lsp.Config
+return {
+    name = "copilot",
+    cmd = {
+        "copilot-language-server",
+        "--stdio",
+    },
+    init_options = {
+        editorInfo = { name = "neovim", version = "0.11" },
+        editorPluginInfo = {
+            name = "Github Copilot LSP for Neovim",
+            version = "0.0.1",
+        },
+    },
+    settings = {
+        nextEditSuggestions = {
+            enabled = true,
+        },
+    },
+    root_markers = { ".git" },
+    on_init = function(client)
+        local nes = require("copilot-lsp.nes")
+        local inline_completion = require("copilot-lsp.completion")
+
+        vim.keymap.set("i", "<c-i>", function()
+            inline_completion.request_inline_completion(1)
+        end)
+
+        vim.keymap.set("n", "<leader>x", function()
+            nes.request_nes(client)
+        end)
+
+        local au = vim.api.nvim_create_augroup("copilot-language-server", { clear = true })
+        vim.api.nvim_create_autocmd("TextChangedI", {
+            callback = function()
+                inline_completion.request_inline_completion(2)
+            end,
+            group = au,
+        })
+
+        vim.api.nvim_create_autocmd("BufEnter", {
+            callback = function()
+                local td_params = vim.lsp.util.make_text_document_params()
+                client:notify("textDocument/didFocus", {
+                    textDocument = {
+                        uri = td_params.uri,
+                    },
+                })
+            end,
+            group = au,
+        })
+    end,
+}

lua/copilot-lsp/completion/init.lua

@@ -0,0 +1,40 @@
+local M = {}
+
+---@param results table<integer, { err: lsp.ResponseError, result: lsp.InlineCompletionList}>
+---@param _ctx lsp.HandlerContext
+---@param _config table
+local function handle_inlineCompletion_response(results, _ctx, _config)
+    -- Filter errors from results
+    local results1 = {} --- @type table<integer,lsp.InlineCompletionList>
+
+    for client_id, resp in pairs(results) do
+        local err, result = resp.err, resp.result
+        if err then
+            vim.lsp.log.error(err.code, err.message)
+        elseif result then
+            results1[client_id] = result
+        end
+    end
+
+    for _, result in pairs(results1) do
+        -- This is where we show the completion results
+        dd(result)
+    end
+end
+
+---@param type lsp.InlineCompletionTriggerKind
+function M.request_inline_completion(type)
+    local params = vim.tbl_deep_extend("keep", vim.lsp.util.make_position_params(0, "utf-16"), {
+        textDocument = vim.lsp.util.make_text_document_params(),
+        position = vim.lsp.util.make_position_params(0, "utf-16"),
+        context = {
+            triggerKind = type,
+        },
+        formattingOptions = {
+            tabSize = 4,
+            insertSpaces = true,
+        },
+    })
+    vim.lsp.buf_request_all(0, "textDocument/inlineCompletion", params, handle_inlineCompletion_response)
+end
+return M

lua/copilot-lsp/errors.lua

@@ -0,0 +1,3 @@
+return {
+    ErrNotStarted = "Copilot LSP not started",
+}

lua/copilot-lsp/nes/init.lua

@@ -0,0 +1,37 @@
+local errs = require("copilot-lsp.errors")
+local utils = require("copilot-lsp.util")
+
+local M = {}
+
+---@param err lsp.ResponseError?
+---@param result copilotInlineEditResponse
+local function handle_nes_response(err, result)
+    if err then
+        vim.notify(err.message)
+        return
+    end
+    if #result.edits > 1 then
+        vim.notify("more than 1 edit, dont know what to do yet")
+        return
+    end
+
+    if #result.edits == 0 then
+        vim.notify("no edits")
+        return
+    end
+
+    local edit = result.edits[1]
+    utils.apply_inline_edit(edit)
+end
+
+---@param copilot_lss vim.lsp.Client?
+function M.request_nes(copilot_lss)
+    local pos_params = vim.lsp.util.make_position_params(0, "utf-16")
+    local version = vim.lsp.util.buf_versions[vim.api.nvim_get_current_buf()]
+    assert(copilot_lss, errs.ErrNotStarted)
+    ---@diagnostic disable-next-line: inject-field
+    pos_params.textDocument.version = version
+    copilot_lss:request("textDocument/copilotInlineEdit", pos_params, handle_nes_response)
+end
+
+return M

lua/copilot-lsp/types.lua

@@ -0,0 +1,8 @@
+---@class copilotInlineEdit
+---@field command lsp.Command
+---@field range lsp.Range
+---@field text string
+---@field textDocument lsp.VersionedTextDocumentIdentifier
+
+---@class copilotInlineEditResponse
+---@field edits copilotInlineEdit[]

lua/copilot-lsp/util.lua

@@ -0,0 +1,20 @@
+local M = {}
+---@param edit copilotInlineEdit
+function M.apply_inline_edit(edit)
+    local bufnr = vim.uri_to_bufnr(edit.textDocument.uri)
+    local multi_line
+    if edit.text:match("\n") then
+        multi_line = vim.split(edit.text, "\n")
+    end
+
+    vim.api.nvim_buf_set_text(
+        bufnr,
+        edit.range.start.line,
+        edit.range.start.character,
+        edit.range["end"].line,
+        edit.range["end"].character,
+        multi_line or { edit.text }
+    )
+end
+
+return M