crsn/crsn/src/asm/mod.rs

use std::cell::RefCell;
use std::sync::Arc;

use sexp::SourcePosition;

use crate::asm::instr::flatten::labels_to_skips;
use crate::asm::parse::{ParserContext, ParserState};
use crate::module::{CrsnExtension, CrsnUniq};
use crate::runtime::program::Program;
use crate::builtin::BuiltinOps;

pub mod data;
pub mod error;
pub mod instr;
pub mod parse;
pub mod patches;

/// Parse a program from string and assemble a low level instruction sequence from it.
pub fn assemble(source: &str, uniq : &CrsnUniq, mut parsers: Vec<Box<dyn CrsnExtension>>) -> Result<Arc<Program>, error::CrsnError> {
    parsers.insert(0, BuiltinOps::new());

    for p in &mut parsers {
        p.init(uniq);
    }

    // remove first line if it looks like a shebang
    let source = if source.starts_with("#!") {
        if let Some(nl) = source.find('\n') {
            &source[nl + 1..]
        } else {
            source
        }
    } else {
        source
    };

    let pcx = ParserContext {
        parsers: &parsers,
        state: RefCell::new(ParserState {
            reg_aliases: Default::default(),
            reg_alias_stack: vec![],
            constants: Default::default(),
        }),
    };

    let ops = parse::parse(source, &SourcePosition::default(), &pcx)?;
    let ops = labels_to_skips(ops)?;

    Ok(Program::new(ops, Arc::new(parsers))?)
}