copilot.lua/lua/copilot/command.lua at master · softstone1/copilot.lua · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
local a = require("copilot.api")
local c = require("copilot.client")
local u = require("copilot.util")

local mod = {}

function mod.version()
  local info = u.get_editor_info()

  local lines = {
    info.editorInfo.name .. " " .. info.editorInfo.version,
    info.editorPluginInfo.name .. " " .. info.editorPluginInfo.version,
    "copilot.lua" .. " " .. u.get_copilot_lua_version(),
  }

  local client = u.get_copilot_client()

  coroutine.wrap(function()
    if client then
      local _, data = a.get_version(client)
      lines[#lines + 1] = "copilot/dist/agent.js" .. " " .. data.version
      lines[#lines + 1] = "Node.js" .. " " .. c.get_node_version()
    else
      lines[#lines + 1] = "copilot/dist/agent.js" .. " " .. "not running"
    end

    vim.api.nvim_echo(
      vim.tbl_map(function(line)
        return { line .. "\n" }
      end, lines),
      true,
      {}
    )
  end)()
end

function mod.status()
  local lines = {}

  local function add_line(line)
    lines[#lines + 1] = { "[Copilot] " .. line .. "\n" }
  end

  local function flush_lines(last_line)
    if last_line then
      add_line(last_line)
    end

    vim.api.nvim_echo(lines, true, {})
  end

  local client = u.get_copilot_client()
  if not client then
    flush_lines("Not running")
    return
  end

  add_line("Online")

  coroutine.wrap(function()
    local cserr, status = a.check_status(client)
    if cserr then
      flush_lines(cserr)
      return
    end

    if not status.user then
      flush_lines("Not authenticated. Run ':Copilot auth'")
      return
    elseif status.status == 'NoTelemetryConsent' then
      flush_lines("Telemetry terms not accepted")
      return
    elseif status.status == 'NotAuthorized' then
      flush_lines("Not authorized")
      return
    end

    local should_attach, no_attach_reason = u.should_attach()
    local is_attached = u.is_attached(client)
    if is_attached then
      if not should_attach then
        add_line("Enabled manually (" .. no_attach_reason .. ")")
      else
        add_line("Enabled for " .. vim.bo.filetype)
      end
    elseif not is_attached then
      if should_attach then
        add_line("Disabled manually for " .. vim.bo.filetype)
      else
        add_line("Disabled (" .. no_attach_reason .. ")")
      end
    end

    if string.lower(a.status.data.status) == "error" then
      add_line(a.status.data.message)
    end

    flush_lines()
  end)()
end

---@param opts? { force?: boolean }
function mod.toggle(opts)
  opts = opts or {}

  local client = u.get_copilot_client()
  if not client then
    return
  end

  if u.is_attached(client) then
    c.buf_detach(client)
    return
  end

  if not opts.force then
    local should_attach, no_attach_reason = u.should_attach()
    if not should_attach then
      vim.api.nvim_echo({
        { "[Copilot] " .. no_attach_reason .. "\n" },
        { "[Copilot] to force enable, run ':Copilot! toggle'" },
      }, true, {})
      return
    end

    opts.force = true
  end

  c.buf_attach(client, opts.force)
end

return mod