local core = require "core" local command = require "core.command" local common = require "core.common" local DocView = require "core.docview" local project = {} local function has_no_locked_children(node) if node.locked then return false end if node.type == "leaf" then return true end return has_no_locked_children(node.a) and has_no_locked_children(node.b) end local function get_unlocked_root(node) if node.type == "leaf" then return not node.locked and node end if has_no_locked_children(node) then return node end return get_unlocked_root(node.a) or get_unlocked_root(node.b) end local function save_view(view) local mt = getmetatable(view) if mt == DocView then return { type = "doc", active = (core.active_view == view), filename = view.doc.filename, selection = { view.doc:get_selection() }, scroll = { x = view.scroll.to.x, y = view.scroll.to.y }, text = not view.doc.filename and view.doc:get_text(1, 1, math.huge, math.huge) } end for name, mod in pairs(package.loaded) do if mod == mt then return { type = "view", active = (core.active_view == view), module = name } end end end local function load_view(t) if t.type == "doc" then local ok, doc = pcall(core.open_doc, t.filename) if not ok then return DocView(core.open_doc()) end local dv = DocView(doc) if t.text then doc:insert(1, 1, t.text) end doc:set_selection(table.unpack(t.selection)) dv.last_line, dv.last_col = doc:get_selection() dv.scroll.x, dv.scroll.to.x = t.scroll.x, t.scroll.x dv.scroll.y, dv.scroll.to.y = t.scroll.y, t.scroll.y return dv end return require(t.module)() end local function save_node(node) local res = {} res.type = node.type if node.type == "leaf" then res.views = {} for _, view in ipairs(node.views) do local t = save_view(view) if t then table.insert(res.views, t) if node.active_view == view then res.active_view = #res.views end end end else res.divider = node.divider res.a = save_node(node.a) res.b = save_node(node.b) end return res end local function load_node(node, t) if t.type == "leaf" then local res for _, v in ipairs(t.views) do local view = load_view(v) if v.active then res = view end node:add_view(view) end if t.active_view then node:set_active_view(node.views[t.active_view]) end return res else node:split(t.type == "hsplit" and "right" or "down") node.divider = t.divider local res1 = load_node(node.a, t.a) local res2 = load_node(node.b, t.b) return res1 or res2 end end function project.save_workspace(filename) local root = get_unlocked_root(core.root_view.root_node) local fp = io.open(filename, "w") if fp then local node_text = common.serialize(save_node(root)) local topdir_entries = {} for _, entry in ipairs(core.project_entries) do if entry.item.topdir then table.insert(topdir_entries, {path = entry.name, type = entry.item.type}) end end local project_entries_text = common.serialize(topdir_entries) fp:write(string.format( "return { project_name = %q, working_dir = %q, documents = %s, project_entries = %s }\n", core.project_name, core.working_dir, node_text, project_entries_text)) fp:close() end end function project.load(name) core.project_name = name local filename = common.path_join(USERDIR, "projects", name .. ".lua") project.load_workspace(filename) core.log("Loaded project %s.", core.project_name) core.reschedule_project_scan() end function project.load_workspace(filename) local load = loadfile(filename) local workspace = load and load() -- FIXME: decide, error or return a success code if not workspace then error("Cannot load workspace") end if workspace then local root = get_unlocked_root(core.root_view.root_node) local active_view = load_node(root, workspace.documents) if active_view then core.set_active_view(active_view) end core.project_name = workspace.project_name core.project_entries = {} for _, entry in ipairs(workspace.project_entries) do if entry.type == "dir" then core.add_project_directory(entry.path) elseif entry.type == "dir" then core.add_project_file(entry.path) end end system.chdir(workspace.working_dir) end end local function suggest_directory(text) text = common.home_expand(text) return common.home_encode_list(text == "" and core.recents_open.dir or common.dir_path_suggest(text)) end command.add(nil, { ["project:save-as"] = function() local entry = core.project_entries[1] if entry then core.command_view:set_text(entry.item.filename) end core.command_view:enter("Save Project As", function(text) -- FIXME: add sanity check of project name. core.project_name = text local filename = common.path_join(USERDIR, "projects", text .. ".lua") save_workspace(filename) core.log("Saved project %s.", core.project_name) end) end, ["project:save"] = function() if core.project_name == "" then core.command_view:enter("Save Project As", function(text) core.project_name = text end) end local filename = common.path_join(USERDIR, "projects", core.project_name .. ".lua") save_workspace(filename) core.set_recent_project(core.project_name) core.log("Saved project %s.", core.project_name) end, ["project:load"] = function() core.command_view:enter("Load Project", function(text) project.load(text) core.set_recent_project(core.project_name) end) end, ["project:open-directory"] = function() core.command_view:enter("Open Directory", function(text, item) text = system.absolute_path(common.home_expand(item and item.text or text)) if text == core.working_dir then return end local path_stat = system.get_file_info(text) if not path_stat or path_stat.type ~= 'dir' then core.error("Cannot open folder %q", text) return end core.confirm_close_all(core.new_project_from_directory, text) end, suggest_directory) end, }) return project