copilot.lua/lua/copilot/command.lua at master · MunifTanjim/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
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
    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
    end

    local should_attach, no_attach_reason = u.should_attach(c.params.filetypes)
    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(c.params.filetypes)
    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