xqcli 0.3.2

Unified CLI for the XQuad Toolchain
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

// Copyright (C) 2026 Postquant Labs Incorporated
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program.  If not, see <https://www.gnu.org/licenses/>.
//
// SPDX-License-Identifier: AGPL-3.0-or-later

//! `xquad asm` subcommand -- assembles XQVM source into bytecode.

use std::io::Write as _;
use std::path::PathBuf;

use clap::Parser;
use miette::{IntoDiagnostic, WrapErr};

/// Assemble an XQVM assembly source file into binary bytecode.
#[derive(Debug, Parser)]
pub(crate) struct Args {
    /// Input assembly source file.
    input: PathBuf,

    /// Output file.  Defaults to `<input>.xqb` when omitted.
    #[arg(short, long)]
    output: Option<PathBuf>,

    /// Write bytecode to stdout instead of a file.
    #[arg(long, conflicts_with = "output")]
    stdout: bool,
}

/// Execute the `asm` subcommand.
pub(crate) fn exec(args: Args) -> miette::Result<()> {
    let source = std::fs::read_to_string(&args.input)
        .into_diagnostic()
        .wrap_err_with(|| format!("failed to read '{}'", args.input.display()))?;

    let name = args.input.display().to_string();
    let lines = xqasm::parse(&source, &name)?;
    let program = xqasm::assemble(&lines, &source, &name)?;
    let encoded = program.encode();

    if args.stdout {
        std::io::stdout()
            .write_all(&encoded)
            .into_diagnostic()
            .wrap_err("failed to write bytecode to stdout")?;
    } else {
        let out_path = args.output.unwrap_or_else(|| {
            let mut p = args.input.clone();
            let _ = p.set_extension("xqb");
            p
        });
        std::fs::write(&out_path, &encoded)
            .into_diagnostic()
            .wrap_err_with(|| format!("failed to write '{}'", out_path.display()))?;
        eprintln!(
            "assembled {} instructions ({} bytes) -> {}",
            instruction_count(program.code()),
            encoded.len(),
            out_path.display(),
        );
    }

    Ok(())
}

/// Count instructions by walking the encoded buffer.
fn instruction_count(buf: &[u8]) -> usize {
    xqvm::InstructionStream::new(buf)
        .filter_map(Result::ok)
        .count()
}