From 2f50f21f571b3432f3056241e2581c4ac7b27f0c Mon Sep 17 00:00:00 2001
From: Francesco Abbate <francesco.bbt@gmail.com>
Date: Sat, 26 Dec 2020 16:48:08 +0100
Subject: [PATCH] Preliminary version of projects with multiple directories

---
 data/core/init.lua        |  4 +-
 data/plugins/treeview.lua | 81 +++++++++++++++++++++++++++------------
 2 files changed, 60 insertions(+), 25 deletions(-)

diff --git a/data/core/init.lua b/data/core/init.lua
index d64960ed..a5da4378 100644
--- a/data/core/init.lua
+++ b/data/core/init.lua
@@ -84,7 +84,7 @@ local function project_scan_thread()
         local file = (path ~= "." and path .. PATHSEP or "") .. file
         local info = system.get_file_info(file)
         if info and info.size < size_limit then
-          info.filename = file
+          info.filename = system.absolute_path(file)
           table.insert(info.type == "dir" and dirs or files, info)
           entries_count = entries_count + 1
           if entries_count > max_entries then break end
@@ -120,6 +120,7 @@ local function project_scan_thread()
           config.max_project_files.." files according to config.max_project_files.")
       end
       core.project_files = t
+      core.project_directories[1] = {filename = system.absolute_path("."), type = "dir"}
       core.redraw = true
     end
 
@@ -252,6 +253,7 @@ function core.init()
   core.docs = {}
   core.threads = setmetatable({}, { __mode = "k" })
   core.project_files = {}
+  core.project_directories = {}
   core.redraw = true
   core.visited_files = {}
   core.restart_request = false
diff --git a/data/plugins/treeview.lua b/data/plugins/treeview.lua
index 8aa53ffc..1e41b2c7 100644
--- a/data/plugins/treeview.lua
+++ b/data/plugins/treeview.lua
@@ -9,7 +9,7 @@ local View = require "core.view"
 config.treeview_size = 200 * SCALE
 
 local function get_depth(filename)
-  local n = 0
+  local n = 1
   for sep in filename:gmatch("[\\/]") do
     n = n + 1
   end
@@ -28,16 +28,36 @@ function TreeView:new()
 end
 
 
-function TreeView:get_cached(item)
+local function relative_filename(filename, dirname)
+  local n = #dirname
+  if filename:sub(1, n) == dirname then
+    return filename:sub(n + 1):match('[/\\](.*)')
+  end
+end
+
+
+local function belongs_to_directory(item, dirname)
+  return relative_filename(item.filename, dirname)
+end
+
+
+function TreeView:get_cached(item, dirname)
   local t = self.cache[item.filename]
   if not t then
     t = {}
-    t.filename = item.filename
-    t.abs_filename = system.absolute_path(item.filename)
+    if dirname then
+      local rel = relative_filename(item.filename, dirname)
+      -- FIXME: rel should never be nil here. to be verified.
+      t.filename = rel or item.filename
+      t.depth = get_depth(t.filename)
+    else
+      t.filename = item.filename:match("[^\\/]+$")
+      t.depth = 0
+    end
+    t.abs_filename = item.filename
     t.name = t.filename:match("[^\\/]+$")
-    t.depth = get_depth(t.filename)
     t.type = item.type
-    self.cache[t.filename] = t
+    self.cache[item.filename] = t
   end
   return t
 end
@@ -73,28 +93,41 @@ function TreeView:each_item()
     local h = self:get_item_height()
 
     local i = 1
-    while i <= #core.project_files do
-      local item = core.project_files[i]
-      local cached = self:get_cached(item)
-
-      coroutine.yield(cached, ox, y, w, h)
+    -- FIXME: make this loop more efficient. Currently it takes O(M * N)
+    -- where M = (directories number) and N = (files number)
+    -- It should take O(N).
+    for idir, dir_item in ipairs(core.project_directories) do
+      local dir_entry = self:get_cached(dir_item)
+      local dir_name = dir_item.filename
+      coroutine.yield(dir_entry, ox, y, w, h)
       y = y + h
-      i = i + 1
+      while i <= #core.project_files do
+        local item = core.project_files[i]
+        if belongs_to_directory(item, dir_name) then
+          local cached = self:get_cached(item, dir_name)
 
-      if not cached.expanded then
-        if cached.skip then
-          i = cached.skip
-        else
-          local depth = cached.depth
-          while i <= #core.project_files do
-            local filename = core.project_files[i].filename
-            if get_depth(filename) <= depth then break end
-            i = i + 1
+          coroutine.yield(cached, ox, y, w, h)
+          y = y + h
+          i = i + 1
+
+          if not cached.expanded then
+            if cached.skip then
+              i = cached.skip
+            else
+              local depth = cached.depth
+              while i <= #core.project_files do
+                local filename = relative_filename(core.project_files[i].filename, dir_name)
+                if get_depth(filename) <= depth then break end
+                i = i + 1
+              end
+              cached.skip = i
+            end
           end
-          cached.skip = i
+        else
+          i = i + 1
         end
-      end
-    end
+      end -- while files
+    end -- for directories
   end)
 end