trion 0.7.2

Trion is an assembler designed to be used with the Raspberry Pico (RP2040) microcontroller.
Documentation 
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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114

use core::fmt;
use std::error::Error;
use std::fs::OpenOptions;
use std::io::{self, Read};
use std::path::PathBuf;

use crate::asm::{Context, ErrorLevel};
use crate::asm::directive::{Directive, DirectiveErrorKind};
use crate::text::Positioned;
use crate::text::parse::{Argument, ArgumentType};
use crate::text::parse::choice::ArgChoice;

#[derive(Clone, Copy, Debug)]
pub struct Include;

impl Directive for Include
{
	fn get_name(&self) -> &str
	{
		"include"
	}
	
	fn apply(&self, ctx: & mut Context, args: Positioned<Vec<Argument>>) -> Result<(), ErrorLevel>
	{
		const NUM_ARGS: usize = 1;
		if args.value.len() != NUM_ARGS
		{
			let dir = self.get_name().to_owned();
			ctx.push_error(args.convert(
				if args.value.len() < NUM_ARGS {DirectiveErrorKind::NotEnoughArguments{dir, need: NUM_ARGS, have: args.value.len()}}
				else {DirectiveErrorKind::TooManyArguments{dir, max: NUM_ARGS, have: args.value.len()}}
			));
			return Err(ErrorLevel::Trivial);
		}
		let path = match args.value[0]
		{
			Argument::String(ref path) =>
			{
				let mut curr = match ctx.curr_file_path()
				{
					None => PathBuf::new(),
					Some(prev) => prev.to_path_buf(),
				};
				if !curr.pop() {curr.push("..");}
				curr.push(path.as_ref());
				curr
			},
			ref arg =>
			{
				let dir = self.get_name().to_owned();
				ctx.push_error(args.convert(DirectiveErrorKind::ArgumentType{dir, idx: 0, expect: ArgChoice::of(ArgumentType::String), have: arg.get_type()}));
				return Err(ErrorLevel::Trivial);
			},
		};
		let mut f = match OpenOptions::new().read(true).open(&path)
		{
			Ok(f) => f,
			Err(err) =>
			{
				let source = Box::new(IncludeError::NoSuchFile{path, err});
				ctx.push_error(args.convert(DirectiveErrorKind::Apply{dir: self.get_name().to_owned(), source}));
				return Err(ErrorLevel::Fatal);
			},
		};
		let mut data = Vec::new();
		if let Err(err) = f.read_to_end(&mut data)
		{
			let source = Box::new(IncludeError::FileRead{path, err});
			ctx.push_error(args.convert(DirectiveErrorKind::Apply{dir: self.get_name().to_owned(), source}));
			return Err(ErrorLevel::Fatal);
		}
		if let (Err(..), path) = ctx.assemble(data.as_ref(), path)
		{
			let source = Box::new(IncludeError::AssemblyFailed{path});
			ctx.push_error(args.convert(DirectiveErrorKind::Apply{dir: self.get_name().to_owned(), source}));
			return Err(ErrorLevel::Fatal);
		}
		Ok(())
	}
}

#[derive(Debug)]
pub enum IncludeError
{
	NoSuchFile{path: PathBuf, err: io::Error},
	FileRead{path: PathBuf, err: io::Error},
	AssemblyFailed{path: PathBuf}, // TODO should reference the original error
}

impl fmt::Display for IncludeError
{
	fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
	{
		match self
		{
			Self::NoSuchFile{path, ..} => write!(f, "no such file {}", path.display()),
			Self::FileRead{path, ..} => write!(f, "could not read file {}", path.display()),
			Self::AssemblyFailed{path} => write!(f, "assembly of {} failed", path.display()),
		}
	}
}

impl Error for IncludeError
{
	fn source(&self) -> Option<&(dyn Error + 'static)>
	{
		match self
		{
			Self::NoSuchFile{err, ..} => Some(err),
			Self::FileRead{err, ..} => Some(err),
			_ => None,
		}
	}
}