From 25df400bf36a2f6888546d58ae04c7afb6ba24a8 Mon Sep 17 00:00:00 2001
From: kyazdani42 <yazdani.kiyan@protonmail.com>
Date: Wed, 5 Feb 2020 01:07:06 +0100
Subject: [PATCH] add first draft

---
 README.md    | 12 ++++++----
 lua/tree.lua | 64 +++++++++++++++++++++++++++++++++++++++++-----------
 2 files changed, 59 insertions(+), 17 deletions(-)

diff --git a/README.md b/README.md
index 90b9c6af..1da0a617 100644
--- a/README.md
+++ b/README.md
@@ -2,19 +2,23 @@
 
 ## Notice
 
-This is under development, do not try to use it, it does not work.
+This is under development, do not try to use it right now. The best you could do is wait or give me tips to work better with lua.
 
-I am working on this plugin to learn lua, become better with neovim api and create a **simple** file explorer with features i need.
+I am working on this plugin to learn lua, learn neovim api and create a **simple** file explorer with features i need.
 
 I really don't like any of the vim trees, they are all too complicated for their purpose. I do have my shell to do most commands.
 
-
 ## Goals / Features
 
+### 0.1
 - [] moving around the file structure like any basic tree
-- [] open file in current buffer or in split with FzF like bindings (CR, C-v, C-x)
+- [x] open file in current buffer or in split with FzF like bindings (CR, C-v, C-x)
+
+### 0.2
 - [] add / delete file in directory
 - [] quickly find file in the directory structure
+
+### 0.3
 - [] icons for files
 - [] simple git integration (color of file changing when staged/changed)
 
diff --git a/lua/tree.lua b/lua/tree.lua
index 99fb74a0..ad15533a 100644
--- a/lua/tree.lua
+++ b/lua/tree.lua
@@ -2,9 +2,22 @@ local api = vim.api
 local function sys(v) return vim.fn.system(v) end
 local function syslist(v) return vim.fn.systemlist(v) end
 
-local dir_struct = syslist('ls')
-local BUF_NAME = 'LuaTree'
-print(dir_struct)
+local BUF_NAME = '_LuaTree_'
+
+-- TODO: think of a better way to implement the whole thing
+-- because right now the code is quite bad
+-- I shouldnt base the code on tree indentation to handle logic
+-- But for now its a first draft. It works a little
+
+-- TODO: maybe this should not be required, as the tree is only used in dev projects.
+-- In the README we should then precise to install vim-rooter.
+-- Or maybe we should just keep this functionnality
+local function add_dotdot(dirs)
+    table.insert(dirs, 1, '..')
+    return dirs
+end
+
+local dir_struct = add_dotdot(syslist('ls'))
 
 local function open()
     local win_width = 30
@@ -21,8 +34,9 @@ local function open()
         api.nvim_buf_set_option(buf, opt, val)
     end
 
-    api.nvim_command('topleft '..win_width..'vnew | set nonumber norelativenumber')
+    api.nvim_command('topleft '..win_width..'vnew')
     api.nvim_win_set_buf(0, buf)
+    api.nvim_command('setlocal nonumber norelativenumber winfixwidth winfixheight')
 end
 
 local function get_buf()
@@ -51,11 +65,13 @@ end
 
 local function update_view()
     local buf = get_buf();
-    if buf == nil then return end
+    if not buf then return end
 
+    local cursor_pos = api.nvim_win_get_cursor(0)
     api.nvim_buf_set_option(buf, 'modifiable', true)
     api.nvim_buf_set_lines(buf, 1, -1, false, dir_struct)
     api.nvim_buf_set_option(buf, 'modifiable', false)
+    api.nvim_win_set_cursor(0, cursor_pos)
 end
 
 local function is_dir(path)
@@ -70,6 +86,9 @@ local function close()
 end
 
 local function set_mappings()
+    local buf = get_buf()
+    if not buf then return end
+
     local mappings = {
         ['<CR>'] = 'open_file("edit")';
         ['<C-v>'] = 'open_file("vsplit")';
@@ -84,17 +103,36 @@ local function set_mappings()
     end
 end
 
-local function open_file(open_type)
-    local str = api.nvim_get_current_line()
-    if is_dir(str) then
-        local cur_dir = syslist('ls')
-        local sub_dir = syslist('ls')
-        -- local final_data = {}
+local function update_struct(folder_name)
+    local dirs = syslist('ls '..folder_name)
 
+    local index = 0
+    for i, v in pairs(dir_struct) do
+        if v == folder_name then
+            index = i
+            break
+        end
+    end
+
+    if string.match(dir_struct[index + 1] or '', '^  .*$') ~= nil then
+        while string.match(dir_struct[index + 1] or '', '^  .*$') ~= nil do
+            table.remove(dir_struct, index + 1)
+        end
+    else
+        for i, v in pairs(dirs) do
+            table.insert(dir_struct, index+i, '  '..v)
+        end
+    end
+end
+
+local function open_file(open_type)
+    local file = api.nvim_get_current_line()
+
+    if is_dir(file) then
+        update_struct(file)
         update_view()
     else
-        print(open_type)
-        -- api.nvim_command(open_type..' '..str)
+        api.nvim_command('wincmd l | '..open_type..' '..file)
     end
 end