main.lua

#!/usr/bin/env luajit
-- obsidian-code: Codebase Function Graph Visualizer
-- Usage:
--   love .                      (GUI mode)
--   luajit main.lua --cli <path> (CLI mode, JSON output)

-- Set up LÖVE2D's internal file loader to find our lua/ modules inside the .love archive
if love and love.filesystem then
    love.filesystem.setRequirePath("?.lua;?/init.lua;lua/?.lua;lua/ui/?.lua")
end

-- Set up standard Lua module paths for filesystem access
local sourceDir = love and love.filesystem.getSource() or (arg[0]:match("(.*/)") or "./")
-- Get the directory containing the .exe (sourceDir may be the .exe path on Windows fused builds)
local sourceParent = sourceDir:match("^(.+)[/\\][^/\\]+$") or sourceDir
package.path = sourceDir .. "/lua/?.lua;" .. sourceDir .. "/lua/ui/?.lua;" .. package.path

-- Add native C module search paths (.so on Linux, .dll on Windows)
local home = os.getenv("HOME") or ""
local isWindows = love and love.system and love.system.getOS() == "Windows"
if isWindows then
    -- On Windows fused builds, DLLs are next to the .exe
    package.cpath = sourceParent .. "/?.dll;"
        .. sourceParent .. "/?51.dll;"
        .. sourceDir .. "/?.dll;"
        .. ".\\?.dll;"
        .. package.cpath
else
    package.cpath = home .. "/.luarocks/lib/lua/5.1/?.so;"
        .. "/usr/local/lib/lua/5.1/?.so;"
        .. "/usr/lib/lua/5.1/?.so;"
        .. package.cpath
end

-- CLI mode: analyze and print JSON, no GUI
if arg and arg[1] == "--cli" then
    local cjson = require("cjson")
    local analyzer = require("analyzer")
    local path = arg[2]
    if not path then
        io.stderr:write("Usage: luajit main.lua --cli <project-path>\n")
        os.exit(1)
    end
    local ok, result = pcall(analyzer.analyze, path)
    if not ok then
        io.stderr:write("Error: " .. tostring(result) .. "\n")
        os.exit(1)
    end
    print(cjson.encode(result))
    os.exit(0)
end

-- GUI mode (LÖVE2D)
local colors = require("ui.colors")
local toolbar = require("ui.toolbar")
local sidebar = require("ui.sidebar")
local graphView = require("ui.graph_view")
local flowView = require("ui.flow_view")

local graphData = nil
local defaultFont = nil

-- On Windows, write to a log file on Desktop (SFX extracts to temp dir)
local logFile = nil
local function log(msg)
    io.stderr:write("[obsidian-code] " .. msg .. "\n")
    if not logFile then
        -- Try Desktop, then USERPROFILE, then next to exe
        local desktop = os.getenv("USERPROFILE")
        if desktop then
            logFile = io.open(desktop .. "\\Desktop\\obsidian-code.log", "w")
            if not logFile then
                logFile = io.open(desktop .. "\\obsidian-code.log", "w")
            end
        end
        if not logFile then
            logFile = io.open(sourceParent .. "/obsidian-code.log", "w")
        end
    end
    if logFile then
        logFile:write("[obsidian-code] " .. msg .. "\n")
        logFile:flush()
    end
end

local function analyzeProject(path)
    -- Wrap everything in pcall so errors never crash the app
    local success, errMsg = pcall(function()
        log("Analyzing: " .. path)
        log("sourceDir: " .. sourceDir)
        log("sourceParent: " .. sourceParent)
        log("package.cpath: " .. package.cpath)

        -- Load native C modules one at a time with logging
        log("Step 1: loading cjson...")
        local ok_cjson, cjson = pcall(require, "cjson")
        if not ok_cjson then
            toolbar.setError("cjson not found: " .. tostring(cjson))
            log("FAIL cjson: " .. tostring(cjson))
            return
        end
        log("Step 1: cjson OK")

        log("Step 2: loading lfs...")
        local ok_lfs, lfs = pcall(require, "lfs")
        if not ok_lfs then
            toolbar.setError("lfs not found: " .. tostring(lfs))
            log("FAIL lfs: " .. tostring(lfs))
            return
        end
        log("Step 2: lfs OK")

        log("Step 3: loading analyzer...")
        local ok_analyzer, analyzer = pcall(require, "analyzer")
        if not ok_analyzer then
            toolbar.setError("analyzer load failed: " .. tostring(analyzer))
            log("FAIL analyzer: " .. tostring(analyzer))
            return
        end
        log("Step 3: analyzer OK")

        -- Tell analyzer where to find tree-sitter .so/.dll files
        analyzer.base_dir = sourceParent
        analyzer.log = log  -- share file-based logging
        log("Step 4: base_dir=" .. analyzer.base_dir)

        log("Step 5: calling analyzer.analyze(" .. path .. ")...")
        local ok, result = pcall(analyzer.analyze, path)
        if not ok then
            toolbar.setError(tostring(result))
            log("FAIL analyze: " .. tostring(result))
            return
        end

        log("Step 6: Done: " .. #result.nodes .. " nodes, " .. #result.edges .. " edges")
        graphData = result

        local w = love.graphics.getWidth()
        local h = love.graphics.getHeight()
        local sidebarW = sidebar.getWidth()
        local toolbarH = toolbar.getHeight()

        graphView.setGraph(result, w - sidebarW, h - toolbarH)
        sidebar.setGraph(result)
        flowView.setGraph(result)
        flowView.reset()
        toolbar.setViewMode("network")
    end)

    if not success then
        toolbar.setError("Crash: " .. tostring(errMsg))
        log("CRASH in analyzeProject: " .. tostring(errMsg))
    end
end

function love.load()
    log("=== obsidian-code starting ===")
    log("OS: " .. (love.system.getOS and love.system.getOS() or "unknown"))
    log("sourceDir: " .. sourceDir)
    log("sourceParent: " .. sourceParent)
    log("package.cpath: " .. package.cpath)
    love.window.setTitle("obsidian-code")
    love.graphics.setBackgroundColor(colors.base)

    defaultFont = love.graphics.newFont(13)
    love.graphics.setFont(defaultFont)

    graphView.init()

    -- Wire up callbacks
    toolbar.setCallbacks(analyzeProject, function(mode)
        -- View mode changed
    end)

    sidebar.setCallbacks(
        function(nodeId) -- select node
            toolbar.setViewMode("flow")
            flowView.selectNode(nodeId)
        end,
        nil -- filter file (not implemented yet)
    )

    flowView.setCallbacks(function(nodeId)
        -- Navigated to a node in flow view
    end)

    -- Load demo data for quick testing
    loadDemo()
end

function loadDemo()
    graphData = {
        root_dir = "/demo",
        files = { "src/auth.ts", "src/api.ts", "src/db.ts", "src/utils.ts" },
        nodes = {
            { id = "src/auth.ts:login:5", name = "login", file_path = "src/auth.ts", line = 5, kind = "function", params = { "username", "password" }, is_exported = true, is_async = true },
            { id = "src/auth.ts:validateToken:20", name = "validateToken", file_path = "src/auth.ts", line = 20, kind = "function", params = { "token" }, is_exported = true, is_async = false },
            { id = "src/auth.ts:hashPassword:35", name = "hashPassword", file_path = "src/auth.ts", line = 35, kind = "arrow", params = { "pw" }, is_exported = false, is_async = false },
            { id = "src/api.ts:fetchUser:3", name = "fetchUser", file_path = "src/api.ts", line = 3, kind = "function", params = { "id" }, is_exported = true, is_async = true },
            { id = "src/api.ts:createUser:15", name = "createUser", file_path = "src/api.ts", line = 15, kind = "function", params = { "data" }, is_exported = true, is_async = true },
            { id = "src/api.ts:deleteUser:30", name = "deleteUser", file_path = "src/api.ts", line = 30, kind = "function", params = { "id" }, is_exported = true, is_async = true },
            { id = "src/db.ts:query:2", name = "query", file_path = "src/db.ts", line = 2, kind = "function", params = { "sql", "params" }, is_exported = true, is_async = true },
            { id = "src/db.ts:connect:20", name = "connect", file_path = "src/db.ts", line = 20, kind = "function", params = {}, is_exported = true, is_async = true },
            { id = "src/db.ts:disconnect:35", name = "disconnect", file_path = "src/db.ts", line = 35, kind = "function", params = {}, is_exported = true, is_async = true },
            { id = "src/utils.ts:formatDate:1", name = "formatDate", file_path = "src/utils.ts", line = 1, kind = "arrow", params = { "date" }, is_exported = true, is_async = false },
            { id = "src/utils.ts:sanitize:10", name = "sanitize", file_path = "src/utils.ts", line = 10, kind = "arrow", params = { "input" }, is_exported = true, is_async = false },
            { id = "src/utils.ts:log:18", name = "log", file_path = "src/utils.ts", line = 18, kind = "arrow", params = { "msg" }, is_exported = false, is_async = false },
        },
        edges = {
            { source = "src/auth.ts:login:5", target = "src/api.ts:fetchUser:3", line = 8, is_conditional = false, is_in_loop = false, order = 1 },
            { source = "src/auth.ts:login:5", target = "src/auth.ts:hashPassword:35", line = 7, is_conditional = false, is_in_loop = false, order = 0 },
            { source = "src/auth.ts:login:5", target = "src/utils.ts:log:18", line = 12, is_conditional = false, is_in_loop = false, order = 2 },
            { source = "src/auth.ts:validateToken:20", target = "src/db.ts:query:2", line = 22, is_conditional = false, is_in_loop = false, order = 0 },
            { source = "src/api.ts:fetchUser:3", target = "src/db.ts:query:2", line = 6, is_conditional = false, is_in_loop = false, order = 0 },
            { source = "src/api.ts:fetchUser:3", target = "src/utils.ts:sanitize:10", line = 9, is_conditional = true, is_in_loop = false, order = 1 },
            { source = "src/api.ts:createUser:15", target = "src/db.ts:query:2", line = 18, is_conditional = false, is_in_loop = false, order = 0 },
            { source = "src/api.ts:createUser:15", target = "src/auth.ts:hashPassword:35", line = 17, is_conditional = false, is_in_loop = false, order = 1 },
            { source = "src/api.ts:deleteUser:30", target = "src/db.ts:query:2", line = 32, is_conditional = false, is_in_loop = false, order = 0 },
            { source = "src/api.ts:deleteUser:30", target = "src/utils.ts:log:18", line = 34, is_conditional = false, is_in_loop = false, order = 1 },
            { source = "src/db.ts:query:2", target = "src/db.ts:connect:20", line = 5, is_conditional = true, is_in_loop = false, order = 0 },
        },
    }

    local w = love.graphics.getWidth()
    local h = love.graphics.getHeight()
    local sidebarW = sidebar.getWidth()
    local toolbarH = toolbar.getHeight()

    graphView.setGraph(graphData, w - sidebarW, h - toolbarH)
    sidebar.setGraph(graphData)
    flowView.setGraph(graphData)
end

function love.update(dt)
    toolbar.update(dt)
    graphView.update(dt)

    -- Update cursor
    local cursor = graphView.getCursor()
    if cursor then
        love.mouse.setCursor(cursor)
    else
        love.mouse.setCursor()
    end
end

function love.draw()
    local w = love.graphics.getWidth()
    local h = love.graphics.getHeight()
    local toolbarH = toolbar.getHeight()
    local sidebarW = sidebar.getWidth()

    -- Toolbar (full width, top)
    love.graphics.setFont(defaultFont)
    toolbar.draw(w)

    -- Sidebar (left, below toolbar)
    sidebar.draw(0, toolbarH, sidebarW, h - toolbarH)

    -- Main content area (right of sidebar, below toolbar)
    local contentX = sidebarW
    local contentY = toolbarH
    local contentW = w - sidebarW
    local contentH = h - toolbarH

    -- Background for content
    love.graphics.setColor(colors.mantle)
    love.graphics.rectangle("fill", contentX, contentY, contentW, contentH)

    love.graphics.setFont(defaultFont)
    if toolbar.getViewMode() == "network" then
        graphView.draw(contentX, contentY, contentW, contentH)
    else
        flowView.draw(contentX, contentY, contentW, contentH)
    end

    -- Welcome screen if no data
    if not graphData then
        love.graphics.setColor(colors.overlay0)
        love.graphics.printf("Enter a project path and click Analyze", contentX, contentY + contentH / 2 - 10, contentW, "center")
    end

    -- Error bar on top of everything
    love.graphics.setFont(defaultFont)
    toolbar.drawError(w)
end

function love.mousepressed(x, y, button)
    local w = love.graphics.getWidth()
    local h = love.graphics.getHeight()
    local toolbarH = toolbar.getHeight()
    local sidebarW = sidebar.getWidth()

    -- Toolbar gets first priority
    if toolbar.mousepressed(x, y, button, w) then return end

    -- Sidebar
    if sidebar.mousepressed(x, y, button, 0, toolbarH, sidebarW, h - toolbarH) then return end

    -- Content area
    local contentX = sidebarW
    local contentY = toolbarH
    local contentW = w - sidebarW
    local contentH = h - toolbarH

    if toolbar.getViewMode() == "network" then
        graphView.mousepressed(x, y, button, contentX, contentY, contentW, contentH)
    else
        flowView.mousepressed(x, y, button, contentX, contentY, contentW, contentH)
    end
end

function love.mousereleased(x, y, button)
    if sidebar.mousereleased(x, y, button) then return end
    local clickedId = graphView.mousereleased(x, y, button)
    if clickedId then
        -- Node was clicked — switch to flow view
        toolbar.setViewMode("flow")
        flowView.selectNode(clickedId)
    end
end

function love.mousemoved(x, y, dx, dy)
    if sidebar.mousemoved(x, y) then return end

    local w = love.graphics.getWidth()
    local h = love.graphics.getHeight()
    local toolbarH = toolbar.getHeight()
    local sidebarW = sidebar.getWidth()
    local contentX = sidebarW
    local contentY = toolbarH
    local contentW = w - sidebarW
    local contentH = h - toolbarH

    graphView.mousemoved(x, y, contentX, contentY, contentW, contentH)
end

function love.wheelmoved(wx, wy)
    local mx, my = love.mouse.getPosition()
    local w = love.graphics.getWidth()
    local h = love.graphics.getHeight()
    local toolbarH = toolbar.getHeight()
    local sidebarW = sidebar.getWidth()

    -- Sidebar scroll
    if sidebar.wheelmoved(wx, wy, mx, my, 0, toolbarH, sidebarW, h - toolbarH) then return end

    -- Flow view scroll
    if toolbar.getViewMode() == "flow" then
        local contentX = sidebarW
        local contentY = toolbarH
        local contentW = w - sidebarW
        local contentH = h - toolbarH
        if flowView.wheelmoved(wx, wy, mx, my, contentX, contentY, contentW, contentH) then return end
    end

    -- Graph zoom
    if toolbar.getViewMode() == "network" then
        graphView.wheelmoved(wx, wy, mx, my, sidebarW, toolbarH)
    end
end

function love.textinput(t)
    toolbar.textinput(t)
end

function love.keypressed(key)
    if toolbar.keypressed(key) then return end

    -- Global shortcuts
    if key == "escape" then
        love.event.quit()
    end
end

function love.resize(w, h)
    -- Re-settle graph for new dimensions if we have data
    if graphData then
        local sidebarW = sidebar.getWidth()
        local toolbarH = toolbar.getHeight()
        graphView.setGraph(graphData, w - sidebarW, h - toolbarH)
    end
end