local function custom_on_attach(client)
    local idris2 = require('idris2')
    idris2.show_implicits()
    idris2.show_machine_names()
    idris2.full_namespace()

    vim.keymap.set('n', '<localleader>o', require('idris2.code_action').generate_def)
    vim.keymap.set('n', '<localleader>c', require('idris2.code_action').case_split)
end

require('idris2').setup({server = {on_attach = custom_on_attach}})