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
#![forbid(unsafe_code)]
use move_deps::{
move_binary_format::file_format::CompiledModule,
move_compiler::{
compiled_unit::{CompiledUnit, NamedCompiledModule},
shared::NumericalAddress,
},
move_package::compilation::compiled_package::CompiledPackage,
};
use once_cell::sync::Lazy;
use std::collections::{BTreeMap, HashSet};
use std::hash::Hash;
const APTOS_FRAMEWORK_DIR: &str = "aptos-framework/sources";
const APTOS_STDLIB_DIR: &str = "aptos-stdlib/sources";
const MOVE_STDLIB_DIR: &str = "move-stdlib/sources";
const TOKEN_MODULES_DIR: &str = "aptos-token/sources";
static APTOS_PKG: Lazy<CompiledPackage> = Lazy::new(|| super::package("aptos-framework"));
static TOKEN_PKG: Lazy<CompiledPackage> = Lazy::new(|| super::package("aptos-token"));
static APTOS_STDLIB_PKG: Lazy<CompiledPackage> = Lazy::new(|| super::package("aptos-stdlib"));
pub fn dedup<T, K: Hash + Eq, F: Fn(&T) -> K>(lists: Vec<T>, f: F) -> Vec<T> {
let mut res = vec![];
let mut keys = HashSet::new();
for l in lists {
let key: K = f(&l);
if keys.insert(key) {
res.push(l);
}
}
res
}
pub fn files() -> Vec<String> {
let mut files = super::move_files_in_path(MOVE_STDLIB_DIR);
files.extend(super::move_files_in_path(APTOS_STDLIB_DIR));
files.extend(super::move_files_in_path(APTOS_FRAMEWORK_DIR));
files.extend(super::move_files_in_path(TOKEN_MODULES_DIR));
files.extend(super::move_files_in_path(APTOS_STDLIB_DIR));
dedup(files, |f| f.to_string())
}
pub fn module_blobs() -> Vec<Vec<u8>> {
let mut framework_blobs = super::module_blobs(&*APTOS_PKG);
framework_blobs.extend(super::module_blobs(&*TOKEN_PKG));
framework_blobs.extend(super::module_blobs(&*APTOS_STDLIB_PKG));
dedup(framework_blobs, |blob| {
CompiledModule::deserialize(blob).unwrap().self_id()
})
}
pub fn named_addresses() -> BTreeMap<String, NumericalAddress> {
let mut framework = super::named_addresses(&*APTOS_PKG);
framework.append(&mut super::named_addresses(&*TOKEN_PKG));
framework.append(&mut super::named_addresses(&*APTOS_STDLIB_PKG));
framework
}
pub fn modules() -> Vec<CompiledModule> {
let mut framework: Vec<CompiledModule> = APTOS_PKG
.all_compiled_units()
.filter_map(|unit| match unit {
CompiledUnit::Module(NamedCompiledModule { module, .. }) => Some(module.clone()),
CompiledUnit::Script(_) => None,
})
.collect();
framework.extend(
TOKEN_PKG
.all_compiled_units()
.filter_map(|unit| match unit {
CompiledUnit::Module(NamedCompiledModule { module, .. }) => Some(module.clone()),
CompiledUnit::Script(_) => None,
})
.collect::<Vec<CompiledModule>>(),
);
framework.extend(
APTOS_STDLIB_PKG
.all_compiled_units()
.filter_map(|unit| match unit {
CompiledUnit::Module(NamedCompiledModule { module, .. }) => Some(module.clone()),
CompiledUnit::Script(_) => None,
})
.collect::<Vec<CompiledModule>>(),
);
dedup(framework, |f| f.self_id())
}
