From 4dbd024b32bd322af317f01036e90202d945f605 Mon Sep 17 00:00:00 2001
From: ThePrimeagen <mpaulson@netflix.com>
Date: Fri, 19 Feb 2021 16:24:07 -0700
Subject: [PATCH] feat(yayayaya): Term and mark are now using configs

---
 lua/harpoon/init.lua     | 66 ++++++++++++++++++++++-----
 lua/harpoon/mark.lua     | 97 ++++++++++------------------------------
 lua/harpoon/term.lua     |  6 ++-
 plugin/manage-a-mark.vim |  2 +-
 plugin/mark.vim          |  5 +++
 5 files changed, 90 insertions(+), 86 deletions(-)
 create mode 100644 plugin/mark.vim

diff --git a/lua/harpoon/init.lua b/lua/harpoon/init.lua
index 51c54aa..4095995 100644
--- a/lua/harpoon/init.lua
+++ b/lua/harpoon/init.lua
@@ -1,24 +1,63 @@
+local Path = require("plenary.path")
 local terminals = require("harpoon.term")
-local manage = require("harpoon.mark")
+local mark = require("harpoon.mark")
 local cwd = cwd or vim.loop.cwd()
 local config_path = vim.fn.stdpath("config")
 local data_path = vim.fn.stdpath("data")
-local user_terminal_config = string.format("%s/harpoon-terminal.json", config_path)
-local cache_terminal_config = string.format("%s/harpoon-terminal.json", data_path)
+local user_config = string.format("%s/harpoon.json", config_path)
+local cache_config = string.format("%s/harpoon.json", data_path)
 
 local M = {}
 
-function expand_dir(projects) 
+--[[
+{
+    projects = {
+        ["/path/to/director"] = {
+            term = {
+                cmds = {
+                }
+                ... is there antyhnig that could be options?
+            },
+            mark = {
+                marks = {
+                }
+                ... is there antyhnig that could be options?
+            }
+        }
+    },
+    ... high level settings
+}
+--]]
+harpoon_config = harpoon_config or {}
+
+function expand_dir(config) 
+    local projects = config.projects or {}
     local expanded_config = {}
     for k in pairs(projects) do
         local expanded_path = Path.new(k):expand()
         projects[expanded_path] = projects[k]
     end
+
+    return config
+end
+
+M.save = function()
+    local term_config = terminals.get_config()
+    local mark_config = mark.get_config()
+
+    if not harpoon_config.projects[cwd] then
+        harpoon_config.projects[cwd] = {}
+    end
+
+    harpoon_config.projects[cwd].term = term_config
+    harpoon_config.projects[cwd].mark = mark_config
+
+    Path:new(cache_config):write(vim.fn.json_encode(harpoon_config), 'w')
 end
 
 -- 1. saved.  Where do we save?
 M.setup = function(config) 
-    function read_terminal_config(config)
+    function read_config(config)
         return vim.fn.json_decode(Path:new(config):read())
     end
 
@@ -26,22 +65,27 @@ M.setup = function(config)
         config = {}
     end
 
-    local ok, user_config = pcall(read_terminal_config, user_terminal_config)
-    local ok2, cache_config = pcall(read_terminal_config, data_terminal_config)
+    local ok, u_config = pcall(read_config, user_terminal_config)
+    local ok2, c_config = pcall(read_config, cache_config)
 
     if not ok then
-        user_config = {}
+        u_config = {}
     end
 
     if not ok2 then
-        cache_config = {}
+        c_config = {}
     end
 
     local complete_config = 
-        vim.tbl_deep_extend("force", {}, cache_config, user_config, config)
+        vim.tbl_deep_extend("force", 
+            {projects = {}}, 
+            expand_dir(c_config), 
+            expand_dir(u_config),
+            expand_dir(config))
 
     terminals.setup(complete_config)
-    manage.setup(complete_config)
+    mark.setup(complete_config)
+    harpoon_config = complete_config
 end
 
 return M
diff --git a/lua/harpoon/mark.lua b/lua/harpoon/mark.lua
index 123c40f..134c7b4 100644
--- a/lua/harpoon/mark.lua
+++ b/lua/harpoon/mark.lua
@@ -1,34 +1,9 @@
 local Path = require('plenary.path')
 
 local M = {}
-
 local cwd = cwd or vim.loop.cwd()
-local data_path = vim.fn.stdpath("data")
-local mark_config = {}
 
-cwd = cwd:gsub("/", "_")
-
-local file_name = string.format("%s/%s.cache", data_path, cwd)
-
---[[
-{
-    projects = {
-        ["/path/to/director"] = {
-            term = {
-                cmds = {
-                }
-                ... is there antyhnig that could be options?
-            },
-            mark = {
-                marks = {
-                }
-                ... is there antyhnig that could be options?
-            }
-        }
-    },
-    ... high level settings
-}
---]]
+mark_config = mark_config or {}
 
 function get_id_or_current_buffer(id)
     if id == nil then
@@ -38,25 +13,9 @@ function get_id_or_current_buffer(id)
     return id
 end
 
-function hydrate_from_cache()
-    ok, res = pcall(function()
-        local results = Path:new(file_name):read()
-        if results == nil then
-            return {}, {}
-        end
-        return vim.fn.json_decode(results), {}
-    end)
-
-    if ok then
-        return res, {}
-    end
-
-    return {}, {}
-end
-
 M.setup = function(config) 
     if not config.projects[cwd] then
-        mark_config = {}
+        mark_config = {marks = {}}
     else 
         mark_config = config.projects[cwd].mark
     end
@@ -66,22 +25,14 @@ M.get_config = function()
     return mark_config
 end
 
-M.save = function()
-    Path:new(file_name):write(vim.fn.json_encode(marked_files), 'w')
-end
-
-if not marked_files then
-    marked_files, marked_offsets = hydrate_from_cache() 
-end
-
 function get_index_of(item)
     if item == nil then
         error("You have provided a nil value to Harpoon, please provide a string rep of the file or the file idx.")
         return
     end
     if type(item) == 'string' then
-        for idx = 1, #marked_files do
-           if marked_files[idx] == item then
+        for idx = 1, #mark_config.marks do
+           if mark_config.marks[idx] == item then
                 return idx
             end
         end
@@ -93,7 +44,7 @@ function get_index_of(item)
         item = item + 1
     end
 
-    if item <= #marked_files and item >= 1 then
+    if item <= #mark_config.marks and item >= 1 then
         return item
     end
 
@@ -101,16 +52,16 @@ function get_index_of(item)
 end
 
 function valid_index(idx) 
-    return idx ~= nil and marked_files[idx] ~= nil
+    return idx ~= nil and mark_config.marks[idx] ~= nil
 end
 
 M.get_index_of = get_index_of
 M.valid_index = valid_index
 
 function swap(a_idx, b_idx) 
-    local tmp = marked_files[a_idx]
-    marked_files[a_idx] = marked_files[b_idx]
-    marked_files[b_idx] = tmp
+    local tmp = mark_config.marks[a_idx]
+    mark_config.marks[a_idx] = mark_config.marks[b_idx]
+    mark_config.marks[b_idx] = tmp
 end
 
 M.add_file = function()
@@ -120,14 +71,14 @@ M.add_file = function()
         return
     end
 
-    for idx = 1, #marked_files do
-        if marked_files[idx] == nil then
-            marked_files[idx] = buf_name
+    for idx = 1, #mark_config.marks do
+        if mark_config.marks[idx] == nil then
+            mark_config.marks[idx] = buf_name
             return
         end
     end
 
-    table.insert(marked_files, buf_name)
+    table.insert(mark_config.marks, buf_name)
 end
 
 M.store_offset = function() 
@@ -159,7 +110,7 @@ M.rm_file = function()
         return
     end
 
-    marked_files[idx] = nil
+    mark_config.marks[idx] = nil
 end
 
 M.trim = function() 
@@ -167,7 +118,7 @@ M.trim = function()
 end
 
 M.clear_all = function()
-    marked_files = {}
+    mark_config.marks = {}
 end
 
 M.promote = function(id)
@@ -194,13 +145,13 @@ end
 
 M.remove_nils = function()
     local next = {}
-    for idx = 1, #marked_files do
-        if marked_files[idx] ~= nil then
-            table.insert(next, marked_files[idx])
+    for idx = 1, #mark_config.marks do
+        if mark_config.marks[idx] ~= nil then
+            table.insert(next, mark_config.marks[idx])
         end
     end
 
-    marked_files = next
+    mark_config.marks = next
 end
 
 M.shorten_list = function(count) 
@@ -216,19 +167,19 @@ M.shorten_list = function(count)
     end
 
     local next = {}
-    local up_to = math.min(count, #marked_files)
+    local up_to = math.min(count, #mark_config.marks)
     for idx = 1, up_to do
-        table.insert(next, marked_files[idx])
+        table.insert(next, mark_config.marks[idx])
     end
-    marked_files = next
+    mark_config.marks = next
 end
 
 M.get_marked_file = function(idx)
-    return marked_files[idx]
+    return mark_config.marks[idx]
 end
 
 M.get_length = function() 
-    return #marked_files
+    return #mark_config.marks
 end
 
 return M
diff --git a/lua/harpoon/term.lua b/lua/harpoon/term.lua
index 98e44d6..27a326f 100644
--- a/lua/harpoon/term.lua
+++ b/lua/harpoon/term.lua
@@ -3,7 +3,7 @@ local cwd = cwd or vim.loop.cwd()
 
 local M = {}
 
-local terminal_config = { }
+terminal_config = terminal_config or { }
 local terminals = {}
 
 function create_terminal() 
@@ -23,6 +23,10 @@ function create_terminal()
     return buf_id, term_id
 end
 
+M.get_config = function() 
+    return terminal_config
+end
+
 --[[
 -- First iteration of the setup script
 lua require("harpoon").setup({
diff --git a/plugin/manage-a-mark.vim b/plugin/manage-a-mark.vim
index 448c706..25ccea4 100644
--- a/plugin/manage-a-mark.vim
+++ b/plugin/manage-a-mark.vim
@@ -1,5 +1,5 @@
 augroup THE_PRIMEAGEN_HARPOON
     autocmd!
-    autocmd VimLeave * :lua require('harpoon.mark').save()
+    autocmd VimLeave * :lua require('harpoon').save()
     autocmd BufLeave * :lua require('harpoon.mark').store_offset()
 augroup END
diff --git a/plugin/mark.vim b/plugin/mark.vim
new file mode 100644
index 0000000..25ccea4
--- /dev/null
+++ b/plugin/mark.vim
@@ -0,0 +1,5 @@
+augroup THE_PRIMEAGEN_HARPOON
+    autocmd!
+    autocmd VimLeave * :lua require('harpoon').save()
+    autocmd BufLeave * :lua require('harpoon.mark').store_offset()
+augroup END