codebase-recall 0.7.2

CLI based application for codebase dumper for LLMs and fast review/recall project
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

//! Serializers for a [`CodeGraph`].

pub mod dot;
pub mod html;
pub mod json;

use anyhow::{bail, Result};

use crate::graph::CodeGraph;

/// Output format selected on the CLI.
#[derive(Clone, Copy, PartialEq, Eq, Debug)]
pub enum Format {
    Html,
    Json,
    Dot,
}

impl Format {
    pub fn parse(token: &str) -> Result<Format> {
        Ok(match token.trim().to_ascii_lowercase().as_str() {
            "html" => Format::Html,
            "json" => Format::Json,
            "dot" | "graphviz" | "gv" => Format::Dot,
            other => bail!("unknown graph format `{other}` (expected html, json, or dot)"),
        })
    }

    pub fn extension(&self) -> &'static str {
        match self {
            Format::Html => "html",
            Format::Json => "json",
            Format::Dot => "dot",
        }
    }
}

pub fn render(graph: &CodeGraph, format: Format) -> Result<String> {
    Ok(match format {
        Format::Html => html::render(graph),
        Format::Json => json::render(graph)?,
        Format::Dot => dot::render(graph),
    })
}