#![feature(set_stdio)]

extern crate argdata;
extern crate cloudabi;

use argdata::Argdata;
use argdata::ArgdataExt;

pub struct Console(cloudabi::fd);

impl std::io::Write for Console {
    fn write(&mut self, buf: &[u8]) -> std::io::Result<usize> {
        unsafe {
            let iovs = [
                cloudabi::ciovec {
                    buf: (buf.as_ptr() as *const _ as *const (), buf.len()),
                },
            ];
            let mut nwritten = std::mem::uninitialized();
            cloudabi::fd_write(self.0, &iovs, &mut nwritten);
        }
        Ok(buf.len())
    }

    fn flush(&mut self) -> std::io::Result<()> {
        Ok(())
    }
}

fn main() {
    let ad = argdata::env::argdata();
    let mut it = ad.read_map().expect("argdata should be a map").iter_map();
    while let Some(Ok((key, val))) = it.next() {
        match key.read_str().expect("keys should be strings") {
            "console" => {
                if let Some(fd) = val.read_fd().ok() {
                    std::io::set_print(Some(Box::new(Console(cloudabi::fd(fd.0 as u32)))));
                    std::io::set_panic(Some(Box::new(Console(cloudabi::fd(fd.0 as u32)))));
                }
            }
            _ => {}
        }
    }

    println!("Hello, world!");
}