luau0_src/
lib.rs

1use std::env;
2use std::fs;
3use std::path::{Path, PathBuf};
4
5pub struct Build {
6    out_dir: Option<PathBuf>,
7    target: Option<String>,
8    host: Option<String>,
9    // Max number of Lua stack slots that a C function can use
10    max_cstack_size: usize,
11    // Use longjmp instead of C++ exceptions
12    use_longjmp: bool,
13    // Enable code generator (jit)
14    enable_codegen: bool,
15    // Vector size, must be 3 (default) or 4
16    vector_size: usize,
17}
18
19pub struct Artifacts {
20    lib_dir: PathBuf,
21    libs: Vec<String>,
22    cpp_stdlib: Option<String>,
23}
24
25impl Default for Build {
26    fn default() -> Self {
27        Build {
28            out_dir: env::var_os("OUT_DIR").map(PathBuf::from),
29            target: env::var("TARGET").ok(),
30            host: env::var("HOST").ok(),
31            max_cstack_size: 1000000,
32            use_longjmp: false,
33            enable_codegen: false,
34            vector_size: 3,
35        }
36    }
37}
38
39impl Build {
40    pub fn new() -> Build {
41        Build::default()
42    }
43
44    pub fn out_dir<P: AsRef<Path>>(&mut self, path: P) -> &mut Build {
45        self.out_dir = Some(path.as_ref().to_path_buf());
46        self
47    }
48
49    #[doc(hidden)]
50    pub fn target(&mut self, target: &str) -> &mut Build {
51        self.target = Some(target.to_string());
52        self
53    }
54
55    #[doc(hidden)]
56    pub fn host(&mut self, host: &str) -> &mut Build {
57        self.host = Some(host.to_string());
58        self
59    }
60
61    pub fn set_max_cstack_size(&mut self, size: usize) -> &mut Build {
62        self.max_cstack_size = size;
63        self
64    }
65
66    pub fn use_longjmp(&mut self, r#use: bool) -> &mut Build {
67        self.use_longjmp = r#use;
68        self
69    }
70
71    pub fn enable_codegen(&mut self, enable: bool) -> &mut Build {
72        self.enable_codegen = enable;
73        self
74    }
75
76    pub fn set_vector_size(&mut self, size: usize) -> &mut Build {
77        assert!(size == 3 || size == 4, "vector size must be 3 or 4");
78        self.vector_size = size;
79        self
80    }
81
82    pub fn build(&mut self) -> Artifacts {
83        let target = &self.target.as_ref().expect("TARGET is not set")[..];
84        let host = &self.host.as_ref().expect("HOST is not set")[..];
85        let out_dir = self.out_dir.as_ref().expect("OUT_DIR is not set");
86        let build_dir = out_dir.join("luau-build");
87
88        let source_base_dir = Path::new(env!("CARGO_MANIFEST_DIR"));
89        let common_include_dir = source_base_dir.join("luau").join("Common").join("include");
90        let vm_source_dir = source_base_dir.join("luau").join("VM").join("src");
91        let vm_include_dir = source_base_dir.join("luau").join("VM").join("include");
92
93        // Cleanup
94        if build_dir.exists() {
95            fs::remove_dir_all(&build_dir).unwrap();
96        }
97
98        // Configure C++
99        let mut config = cc::Build::new();
100        config
101            .warnings(false)
102            .cargo_metadata(false)
103            .std("c++17")
104            .cpp(true);
105
106        if target.ends_with("emscripten") {
107            // Enable c++ exceptions for emscripten (it's disabled by default)
108            // Later we should switch to wasm exceptions
109            config.flag_if_supported("-fexceptions");
110        }
111
112        // Common defines
113        config.define("LUAI_MAXCSTACK", &*self.max_cstack_size.to_string());
114        config.define("LUA_VECTOR_SIZE", &*self.vector_size.to_string());
115        config.define("LUA_API", "extern \"C\"");
116
117        if self.use_longjmp {
118            config.define("LUA_USE_LONGJMP", "1");
119        }
120
121        if cfg!(debug_assertions) {
122            config.define("LUAU_ENABLE_ASSERT", None);
123        } else {
124            // this flag allows compiler to lower sqrt() into a single CPU instruction
125            config.flag_if_supported("-fno-math-errno");
126        }
127
128        config.include(&common_include_dir);
129
130        // Build `Ast` library
131        let ast_lib_name = "luauast";
132        let ast_source_dir = source_base_dir.join("luau").join("Ast").join("src");
133        let ast_include_dir = source_base_dir.join("luau").join("Ast").join("include");
134        config
135            .clone()
136            .include(&ast_include_dir)
137            .add_files_by_ext(&ast_source_dir, "cpp")
138            .out_dir(&build_dir)
139            .compile(ast_lib_name);
140
141        // Build `CogeGen` library
142        let codegen_lib_name = "luaucodegen";
143        let codegen_source_dir = source_base_dir.join("luau").join("CodeGen").join("src");
144        let codegen_include_dir = source_base_dir.join("luau").join("CodeGen").join("include");
145        if self.enable_codegen {
146            if target.ends_with("emscripten") {
147                panic!("codegen (jit) is not supported on emscripten");
148            }
149
150            config
151                .clone()
152                .include(&codegen_include_dir)
153                .include(&vm_include_dir)
154                .include(&vm_source_dir)
155                .define("LUACODEGEN_API", "extern \"C\"")
156                .add_files_by_ext(&codegen_source_dir, "cpp")
157                .out_dir(&build_dir)
158                .compile(codegen_lib_name);
159        }
160
161        // Build `Compiler` library
162        let compiler_lib_name = "luaucompiler";
163        let compiler_source_dir = source_base_dir.join("luau").join("Compiler").join("src");
164        let compiler_include_dir = (source_base_dir.join("luau").join("Compiler")).join("include");
165        config
166            .clone()
167            .include(&compiler_include_dir)
168            .include(&ast_include_dir)
169            .define("LUACODE_API", "extern \"C\"")
170            .add_files_by_ext(&compiler_source_dir, "cpp")
171            .out_dir(&build_dir)
172            .compile(compiler_lib_name);
173
174        // Build customization library
175        let custom_lib_name = "luaucustom";
176        let custom_source_dir = source_base_dir.join("luau").join("Custom").join("src");
177        config
178            .clone()
179            .include(&vm_include_dir)
180            .include(&vm_source_dir)
181            .add_files_by_ext(&custom_source_dir, "cpp")
182            .out_dir(&build_dir)
183            .compile(custom_lib_name);
184
185        // Build `Require` library
186        let require_lib_name = "luaurequire";
187        let require_base_dir = source_base_dir.join("luau").join("Require");
188        let require_source_dirs = &[
189            require_base_dir.join("Navigator").join("src"),
190            require_base_dir.join("Runtime").join("src"),
191            source_base_dir.join("luau").join("Config").join("src"),
192        ];
193        let require_include_dirs = &[
194            require_base_dir.join("Navigator").join("include"),
195            require_base_dir.join("Runtime").join("include"),
196            source_base_dir.join("luau").join("Config").join("include"),
197        ];
198        let mut require_config = config.clone();
199        for (source_dir, include_dir) in require_source_dirs.iter().zip(require_include_dirs) {
200            require_config
201                .include(include_dir)
202                .add_files_by_ext(source_dir, "cpp");
203        }
204        require_config
205            .include(&ast_include_dir)
206            .include(&vm_include_dir)
207            .out_dir(&build_dir)
208            .compile(require_lib_name);
209
210        // Build VM
211        let vm_lib_name = "luauvm";
212        config
213            .clone()
214            .include(&vm_include_dir)
215            .add_files_by_ext(&vm_source_dir, "cpp")
216            .out_dir(&build_dir)
217            .compile(vm_lib_name);
218
219        let mut artifacts = Artifacts {
220            lib_dir: build_dir,
221            libs: vec![
222                vm_lib_name.to_string(),
223                compiler_lib_name.to_string(),
224                ast_lib_name.to_string(),
225                custom_lib_name.to_string(),
226                require_lib_name.to_string(),
227            ],
228            cpp_stdlib: Self::get_cpp_link_stdlib(target, host),
229        };
230
231        if self.enable_codegen {
232            artifacts.libs.push(codegen_lib_name.to_string());
233        }
234
235        artifacts
236    }
237
238    /// Returns the C++ standard library:
239    /// 1) Uses `CXXSTDLIB` environment variable if set
240    /// 2) The default `c++` for OS X and BSDs
241    /// 3) `c++_shared` for Android
242    /// 4) `None` for MSVC
243    /// 5) `stdc++` for anything else.
244    ///
245    /// Inspired by the `cc` crate.
246    fn get_cpp_link_stdlib(target: &str, host: &str) -> Option<String> {
247        // Try to get value from the `CXXSTDLIB` env variable
248        let kind = if host == target { "HOST" } else { "TARGET" };
249        let res = env::var(format!("CXXSTDLIB_{target}"))
250            .or_else(|_| env::var(format!("CXXSTDLIB_{}", target.replace('-', "_"))))
251            .or_else(|_| env::var(format!("{kind}_CXXSTDLIB")))
252            .or_else(|_| env::var("CXXSTDLIB"))
253            .ok();
254        if res.is_some() {
255            return res;
256        }
257
258        if target.contains("msvc") {
259            None
260        } else if target.contains("apple") | target.contains("freebsd") | target.contains("openbsd")
261        {
262            Some("c++".to_string())
263        } else if target.contains("android") {
264            Some("c++_shared".to_string())
265        } else {
266            Some("stdc++".to_string())
267        }
268    }
269}
270
271impl Artifacts {
272    pub fn lib_dir(&self) -> &Path {
273        &self.lib_dir
274    }
275
276    pub fn libs(&self) -> &[String] {
277        &self.libs
278    }
279
280    pub fn print_cargo_metadata(&self) {
281        println!("cargo:rustc-link-search=native={}", self.lib_dir.display());
282        for lib in &self.libs {
283            println!("cargo:rustc-link-lib=static={lib}");
284        }
285        if let Some(ref cpp_stdlib) = self.cpp_stdlib {
286            println!("cargo:rustc-link-lib={cpp_stdlib}");
287        }
288        if let Some(version) = self.version() {
289            println!("cargo:rustc-env=LUAU_VERSION={version}");
290        }
291    }
292
293    pub fn version(&self) -> Option<String> {
294        let pkg_version = env!("CARGO_PKG_VERSION");
295        let (_, luau_version) = pkg_version.split_once("+luau")?;
296        Some(format!("0.{luau_version}"))
297    }
298}
299
300trait AddFilesByExt {
301    fn add_files_by_ext(&mut self, dir: &Path, ext: &str) -> &mut Self;
302}
303
304impl AddFilesByExt for cc::Build {
305    fn add_files_by_ext(&mut self, dir: &Path, ext: &str) -> &mut Self {
306        for entry in fs::read_dir(dir)
307            .unwrap()
308            .filter_map(|e| e.ok())
309            .filter(|e| e.path().extension() == Some(ext.as_ref()))
310        {
311            self.file(entry.path());
312        }
313        self
314    }
315}
luau0_src/lib.rs

luau0_src/
lib.rs
