githubmirror/lapce

lapce-proxy/src/cli.rs
219 строк6 KB
Dongdong Zhou
gutter for diff editor
07 июл 2023, 21:48
07 июл 2023, 21:48f3d459b
use std::path::PathBuf;

use anyhow::{anyhow, Error, Result};
use lapce_core::directory::Directory;
use lapce_rpc::{
    file::{LineCol, PathObject},
    proxy::{ProxyMessage, ProxyNotification},
    RpcMessage,
};

#[derive(Default, Debug, Clone, PartialEq, Eq, PartialOrd, Ord)]
pub enum PathObjectType {
    #[default]
    Directory,
    File,
}

pub fn parse_file_line_column(path: &str) -> Result<PathObject, Error> {
    if let Ok(path) = PathBuf::from(path).canonicalize() {
        return Ok(PathObject {
            is_dir: path.is_dir(),
            path,
            linecol: None,
        });
    }

    let pwd = std::env::current_dir().unwrap_or_default();

    let mut splits = path.rsplit(':').peekable();
    let (path, linecol) = if let Some(first_rhs) =
        splits.peek().and_then(|s| s.parse::<usize>().ok())
    {
        splits.next();
        if let Some(second_rhs) = splits.peek().and_then(|s| s.parse::<usize>().ok())
        {
            splits.next();
            let remaning: Vec<&str> = splits.rev().collect();
            let path = remaning.join(":");
            let path = PathBuf::from(path);
            let path = if let Ok(path) = path.canonicalize() {
                path
            } else {
                pwd.join(&path)
            };
            (
                path,
                Some(LineCol {
                    line: second_rhs,
                    column: first_rhs,
                }),
            )
        } else {
            let remaning: Vec<&str> = splits.rev().collect();
            let path = remaning.join(":");
            let path = PathBuf::from(path);
            let path = if let Ok(path) = path.canonicalize() {
                path
            } else {
                pwd.join(&path)
            };
            (
                path,
                Some(LineCol {
                    line: first_rhs,
                    column: 1,
                }),
            )
        }
    } else {
        (pwd.join(path), None)
    };

    Ok(PathObject {
        is_dir: path.is_dir(),
        path,
        linecol,
    })
}

pub fn try_open_in_existing_process(paths: &[PathObject]) -> Result<()> {
    let local_socket = Directory::local_socket()
        .ok_or_else(|| anyhow!("can't get local socket folder"))?;
    let mut socket =
        interprocess::local_socket::LocalSocketStream::connect(local_socket)?;

    let msg: ProxyMessage = RpcMessage::Notification(ProxyNotification::OpenPaths {
        paths: paths.to_vec(),
    });
    lapce_rpc::stdio::write_msg(&mut socket, msg)?;
    Ok(())
}

#[cfg(test)]
mod tests {
    use std::{env, path::PathBuf};

    use super::parse_file_line_column;
    use crate::cli::PathObject;

    #[test]
    #[cfg(windows)]
    fn test_absolute_path() {
        assert_eq!(
            parse_file_line_column("C:\\Cargo.toml:55").unwrap(),
            PathObject::new(PathBuf::from("C:\\Cargo.toml"), false, 55, 1),
        );
    }

    #[test]
    #[cfg(windows)]
    fn test_relative_path() {
        assert_eq!(
            parse_file_line_column(".\\..\\Cargo.toml:55").unwrap(),
            PathObject::new(
                PathBuf::from(".\\..\\Cargo.toml").canonicalize().unwrap(),
                false,
                55,
                1
            ),
        );
    }

    #[test]
    #[cfg(windows)]
    fn test_directory_looking_like_file() {
        assert_eq!(
            parse_file_line_column(".\\Cargo.toml\\").unwrap(),
            PathObject::from_path(
                env::current_dir().unwrap().join("Cargo.toml"),
                false
            ),
        );
    }

    #[test]
    #[cfg(unix)]
    fn test_absolute_path() {
        assert_eq!(
            parse_file_line_column("/tmp/Cargo.toml:55").unwrap(),
            PathObject::new(PathBuf::from("/tmp/Cargo.toml"), false, 55, 1),
        );
    }

    #[test]
    #[cfg(unix)]
    fn test_relative_path() {
        assert_eq!(
            parse_file_line_column("./../Cargo.toml").unwrap(),
            PathObject::from_path(
                PathBuf::from("./../Cargo.toml").canonicalize().unwrap(),
                false,
            ),
        );
    }

    #[test]
    #[cfg(unix)]
    fn test_directory_looking_like_file() {
        assert_eq!(
            parse_file_line_column("./Cargo.toml/").unwrap(),
            PathObject::from_path(
                env::current_dir().unwrap().join("Cargo.toml"),
                false
            ),
        );
    }

    #[test]
    fn test_current_dir() {
        assert_eq!(
            parse_file_line_column(".").unwrap(),
            PathObject::from_path(
                env::current_dir().unwrap().canonicalize().unwrap(),
                true
            ),
        );
    }

    #[test]
    fn test_relative_path_with_line() {
        assert_eq!(
            parse_file_line_column("Cargo.toml:55").unwrap(),
            PathObject::new(
                PathBuf::from("Cargo.toml").canonicalize().unwrap(),
                false,
                55,
                1
            ),
        );
    }

    #[test]
    fn test_relative_path_with_linecol() {
        assert_eq!(
            parse_file_line_column("Cargo.toml:55:3").unwrap(),
            PathObject::new(
                PathBuf::from("Cargo.toml").canonicalize().unwrap(),
                false,
                55,
                3
            ),
        );
    }

    #[test]
    fn test_relative_path_with_none() {
        assert_eq!(
            parse_file_line_column("Cargo.toml:12:623:352").unwrap(),
            PathObject::new(
                env::current_dir()
                    .unwrap()
                    .join(PathBuf::from("Cargo.toml:12")),
                false,
                623,
                352
            ),
        );
    }
}