init.lua

-- SET USERNAME BELOW

local username = "nnadar" -- put purdue username between quotes


------------------------------------------------------------
-- Leader Key
------------------------------------------------------------
vim.g.mapleader = " "

------------------------------------------------------------
-- Basic UI
------------------------------------------------------------
vim.opt.number = true
vim.opt.relativenumber = true
vim.opt.cursorline = true
vim.opt.mouse = "a"
vim.opt.splitright = true
vim.opt.splitbelow = true
vim.opt.clipboard = "unnamedplus"
vim.opt.termguicolors = true
vim.opt.ignorecase = true
vim.opt.smartcase = true
vim.opt.expandtab = true
vim.opt.shiftwidth = 4
vim.opt.tabstop = 4
vim.opt.smartindent = true
vim.opt.scrolloff = 8
vim.opt.signcolumn = "yes"
vim.opt.wrap = false

-- Reduce the time Neovim waits for a mapped sequence to complete (300ms from 1000ms)
vim.opt.timeoutlen = 300

------------------------------------------------------------
-- Transparency
------------------------------------------------------------
vim.cmd [[
  hi Normal guibg=NONE ctermbg=NONE
  hi NormalNC guibg=NONE ctermbg=NONE
  hi VertSplit guibg=NONE ctermbg=NONE
  hi StatusLine guibg=NONE ctermbg=NONE
  hi LineNr guibg=NONE ctermbg=NONE
  hi CursorLine guibg=NONE ctermbg=NONE
]]
vim.o.winblend = 20

------------------------------------------------------------
-- Safety: Clear existing maps to ensure clean slate
------------------------------------------------------------
pcall(vim.keymap.del, 'n', '<leader>t')
pcall(vim.keymap.del, 'n', '<leader>e')
pcall(vim.keymap.del, 'n', '<leader>l')
pcall(vim.keymap.del, 't', '<leader>t')
pcall(vim.keymap.del, 't', '<leader>e')
pcall(vim.keymap.del, 't', '<leader>l')

------------------------------------------------------------
-- Keymaps
------------------------------------------------------------
local map = vim.keymap.set
local opts = { noremap = true, silent = true }

-- Save/Quit
map("n", "<leader>w", ":w<CR>", opts)
map("n", "<leader>q", ":q<CR>", opts)

-- Window navigation
map("n", "<C-h>", "<C-w>h", opts)
map("n", "<C-j>", "<C-w>j", opts)
map("n", "<C-k>", "<C-w>k", opts)
map("n", "<C-l>", "<C-w>l", opts)

-- Split management
map("n", "<leader>sv", ":vsplit<CR>", opts)
map("n", "<leader>sh", ":split<CR>", opts)
map("n", "<leader>sc", ":close<CR>", opts)

------------------------------------------------------------
-- File Explorer (netrw) — <leader>E (Uppercase)
------------------------------------------------------------
vim.g.netrw_banner = 0
vim.g.netrw_liststyle = 3
vim.g.netrw_browse_split = 4
vim.g.netrw_altv = 1
vim.g.netrw_winsize = 25

map("n", "<leader>E", function()
    if vim.bo.filetype == "netrw" then
        vim.cmd("bd")
    else
        vim.cmd("Lexplore 25")
    end
end, opts)

------------------------------------------------------------
-- VS Code Style Copy / Paste
------------------------------------------------------------
map({'n','v'}, '<C-c>', '"+y', opts)
map('n', '<C-v>', '"+p', opts)
map('v', '<C-v>', '"+p', opts)
map('i', '<C-v>', '<C-r>+', opts)
map('i', '<C-c>', '<Esc>"+yyi', opts)

---
-- 🤖 AI Code Editor Integration ($\text{<leader>K}$)
---
-- Function to pipe code to the python script with an instruction
function _G.ai_edit_or_add_feature()
    -- 1. Determine the mode for the filter command
    local current_mode = vim.api.nvim_get_mode().mode

    local cmd_prefix
    if current_mode == 'v' or current_mode == 'V' or current_mode == 's' or current_mode == 'S' then
        -- Visual/Select mode: command applies only to the selection range (:'<,'>!)
        cmd_prefix = "'<,'>!"
        print("AI: Editing selected code...")
    else
        -- Normal mode: command applies to the entire file (: %!) <-- FIXED
        cmd_prefix = "%!"
        print("AI: Adding feature/editing entire file...")
    end

    -- 2. Ask user for the instruction
    vim.ui.input({ prompt = "AI Edit/Feature Instruction: " }, function(input)

        -- *** Exit/Cancel Logic ***
        -- If user presses <Esc> (input == nil) or presses Enter on empty prompt, cancel.
        if not input or input == "" then
            print("AI request cancelled. Press <Enter> to continue.")
            return
        end

        -- Escape single quotes in the instruction for bash safety
        local safe_input = input:gsub("'", "'\\''")

        -- Construct the full command

        local cmd = "bash -c 'source /homes/" .. username .. "/.config/nvim/ai/env/bin/activate && python3 /homes/ " .. username .. "/.config/nvim/ai/llm-editor.py \"" .. safe_input .. "\"'"

        -- Execute the filter command
        vim.cmd(cmd_prefix .. cmd)
    end)
end

-- Map K (Shift+k) in both Normal and Visual Mode to trigger the edit/feature-add
map({"n", "v"}, "<leader>K", "<cmd>lua ai_edit_or_add_feature()<CR>", opts)

---
-- 🖥️ Terminal Management
---

------------------------------------------------------------
-- Terminal Management (two terminals: main + llm)
------------------------------------------------------------
local main_term_buf = nil
local llm_term_buf = nil

-- === Bottom Terminal (<leader>T) ===
function _G.toggle_main_term()
    if main_term_buf and vim.api.nvim_buf_is_valid(main_term_buf) then
        for _, win in ipairs(vim.api.nvim_list_wins()) do
            if vim.api.nvim_win_get_buf(win) == main_term_buf then
                vim.api.nvim_win_close(win, true)
                return
            end
        end
        vim.cmd("botright split")
        vim.api.nvim_win_set_buf(0, main_term_buf)
        vim.cmd("resize 15")
        vim.cmd("startinsert")
        return
    end

    vim.cmd("botright split term://$SHELL")
    vim.cmd("resize 15")
    main_term_buf = vim.api.nvim_get_current_buf()
    vim.cmd("startinsert")
end

map("n", "<leader>T", "<cmd>lua toggle_main_term()<CR>", opts)
map("t", "<leader>T", "<C-\\><C-n><cmd>lua toggle_main_term()<CR>", opts)

-- === LLM Terminal (<leader>L) ===
function _G.toggle_llm_term()
    if llm_term_buf and vim.api.nvim_buf_is_valid(llm_term_buf) then
        for _, win in ipairs(vim.api.nvim_list_wins()) do
            if vim.api.nvim_win_get_buf(win) == llm_term_buf then
                vim.api.nvim_win_close(win, true)
                return
            end
        end
        vim.cmd("vsplit")
        vim.api.nvim_win_set_buf(0, llm_term_buf)
        vim.cmd("vertical resize 80")
        vim.cmd("startinsert")
        return
    end

    -- 1. Create the terminal buffer

    local cmd = "bash -c 'source /homes/" .. username .. "/.config/nvim/ai/env/bin/activate && python3 /homes/" .. username .. "/.config/nvim/ai/llm-manager.py'"
    vim.cmd("vsplit term://" .. cmd)
    vim.cmd("vertical resize 80")

    -- 2. Capture the new buffer ID
    llm_term_buf = vim.api.nvim_get_current_buf()

    -- 3. AutoCmd to close the window when the script finishes (FIX)
    vim.api.nvim_create_autocmd("TermClose", {
        buffer = llm_term_buf,
        once = true,
        callback = function()
            pcall(vim.cmd, "bdelete!")
            llm_term_buf = nil
        end
    })

    vim.cmd("startinsert")
end

map("n", "<leader>L", "<cmd>lua toggle_llm_term()<CR>", opts)
map("t", "<leader>L", "<C-\\><C-n><cmd>lua toggle_llm_term()<CR>", opts)

---
-- ⚙️ Quality of Life
---

------------------------------------------------------------
-- Format Key (Shift + Alt + F)
------------------------------------------------------------
map("n", "<S-A-f>", function()
    if vim.lsp and vim.lsp.buf and vim.lsp.buf.format then
        vim.lsp.buf.format({ async = true })
    else
        vim.cmd("normal! gg=G")
    end
end, opts)

------------------------------------------------------------
-- Buffer / Tab Management
------------------------------------------------------------
map("n", "<leader>bn", ":bnext<CR>", opts)
map("n", "<leader>bp", ":bprevious<CR>", opts)
map("n", "<leader>bd", ":bdelete<CR>", opts)

------------------------------------------------------------
-- Quality of Life
------------------------------------------------------------
vim.api.nvim_create_autocmd("TextYankPost", { callback = function()
    vim.highlight.on_yank({ timeout = 200 })
end })

vim.api.nvim_create_autocmd("BufWritePre", {
  pattern = "*",
  command = [[%s/\s\+$//e]],
})

vim.api.nvim_create_autocmd("BufReadPost", {
  callback = function()
      local mark = vim.api.nvim_buf_get_mark(0, '"')
      local lcount = vim.api.nvim_buf_line_count(0)
      if mark[1] > 0 and mark[1] <= lcount then
          pcall(vim.api.nvim_win_set_cursor, 0, mark)
      end
  end,
})

------------------------------------------------------------
-- End of Config
------------------------------------------------------------