forked from argotorg/fe
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathdb.rs
More file actions
100 lines (86 loc) · 3.53 KB
/
Copy pathdb.rs
File metadata and controls
100 lines (86 loc) · 3.53 KB
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
#![allow(clippy::arc_with_non_send_sync)]
use std::rc::Rc;
use fe_abi::{contract::AbiContract, event::AbiEvent, function::AbiFunction, types::AbiType};
use fe_analyzer::{
db::AnalyzerDbStorage,
namespace::items::{ContractId, ModuleId},
AnalyzerDb,
};
use fe_common::db::{SourceDb, SourceDbStorage, Upcast, UpcastMut};
use fe_mir::{
db::{MirDb, MirDbStorage},
ir::{FunctionBody, FunctionId, FunctionSignature, TypeId},
};
mod queries;
#[salsa::query_group(CodegenDbStorage)]
pub trait CodegenDb: MirDb + Upcast<dyn MirDb> + UpcastMut<dyn MirDb> {
#[salsa::invoke(queries::function::legalized_signature)]
fn codegen_legalized_signature(&self, function_id: FunctionId) -> Rc<FunctionSignature>;
#[salsa::invoke(queries::function::legalized_body)]
fn codegen_legalized_body(&self, function_id: FunctionId) -> Rc<FunctionBody>;
#[salsa::invoke(queries::function::symbol_name)]
fn codegen_function_symbol_name(&self, function_id: FunctionId) -> Rc<String>;
#[salsa::invoke(queries::types::legalized_type)]
fn codegen_legalized_type(&self, ty: TypeId) -> TypeId;
#[salsa::invoke(queries::abi::abi_type)]
fn codegen_abi_type(&self, ty: TypeId) -> AbiType;
#[salsa::invoke(queries::abi::abi_function)]
fn codegen_abi_function(&self, function_id: FunctionId) -> AbiFunction;
#[salsa::invoke(queries::abi::abi_event)]
fn codegen_abi_event(&self, ty: TypeId) -> AbiEvent;
#[salsa::invoke(queries::abi::abi_contract)]
fn codegen_abi_contract(&self, contract: ContractId) -> AbiContract;
#[salsa::invoke(queries::abi::abi_module_events)]
fn codegen_abi_module_events(&self, module: ModuleId) -> Vec<AbiEvent>;
#[salsa::invoke(queries::abi::abi_type_maximum_size)]
fn codegen_abi_type_maximum_size(&self, ty: TypeId) -> usize;
#[salsa::invoke(queries::abi::abi_type_minimum_size)]
fn codegen_abi_type_minimum_size(&self, ty: TypeId) -> usize;
#[salsa::invoke(queries::abi::abi_function_argument_maximum_size)]
fn codegen_abi_function_argument_maximum_size(&self, contract: FunctionId) -> usize;
#[salsa::invoke(queries::abi::abi_function_return_maximum_size)]
fn codegen_abi_function_return_maximum_size(&self, function: FunctionId) -> usize;
#[salsa::invoke(queries::contract::symbol_name)]
fn codegen_contract_symbol_name(&self, contract: ContractId) -> Rc<String>;
#[salsa::invoke(queries::contract::deployer_symbol_name)]
fn codegen_contract_deployer_symbol_name(&self, contract: ContractId) -> Rc<String>;
#[salsa::invoke(queries::constant::string_symbol_name)]
fn codegen_constant_string_symbol_name(&self, data: String) -> Rc<String>;
}
// TODO: Move this to driver.
#[salsa::database(SourceDbStorage, AnalyzerDbStorage, MirDbStorage, CodegenDbStorage)]
#[derive(Default)]
pub struct Db {
storage: salsa::Storage<Db>,
}
impl salsa::Database for Db {}
impl Upcast<dyn MirDb> for Db {
fn upcast(&self) -> &(dyn MirDb + 'static) {
self
}
}
impl UpcastMut<dyn MirDb> for Db {
fn upcast_mut(&mut self) -> &mut (dyn MirDb + 'static) {
&mut *self
}
}
impl Upcast<dyn SourceDb> for Db {
fn upcast(&self) -> &(dyn SourceDb + 'static) {
self
}
}
impl UpcastMut<dyn SourceDb> for Db {
fn upcast_mut(&mut self) -> &mut (dyn SourceDb + 'static) {
&mut *self
}
}
impl Upcast<dyn AnalyzerDb> for Db {
fn upcast(&self) -> &(dyn AnalyzerDb + 'static) {
self
}
}
impl UpcastMut<dyn AnalyzerDb> for Db {
fn upcast_mut(&mut self) -> &mut (dyn AnalyzerDb + 'static) {
&mut *self
}
}