dotfiles/nvim/init.lua

require "kosh.options"
require "kosh.keymaps"
require "kosh.plugins"


-- Fold
require("fold-preview").setup()
require("pretty-fold").setup({
    sections = {
        left = {
            'content',
        },
        right = {
            ' ', 'number_of_folded_lines',
            function(config) return config.fill_char:rep(3) end
        }
    },
    fill_char = " "
})
require("nvim-treesitter").setup {
    highlight = {
        enable = true,
    },
    endent = { enable = true },
    fold = { enable = true },
}


-- Scratchpad setup
file_path = vim.uri_to_fname(vim.uri_from_bufnr(0))

if (file_path == "/home/kosh/.backup/notes.txt") then
    vim.api.nvim_set_keymap("n", "ZZ", ":w<CR>:!i3-msg \"[instance=\"notes\"] move scratchpad\"<CR><CR>", {noremap = true})
end


-- MISC
vim.cmd "colorscheme onedarker"

require("lualine").setup {
    theme = "palenight"
}

-- AS
require("autosave").setup({
    enabled = true
})


-- Git signs
require('gitsigns').setup()

-- tabline
require("bufferline").setup{}
require('nvim-web-devicons').setup{}

-- Jupyter
require('notebook').setup()
local api = require("notebook.api")
local settings = require("notebook.settings")

function _G.define_cell(extmark)
    if extmark == nil then
        local line = vim.fn.line(".")
        extmark, _ = api.current_extmark(line)
    end
    local start_line = extmark[1] + 1
    local end_line = extmark[3].end_row
    pcall(function() vim.fn.MagmaDefineCell(start_line, end_line) end)
end

function _G.define_all_cells()
    local buffer = vim.api.nvim_get_current_buf()
    local extmarks = settings.extmarks[buffer]
    for id, cell in pairs(extmarks) do
        local extmark = vim.api.nvim_buf_get_extmark_by_id(
            0, settings.plugin_namespace, id, { details = true }
        )
        if cell.cell_type == "code" then
            define_cell(extmark)
        end
    end
end

vim.api.nvim_create_autocmd(
    { "BufRead", },
    { pattern = { "*.ipynb" }, command = "MagmaInit" }
)
vim.api.nvim_create_autocmd(
     "User",
    { pattern = { "MagmaInitPost", "NBPostRender" }, callback = _G.define_all_cells }
)