flk 0.6.2

A CLI tool for managing flake.nix devShell environments
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
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

//! # Direnv Integration Handler
//!
//! Manage direnv integration for automatic environment activation.

use anyhow::{bail, Context, Result};
use colored::Colorize;
use std::fs;
use std::path::Path;

const DIRENV_FLK_DIRECTIVE: &str = r#"# Watch flk config files so nix-direnv re-evaluates on changes
watch_file .flk/default.nix
watch_file .flk/pins.nix
watch_file .flk/overlays.nix
for f in .flk/profiles/*.nix; do watch_file "$f"; done

use flake "${FLK_PROFILE:-.#}" --impure"#;

/// Create a new `.envrc` file with the flk directive.
pub fn direnv_init() -> Result<()> {
    let direnv_path = Path::new(".envrc");

    // Check if .envrc already exists
    if direnv_path.exists() {
        bail!(".envrc already exists! Please back it up before proceeding.");
    }

    fs::write(direnv_path, DIRENV_FLK_DIRECTIVE).context("Failed to write .envrc")?;

    println!(
        "{} Created .envrc for direnv successfully!",
        "✓".green().bold()
    );
    println!("\n{}", "Next steps:".bold());
    println!("  1. Review the .envrc file");
    println!("  2. Run {} to allow direnv", "direnv allow".cyan());
    println!("  3. Enjoy your development environment with direnv and flk!");

    Ok(())
}

/// Append the flk directive to an existing `.envrc` file.
pub fn direnv_attach() -> Result<()> {
    let direnv_path = Path::new(".envrc");

    // Check if .envrc exists
    if !direnv_path.exists() {
        bail!(".envrc does not exist! Please run 'flk direnv-init' instead.");
    }

    let mut direnv_content = fs::read_to_string(direnv_path).context("Failed to read .envrc")?;

    // Check if use flake directive already exists
    if direnv_content.contains("use flake") {
        bail!("The .envrc already contains the use flake directive!");
    }

    direnv_content.push('\n');
    direnv_content.push_str(DIRENV_FLK_DIRECTIVE);
    fs::write(direnv_path, direnv_content).context("Failed to update .envrc")?;

    println!(
        "{} Updated .envrc for direnv successfully!",
        "✓".green().bold()
    );
    println!("\n{}", "Next steps:".bold());
    println!("  1. Review the updated .envrc file");
    println!("  2. Run {} to reload direnv", "direnv reload".cyan());
    println!("  3. Enjoy your development environment with direnv and flk!");

    Ok(())
}

/// Remove the flk directive from an existing `.envrc` file.
pub fn direnv_detach() -> Result<()> {
    let direnv_path = Path::new(".envrc");
    // Check if .envrc exists
    if !direnv_path.exists() {
        bail!(".envrc does not exist!");
    }
    let direnv_content = fs::read_to_string(direnv_path).context("Failed to read .envrc")?;
    // Remove flk directives (use flake and watch_file for .flk/)
    let updated_content: String = direnv_content
        .lines()
        .filter(|line| {
            let trimmed = line.trim();
            trimmed != r#"use flake "${FLK_PROFILE:-.#}" --impure"#
                && !trimmed.starts_with("watch_file .flk/")
                && !trimmed.starts_with("for f in .flk/profiles")
                && trimmed != "# Watch flk config files so nix-direnv re-evaluates on changes"
        })
        .map(|line| format!("{}\n", line))
        .collect();
    fs::write(direnv_path, updated_content).context("Failed to update .envrc")?;
    println!(
        "{} Removed use flake directive from .envrc successfully!",
        "✓".green().bold()
    );
    Ok(())
}