forked from extern/nushell
74 lines
1.4 KiB
Rust
74 lines
1.4 KiB
Rust
use std::ops::{Index, IndexMut};
|
|
|
|
use crate::{DeclId, Signature};
|
|
|
|
use super::Statement;
|
|
|
|
#[derive(Debug, Clone)]
|
|
pub struct Block {
|
|
pub signature: Box<Signature>,
|
|
pub stmts: Vec<Statement>,
|
|
pub exports: Vec<(Vec<u8>, DeclId)>, // Assuming just defs for now
|
|
}
|
|
|
|
impl Block {
|
|
pub fn len(&self) -> usize {
|
|
self.stmts.len()
|
|
}
|
|
|
|
pub fn is_empty(&self) -> bool {
|
|
self.stmts.is_empty()
|
|
}
|
|
}
|
|
|
|
impl Index<usize> for Block {
|
|
type Output = Statement;
|
|
|
|
fn index(&self, index: usize) -> &Self::Output {
|
|
&self.stmts[index]
|
|
}
|
|
}
|
|
|
|
impl IndexMut<usize> for Block {
|
|
fn index_mut(&mut self, index: usize) -> &mut Self::Output {
|
|
&mut self.stmts[index]
|
|
}
|
|
}
|
|
|
|
impl Default for Block {
|
|
fn default() -> Self {
|
|
Self::new()
|
|
}
|
|
}
|
|
|
|
impl Block {
|
|
pub fn new() -> Self {
|
|
Self {
|
|
signature: Box::new(Signature::new("")),
|
|
stmts: vec![],
|
|
exports: vec![],
|
|
}
|
|
}
|
|
|
|
pub fn with_exports(self, exports: Vec<(Vec<u8>, DeclId)>) -> Self {
|
|
Self {
|
|
signature: self.signature,
|
|
stmts: self.stmts,
|
|
exports,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl<T> From<T> for Block
|
|
where
|
|
T: Iterator<Item = Statement>,
|
|
{
|
|
fn from(stmts: T) -> Self {
|
|
Self {
|
|
signature: Box::new(Signature::new("")),
|
|
stmts: stmts.collect(),
|
|
exports: vec![],
|
|
}
|
|
}
|
|
}
|