cambridge-asm 0.1.1

// Copyright (c) 2021 Saadi Save
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

use super::{Context, Op};

pub fn end(ctx: &mut Context, _: Op) {
    ctx.increment();
}

pub fn out(ctx: &mut Context, _: Op) {
    let x = ctx.acc;

    if x > 127 {
        panic!("{} is not valid ASCII", &x)
    }

    let x = [x as u8];

    let out = std::str::from_utf8(&x).unwrap();

    println!("{}", &out);

    ctx.increment();
}

pub fn inp(ctx: &mut Context, _: Op) {
    let mut x = String::new();

    std::io::stdin()
        .read_line(&mut x)
        .expect("Unable to read stdin");

    if x.ends_with('\n') && x.chars().count() == 2 {
        ctx.acc = x.chars().next().unwrap() as usize;
    } else {
        panic!("More than one character typed");
    }

    ctx.increment();
}

// Custom instruction for debug logging
pub fn dbg(ctx: &mut Context, op: Op) {
    let x = op
        .expect("No operand");

    let out = match x.as_str() {
        "ix" | "IX" => ctx.ix,
        "acc" | "ACC" => ctx.acc,
        _ => match x.parse() {
            Ok(s) => ctx.mem.get(&s),
            Err(_) => panic!("{} is not a register or a memory address", &x)
        }
    };

    println!("{}", &out);

    ctx.increment();
}

// Raw input - directly input integers
pub fn rin(ctx: &mut Context, _: Op) {
    let mut x = String::new();

    std::io::stdin()
        .read_line(&mut x)
        .expect("Unable to read stdin");
    
    x.pop();

    ctx.acc = x.parse().unwrap_or_else(|_| panic!("'{}' is not an integer", &x));

    ctx.increment();
}