cdx-rs 0.4.0

A fast and interactive cd command alternative for directory navigation
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136

// Copyright (c) 2026 Cinnamon-jp
// This software is released under the MIT License.
// https://opensource.org/licenses/MIT

use std::fmt;
use std::str::FromStr;

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum Shell {
    Bash,
    Zsh,
    Fish,
}

// pub type ShellType = Shell;

impl Shell {
    pub fn as_str(&self) -> &'static str {
        match self {
            Shell::Bash => "bash",
            Shell::Zsh => "zsh",
            Shell::Fish => "fish",
        }
    }
}

impl fmt::Display for Shell {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        write!(f, "{}", self.as_str())
    }
}

impl FromStr for Shell {
    type Err = String;

    fn from_str(s: &str) -> Result<Self, Self::Err> {
        match s.trim().to_lowercase().as_str() {
            "bash" => Ok(Shell::Bash),
            "zsh" => Ok(Shell::Zsh),
            "fish" => Ok(Shell::Fish),
            _ => Err(format!(
                "Unsupported shell '{}'. Supported shells: bash, zsh, fish",
                s.trim()
            )),
        }
    }
}

/// 指定されたシェル用のラッパー関数初期化スクリプトを返す
pub fn get_init_script(shell: Shell) -> &'static str {
    match shell {
        Shell::Bash | Shell::Zsh => {
            r#"function cdx() {
    if [ "$1" = "init" ]; then
        command cdx "$@"
        return
    fi
    local dest
    dest=$(command cdx "$@")
    if [ -n "$dest" ] && [ -d "$dest" ]; then
        builtin cd "$dest"
    fi
}
"#
        }
        Shell::Fish => {
            r#"function cdx
    if test (count $argv) -gt 0 -a "$argv[1]" = "init"
        command cdx $argv
        return
    end
    set dest (command cdx $argv)
    if test -n "$dest" -a -d "$dest"
        builtin cd "$dest"
    end
end
"#
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_shell_from_str_valid() {
        assert_eq!("bash".parse::<Shell>(), Ok(Shell::Bash));
        assert_eq!("BASH".parse::<Shell>(), Ok(Shell::Bash));
        assert_eq!(" bash ".parse::<Shell>(), Ok(Shell::Bash));
        assert_eq!("zsh".parse::<Shell>(), Ok(Shell::Zsh));
        assert_eq!("Zsh".parse::<Shell>(), Ok(Shell::Zsh));
        assert_eq!("fish".parse::<Shell>(), Ok(Shell::Fish));
        assert_eq!("FISH".parse::<Shell>(), Ok(Shell::Fish));
    }

    #[test]
    fn test_shell_from_str_invalid() {
        assert!("unknown".parse::<Shell>().is_err());
        assert!("sh".parse::<Shell>().is_err());
        assert!("".parse::<Shell>().is_err());
    }

    #[test]
    fn test_shell_display_and_as_str() {
        assert_eq!(Shell::Bash.as_str(), "bash");
        assert_eq!(Shell::Zsh.as_str(), "zsh");
        assert_eq!(Shell::Fish.as_str(), "fish");
        assert_eq!(Shell::Bash.to_string(), "bash");
        assert_eq!(Shell::Zsh.to_string(), "zsh");
        assert_eq!(Shell::Fish.to_string(), "fish");
    }

    #[test]
    fn test_get_init_script_bash_zsh() {
        let script_bash = get_init_script(Shell::Bash);
        assert!(script_bash.contains("function cdx()"));
        assert!(script_bash.contains("command cdx \"$@\""));
        assert!(script_bash.contains("builtin cd \"$dest\""));
        assert!(script_bash.contains("[ \"$1\" = \"init\" ]"));

        let script_zsh = get_init_script(Shell::Zsh);
        assert_eq!(script_bash, script_zsh);
        assert_eq!(get_init_script(Shell::Bash), script_bash);
    }

    #[test]
    fn test_get_init_script_fish() {
        let script_fish = get_init_script(Shell::Fish);
        assert!(script_fish.contains("function cdx"));
        assert!(script_fish.contains("command cdx $argv"));
        assert!(script_fish.contains("builtin cd \"$dest\""));
        assert!(script_fish.contains("\"$argv[1]\" = \"init\""));
        assert_eq!(get_init_script(Shell::Fish), script_fish);
    }
}