lineprior-cli 0.3.0

CLI for building and querying lineprior prior books.
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

use anyhow::Result;
use clap::Args;
use lineprior::load_prior_book;
use std::fs::File;
use std::path::PathBuf;
use std::process::ExitCode;

#[derive(Args)]
pub struct QueryArgs {
    /// Prior book JSONL produced by `build`.
    input: PathBuf,

    /// State key to look up.
    #[arg(long)]
    state: String,

    /// Return at most this many candidates.
    #[arg(long)]
    top_k: Option<usize>,
}

pub fn run(args: QueryArgs) -> Result<ExitCode> {
    let file = match File::open(&args.input) {
        Ok(f) => f,
        Err(err) => {
            eprintln!("error: opening {}: {err}", args.input.display());
            return Ok(ExitCode::from(3));
        }
    };

    let book = match load_prior_book(file) {
        Ok(book) => book,
        Err(err) => {
            eprintln!("error: {err}");
            return Ok(super::exit_code_for_lineprior_error(&err));
        }
    };

    // An unseen state is not an error: lineprior never invents actions,
    // so an empty result here is normal, successful output.
    for candidate in book.query(&args.state, args.top_k) {
        println!("{}", serde_json::to_string(&candidate)?);
    }
    Ok(ExitCode::from(0))
}