lux_lib/lua_version/
mod.rs

1use std::{
2    fmt::Display,
3    path::{Path, PathBuf},
4    str::FromStr,
5};
6
7use crate::{
8    config::Config,
9    package::{PackageVersion, PackageVersionReq},
10};
11use itertools::Itertools;
12use miette::Diagnostic;
13use serde::{Deserialize, Serialize};
14use strum::IntoEnumIterator;
15use strum_macros::EnumIter;
16use thiserror::Error;
17
18#[derive(Debug, Clone, PartialEq, Eq, Deserialize, Serialize, EnumIter)]
19pub enum LuaVersion {
20    #[serde(rename = "5.1")]
21    Lua51,
22    #[serde(rename = "5.2")]
23    Lua52,
24    #[serde(rename = "5.3")]
25    Lua53,
26    #[serde(rename = "5.4")]
27    Lua54,
28    #[serde(rename = "5.5")]
29    Lua55,
30    #[serde(rename = "jit")]
31    LuaJIT,
32    #[serde(rename = "jit5.2")]
33    LuaJIT52,
34    // TODO(vhyrro): Support luau?
35    // LuaU,
36}
37
38#[derive(Debug, Error, Diagnostic)]
39pub enum LuaVersionError {
40    #[error("unsupported Lua version: {0}")]
41    UnsupportedLuaVersion(PackageVersion),
42}
43
44impl LuaVersion {
45    pub fn as_version(&self) -> PackageVersion {
46        unsafe {
47            match self {
48                LuaVersion::Lua51 => "5.1.0".parse().unwrap_unchecked(),
49                LuaVersion::Lua52 => "5.2.0".parse().unwrap_unchecked(),
50                LuaVersion::Lua53 => "5.3.0".parse().unwrap_unchecked(),
51                LuaVersion::Lua54 => "5.4.0".parse().unwrap_unchecked(),
52                LuaVersion::Lua55 => "5.5.0".parse().unwrap_unchecked(),
53                LuaVersion::LuaJIT => "5.1.0".parse().unwrap_unchecked(),
54                LuaVersion::LuaJIT52 => "5.2.0".parse().unwrap_unchecked(),
55            }
56        }
57    }
58    pub fn version_compatibility_str(&self) -> String {
59        match self {
60            LuaVersion::Lua51 | LuaVersion::LuaJIT => "5.1".into(),
61            LuaVersion::Lua52 | LuaVersion::LuaJIT52 => "5.2".into(),
62            LuaVersion::Lua53 => "5.3".into(),
63            LuaVersion::Lua54 => "5.4".into(),
64            LuaVersion::Lua55 => "5.5".into(),
65        }
66    }
67    pub fn as_version_req(&self) -> PackageVersionReq {
68        unsafe {
69            format!("~> {}", self.version_compatibility_str())
70                .parse()
71                .unwrap_unchecked()
72        }
73    }
74
75    /// Get the LuaVersion from a version that has been parsed from the `lua -v` output
76    pub fn from_version(version: PackageVersion) -> Result<LuaVersion, LuaVersionError> {
77        // NOTE: Special case. luajit -v outputs 2.x.y as a version
78        let luajit_version_req: PackageVersionReq = unsafe { "~> 2".parse().unwrap_unchecked() };
79        if luajit_version_req.matches(&version) {
80            Ok(LuaVersion::LuaJIT)
81        } else if LuaVersion::Lua51.as_version_req().matches(&version) {
82            Ok(LuaVersion::Lua51)
83        } else if LuaVersion::Lua52.as_version_req().matches(&version) {
84            Ok(LuaVersion::Lua52)
85        } else if LuaVersion::Lua53.as_version_req().matches(&version) {
86            Ok(LuaVersion::Lua53)
87        } else if LuaVersion::Lua54.as_version_req().matches(&version) {
88            Ok(LuaVersion::Lua54)
89        } else if LuaVersion::Lua55.as_version_req().matches(&version) {
90            Ok(LuaVersion::Lua55)
91        } else {
92            Err(LuaVersionError::UnsupportedLuaVersion(version))
93        }
94    }
95
96    pub(crate) fn is_luajit(&self) -> bool {
97        matches!(self, Self::LuaJIT | Self::LuaJIT52)
98    }
99
100    /// Searches for the path to the lux-lua library for this version
101    pub fn lux_lib_dir(&self) -> Option<PathBuf> {
102        option_env!("LUX_LIB_DIR")
103            .map(PathBuf::from)
104            .map(|path| path.join(self.to_string()))
105            .or_else(|| {
106                let lib_name = format!("lux-lua{self}");
107                pkg_config::Config::new()
108                    .print_system_libs(false)
109                    .cargo_metadata(false)
110                    .env_metadata(false)
111                    .probe(&lib_name)
112                    .ok()
113                    .and_then(|library| library.link_paths.first().cloned())
114            })
115            .or_else(|| lux_lib_resource_dir().map(|path| path.join(self.to_string())))
116    }
117}
118
119#[derive(Error, Debug, Diagnostic)]
120#[error("lua version not set")]
121pub struct LuaVersionUnset {
122    #[help]
123    help: String,
124}
125
126impl LuaVersion {
127    pub fn from(config: &Config) -> Result<&Self, LuaVersionUnset> {
128        config.lua_version().ok_or(LuaVersionUnset {
129            help: format!(
130                r#"provide a version through `lx --lua-version <ver> <cmd>`
131or make sure Lux can discover your Lua installation on your PATH.
132valid versions are: {}
133"#,
134                LuaVersion::iter().join(",")
135            ),
136        })
137    }
138}
139
140impl FromStr for LuaVersion {
141    type Err = String;
142
143    fn from_str(s: &str) -> std::result::Result<Self, Self::Err> {
144        match s {
145            "5.1" | "51" => Ok(LuaVersion::Lua51),
146            "5.2" | "52" => Ok(LuaVersion::Lua52),
147            "5.3" | "53" => Ok(LuaVersion::Lua53),
148            "5.4" | "54" => Ok(LuaVersion::Lua54),
149            "5.5" | "55" => Ok(LuaVersion::Lua55),
150            "jit" | "luajit" => Ok(LuaVersion::LuaJIT),
151            "jit52" | "luajit52" => Ok(LuaVersion::LuaJIT52),
152            _ => Err(format!(
153                r#"unrecognized Lua version.
154supported versions: {}
155            "#,
156                LuaVersion::iter().join(",")
157            )),
158        }
159    }
160}
161
162impl Display for LuaVersion {
163    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
164        f.write_str(match self {
165            LuaVersion::Lua51 => "5.1",
166            LuaVersion::Lua52 => "5.2",
167            LuaVersion::Lua53 => "5.3",
168            LuaVersion::Lua54 => "5.4",
169            LuaVersion::Lua55 => "5.5",
170            LuaVersion::LuaJIT => "jit",
171            LuaVersion::LuaJIT52 => "jit52",
172        })
173    }
174}
175
176/// Searches for the lux-lib directory in a binary distribution's resources
177fn lux_lib_resource_dir() -> Option<PathBuf> {
178    if cfg!(target_env = "msvc") {
179        // The msvc .exe and .msi binary installers install lux-lua to the executable's directory.
180        std::env::current_exe()
181            .ok()
182            .and_then(|exe_path| exe_path.parent().map(Path::to_path_buf))
183            .and_then(|exe_dir| {
184                let lib_dir = exe_dir.join("lux-lua");
185                if lib_dir.is_dir() {
186                    Some(lib_dir)
187                } else {
188                    None
189                }
190            })
191    } else if cfg!(target_os = "macos") {
192        // Currently, we only bundle resources with an .app ApplicationBundle
193        std::env::current_exe()
194            .ok()
195            .and_then(|exe_path| exe_path.parent().map(Path::to_path_buf))
196            .and_then(|macos_dir| macos_dir.parent().map(Path::to_path_buf))
197            .and_then(|contents_dir| {
198                let lib_dir = contents_dir.join("Resources").join("lux-lua");
199                if lib_dir.is_dir() {
200                    Some(lib_dir)
201                } else {
202                    None
203                }
204            })
205    } else {
206        // .deb and AppImage packages
207        let lib_dir = PathBuf::from("/usr/share/lux-lua");
208        if lib_dir.is_dir() {
209            Some(lib_dir)
210        } else {
211            None
212        }
213    }
214}
lux_lib/lua_version/mod.rs

lux_lib/lua_version/
mod.rs
