forked from M-Labs/nac3
1
0
Fork 0
nac3/nac3core/src/location.rs

29 lines
572 B
Rust
Raw Normal View History

2021-06-28 14:48:04 +08:00
use rustpython_parser::ast;
use std::vec::Vec;
#[derive(Clone, Copy, PartialEq)]
pub struct FileID(u32);
#[derive(Clone, Copy, PartialEq)]
pub enum Location {
CodeRange(FileID, ast::Location),
2021-08-05 14:55:09 +08:00
Builtin,
2021-06-28 14:48:04 +08:00
}
2021-09-22 17:56:48 +08:00
#[derive(Default)]
2021-06-28 14:48:04 +08:00
pub struct FileRegistry {
files: Vec<String>,
}
impl FileRegistry {
pub fn add_file(&mut self, path: &str) -> FileID {
let index = self.files.len() as u32;
self.files.push(path.to_owned());
FileID(index)
}
pub fn query_file(&self, id: FileID) -> &str {
&self.files[id.0 as usize]
}
}