tabby 0.0.6

A fast, lightweight windows powershell tab completion library.
Documentation 
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

use std::{fs::read_to_string, io::Write};

use crate::Completer;
use anyhow::Result;
use std::fs::OpenOptions;

pub const POWERSHELL_SETUP: &str = r#"
# <cli-name> tab completion
Register-ArgumentCompleter -Native -CommandName <cli-name> -ScriptBlock {
    param($wordToComplete, $commandAst, $cursorPosition)
        [Console]::InputEncoding = [Console]::OutputEncoding = $OutputEncoding = [System.Text.Utf8Encoding]::new()
        $Local:word = $wordToComplete.Replace('"', '""')
        $Local:ast = $commandAst.ToString().Replace('"', '""')
        <cli-name>.exe complete --word="$Local:word" --commandline "$Local:ast" --position $cursorPosition | ForEach-Object {
            [System.Management.Automation.CompletionResult]::new($_, $_, 'ParameterValue', $_)
        }
}"#;

pub fn verify_setup(completer: &Completer) -> Result<bool> {
    let mut profile_string: String = String::new();

    let user_profile = home::home_dir()
        .unwrap()
        .join("Documents/WindowsPowerShell/Microsoft.PowerShell_profile.ps1");

    let global_profile = home::home_dir()
        .unwrap()
        .join("Documents/WindowsPowerShell/profile.ps1");

    profile_string.push_str(&read_to_string(user_profile)?);

    profile_string.push_str(&read_to_string(global_profile)?);

    if profile_string.contains(
        format!(
            "Register-ArgumentCompleter -Native -CommandName {} -ScriptBlock",
            &completer.completion_data.cli_name
        )
        .as_str(),
    ) {
        return Ok(true);
    } else {
        return Ok(false);
    }
}

pub fn install(completer: &Completer) -> Result<()> {
    let global_profile = home::home_dir()
        .unwrap()
        .join("Documents/WindowsPowerShell/profile.ps1");

    let setup_completer =
        POWERSHELL_SETUP.replace("<cli-name>", &completer.completion_data.cli_name);

    let mut profile = OpenOptions::new()
        .write(true)
        .append(true)
        .open(global_profile)?;

    profile.write(setup_completer.as_bytes())?;

    Ok(())
}