feat(nvim): get project info

Author: kkharji

lua/xcodebase/init.lua

@@ -8,7 +8,15 @@ local address = vim.env.NVIM_LISTEN_ADDRESS
 ---@field ensure fun():boolean: When the a new server started it should return true
 ---@field is_running fun():boolean
 ---@field register fun(pid: number, root: string):boolean
-M.daemon = lib.daemon
+M.daemon = {}
+
+M.daemon.project_info = function(root)
+  require("xcodebase.state").projects[root] = nil
+  lib.daemon.project_info(pid, root)
+  -- while require("xcodebase.state").projects[root] == nil do
+  --   print "Wating"
+  -- end
+end
 
 ---@class XcodeBaseCommand
 local command = lib.command
@@ -37,8 +45,8 @@ M.try_register = function(opts)
   local root = vim.loop.cwd()
 
   if M.should_register(root, opts) then
-    local _ = M.daemon.ensure()
-    M.daemon.register(pid, root, address)
+    local _ = lib.daemon.ensure()
+    lib.daemon.register(pid, root, address)
     vim.cmd [[ autocmd VimLeavePre * lua require'xcodebase'.daemon.drop(vim.fn.getpid(), vim.loop.cwd())]]
   else
     return

lua/xcodebase/state.lua

@@ -0,0 +1,17 @@
+local M = {}
+
+---@class XcodeTarget
+---@field type string
+---@field platform string
+---@field sources string[]
+
+---@class XcodeProject
+---@field name string @Project name
+---@field targets string @Project name
+---@field root string @Project root
+
+---Holds project informations
+---@type table<string, XcodeProject>
+M.projects = {}
+
+return M

src/daemon.rs

@@ -40,6 +40,7 @@ pub enum DaemonRequest {
     Run(Run),
     RenameFile(RenameFile),
     Register(Register),
+    ProjectInfo(ProjectInfo),
     Drop(Drop),
 }
 
@@ -53,6 +54,7 @@ impl DaemonRequest {
             DaemonRequest::RenameFile(c) => c.handle(state).await,
             DaemonRequest::Register(c) => c.handle(state).await,
             DaemonRequest::Drop(c) => c.handle(state).await,
+            DaemonRequest::ProjectInfo(c) => c.handle(state).await,
         }
     }
 
@@ -65,6 +67,7 @@ impl DaemonRequest {
             RenameFile::KEY => Self::RenameFile(RenameFile::parse(args)?),
             Register::KEY => Self::Register(Register::parse(args)?),
             Drop::KEY => Self::Drop(Drop::parse(args)?),
+            ProjectInfo::KEY => Self::ProjectInfo(ProjectInfo::parse(args)?),
             cmd => anyhow::bail!("Unknown command messsage: {cmd}"),
         })
     }
@@ -79,6 +82,7 @@ impl Daemon {
         table.set("ensure", lua.create_function(Self::ensure)?)?;
         table.set("register", lua.create_function(Register::lua)?)?;
         table.set("drop", lua.create_function(Drop::lua)?)?;
+        table.set("project_info", lua.create_function(ProjectInfo::lua)?)?;
         Ok(table)
     }
 

src/daemon/requests.rs

@@ -1,11 +1,13 @@
 mod build;
 mod drop;
+mod project_info;
 mod register;
 mod rename_file;
 mod run;
 
 pub use build::Build;
 pub use drop::Drop;
+pub use project_info::ProjectInfo;
 pub use register::Register;
 pub use rename_file::RenameFile;
 pub use run::Run;

src/daemon/requests/project_info.rs

@@ -0,0 +1,70 @@
+#[cfg(feature = "mlua")]
+use crate::daemon::Daemon;
+
+#[cfg(feature = "daemon")]
+use crate::daemon::{DaemonRequestHandler, DaemonState};
+
+#[cfg(feature = "daemon")]
+use anyhow::{bail, Result};
+
+/// Return curr working direcotry project info
+#[derive(Debug)]
+pub struct ProjectInfo {
+    pub pid: i32,
+    pub root: String,
+}
+
+impl ProjectInfo {
+    pub const KEY: &'static str = "project_info";
+}
+
+#[cfg(feature = "daemon")]
+#[async_trait::async_trait]
+impl DaemonRequestHandler<ProjectInfo> for ProjectInfo {
+    fn parse(args: Vec<&str>) -> Result<Self> {
+        if let (Some(pid), Some(root)) = (args.get(0), args.get(1)) {
+            Ok(Self {
+                pid: pid.parse::<i32>()?,
+                root: root.to_string(),
+            })
+        } else {
+            anyhow::bail!("Missing arugments: {:?}", args)
+        }
+    }
+
+    async fn handle(&self, state: DaemonState) -> Result<()> {
+        tracing::info!("Getting info for {}", self.root);
+        let state = state.lock().await;
+
+        let workspace = match state.workspaces.get(&self.root) {
+            Some(o) => o,
+            None => bail!("No workspace for {}", self.root),
+        };
+
+        let nvim = match workspace.clients.get(&self.pid) {
+            Some(o) => o,
+            None => bail!("No nvim instance for {}", self.pid),
+        };
+
+        nvim.exec_lua(
+            &format!(
+                "require'xcodebase.state'.projects['{}'] = vim.json.decode([[{}]])",
+                self.root,
+                serde_json::to_string(&workspace.project)?
+            ),
+            vec![],
+        )
+        .await?;
+
+        Ok(())
+    }
+}
+
+#[cfg(feature = "lua")]
+impl ProjectInfo {
+    pub fn lua(lua: &mlua::Lua, (pid, root): (i32, String)) -> mlua::Result<()> {
+        use crate::util::mlua::LuaExtension;
+        lua.trace(&format!("grapping project info"))?;
+        Daemon::execute(&[Self::KEY, pid.to_string().as_str(), root.as_str()])
+    }
+}

src/daemon/state/project.rs

@@ -1,18 +1,21 @@
+#[cfg(feature = "daemon")]
+use serde::{Deserialize, Serialize};
+
 use std::collections::HashMap;
 use std::path::PathBuf;
 
 /// Represent Xcode Target
 #[derive(Debug)]
 #[allow(dead_code)]
-#[cfg_attr(feature = "serial", derive(serde::Deserialize))]
+#[cfg_attr(feature = "serial", derive(Deserialize, Serialize))]
 pub struct Target {
     r#type: String,
     platform: String,
     sources: Vec<PathBuf>,
 }
 
 #[derive(Debug, Default)]
-#[cfg_attr(feature = "serial", derive(serde::Deserialize))]
+#[cfg_attr(feature = "serial", derive(Deserialize, Serialize))]
 pub struct LocalConfig {
     pub ignore: Vec<String>,
 }
@@ -21,7 +24,7 @@ pub type TargetMap = HashMap<String, Target>;
 
 /// Represent XcodeGen Project
 #[derive(Debug)]
-#[cfg_attr(feature = "serial", derive(serde::Deserialize))]
+#[cfg_attr(feature = "serial", derive(Deserialize, Serialize))]
 pub struct Project {
     /// Project Name or rather xproj generated file name.
     name: String,