lux_lib/lua_rockspec/

mod.rs

mod build;
mod dependency;
mod deploy;
mod partial;
mod platform;
mod rock_source;
mod serde_util;
mod test_spec;

use std::{
    collections::HashMap, convert::Infallible, fmt::Display, io, path::PathBuf, str::FromStr,
};

use ottavino::{Closure, Executor, Fuel};
use ottavino_util::serde::from_value;
use serde::{de::DeserializeOwned, Deserialize, Serialize};

pub use build::*;
pub use dependency::*;
pub use deploy::*;
use miette::{Diagnostic, NamedSource, SourceSpan};
pub use partial::*;
pub use platform::*;
pub use rock_source::*;
use ssri::Integrity;
pub use test_spec::*;
use thiserror::Error;
use url::Url;

pub(crate) use lux_macros::DisplayAsLuaKV;
pub(crate) use serde_util::*;

impl DisplayAsLuaValue for Url {
    fn display_lua_value(&self) -> DisplayLuaValue {
        DisplayLuaValue::String(self.to_string())
    }
}

use crate::{
    hash::HasIntegrity,
    lua_version::{LuaVersion, LuaVersionUnset},
    package::{PackageName, PackageSpec, PackageVersion, PackageVersionReq},
    project::{project_toml::ProjectTomlError, ProjectRoot},
    rockspec::{lua_dependency::LuaDependencySpec, Rockspec},
    ROCKSPEC_FUEL_LIMIT,
};

#[derive(Error, Debug, Diagnostic)]
#[non_exhaustive]
pub enum Error {
    #[error("manifest exceeds computational limit of {ROCKSPEC_FUEL_LIMIT} steps")]
    #[diagnostic(
        code(lux_lib::lua_rockspec::fuel_limit_exceeded),
        help(
            r#"the rockspec may be too complex or contain too many dependencies.
try reducing the number of dependencies or simplifying build instructions.
"#
        )
    )]
    FuelLimitExceeded,
    #[error("could not parse rockspec")]
    #[diagnostic(code(lux_lib::lua_rockspec::execution),
        help("check the rockspec or lux.toml for valid syntax and make sure it matches the specification."),
        url("https://lux.lumen-labs.org/reference/lux-toml")
    )]
    ExecutionError {
        cause: Box<String>,
        #[source_code]
        content: NamedSource<String>,
        #[label("this package")]
        pkg_span: SourceSpan,
        #[label("{cause}")]
        err_span: Option<SourceSpan>,
    },
    #[error("{}copy_directories cannot contain the rockspec name", ._0.as_ref().map(|p| format!("{p}: ")).unwrap_or_default())]
    CopyDirectoriesContainRockspecName(Option<String>),
    #[error("cannot create Lua rockspec with off-spec dependency")]
    #[diagnostic(
        code(lux_lib::lua_rockspec::off_spec_dependency),
        url("https://lux.lumen-labs.org/reference/lux-toml#dependencies")
    )]
    OffSpecDependency(PackageName),
    #[error("cannot create Lua rockspec with off-spec build dependency")]
    #[diagnostic(
        code(lux_lib::lua_rockspec::off_spec_build_dependency),
        url("https://lux.lumen-labs.org/reference/lux-toml#dependencies")
    )]
    OffSpecBuildDependency(PackageName),
    #[error("cannot create Lua rockspec with off-spec test dependency")]
    #[diagnostic(
        code(lux_lib::lua_rockspec::off_spec_test_dependency),
        url("https://lux.lumen-labs.org/reference/lux-toml#dependencies")
    )]
    OffSpecTestDependency(PackageName),
    #[error(transparent)]
    #[diagnostic(transparent)]
    ProjectToml(#[from] ProjectTomlError),
}

impl Error {
    fn from_ottavino(cause: ottavino::ExternError, rockspec_content: &str) -> Self {
        let msg = match &cause {
            ottavino::ExternError::Runtime(rt) => rt.to_string(),
            _ => cause.to_string(),
        };
        let field = msg
            .strip_prefix("invalid rockspec field '")
            .and_then(|rest| rest.split('\'').next())
            .map(|f| f.to_owned());
        let err_span = field.as_deref().and_then(|field| {
            let span = find_key_span(rockspec_content, field);
            if span.is_empty() {
                None
            } else {
                Some(span)
            }
        });
        Error::ExecutionError {
            cause: Box::new(msg),
            content: NamedSource::new("rockspec", rockspec_content.to_string()),
            pkg_span: find_value_span(rockspec_content, "package"),
            err_span,
        }
    }
}

/// This error can be returned within `ottavino::Lua::try_enter`.
/// ottavino wraps it in an [`ottavino::ExternError`], which Lux wraps inn
/// a [`Error::ExecutionError`]
#[derive(Error, Debug, Diagnostic)]
#[non_exhaustive]
enum ParseLuaRockspecFieldError {
    #[error("invalid rockspec field '{field}': {source}")]
    Deserialize {
        field: String,
        source: Box<ottavino_util::serde::de::Error>,
    },
    #[error(
        r#"invalid rockspec field '{field}'.
Failed to parse a table field: {source}"#
    )]
    DeserializeTable {
        field: String,
        source: Box<ottavino_util::serde::de::Error>,
    },
    #[error("invalid rockspec field '{field}'. expected a table, but got: {invalid_type}")]
    NotATable { field: String, invalid_type: String },
}

/// RockSpec for a local rock installation, deserialized from a `.rockspec` file
#[derive(Clone, Debug)]
#[cfg_attr(test, derive(PartialEq))]
pub struct LocalLuaRockspec {
    /// The file format version. Example: "1.0"
    rockspec_format: Option<RockspecFormat>,
    /// The name of the package. Example: "luasocket"
    package: PackageName,
    /// The version of the package, plus a suffix indicating the revision of the rockspec. Example: "2.0.1-1"
    version: PackageVersion,
    description: RockDescription,
    supported_platforms: PlatformSupport,
    /// The Lua version requirement for this rock
    lua: PackageVersionReq,
    dependencies: PerPlatform<Vec<LuaDependencySpec>>,
    build_dependencies: PerPlatform<Vec<LuaDependencySpec>>,
    external_dependencies: PerPlatform<HashMap<String, ExternalDependencySpec>>,
    test_dependencies: PerPlatform<Vec<LuaDependencySpec>>,
    build: PerPlatform<BuildSpec>,
    source: PerPlatform<RemoteRockSource>,
    test: PerPlatform<TestSpec>,
    deploy: PerPlatform<DeploySpec>,
    /// The original content of this rockspec, needed by luarocks
    raw_content: String,
}

trait HasRockspecKey<'gc> {
    fn get_rockspec_key<V: Deserialize<'gc>>(
        &self,
        ctx: ottavino::Context<'gc>,
        key: &str,
    ) -> Result<V, ParseLuaRockspecFieldError>;
}

impl<'gc> HasRockspecKey<'gc> for ottavino::Table<'gc> {
    fn get_rockspec_key<V: Deserialize<'gc>>(
        &self,
        ctx: ottavino::Context<'gc>,
        key: &str,
    ) -> Result<V, ParseLuaRockspecFieldError> {
        let value = self.get_value(ctx, key.to_string());
        let is_table = matches!(&value, ottavino::Value::Table(_));
        from_value(value).map_err(|source| {
            if is_table {
                ParseLuaRockspecFieldError::DeserializeTable {
                    field: key.into(),
                    source: Box::new(source),
                }
            } else {
                ParseLuaRockspecFieldError::Deserialize {
                    field: key.into(),
                    source: Box::new(source),
                }
            }
        })
    }
}

impl LocalLuaRockspec {
    /// Builds a [`LocalLuaRockspec`] from an already-parsed project.
    pub(crate) fn from_project_toml(
        project: &crate::project::project_toml::LocalProjectToml,
        raw_content: String,
    ) -> Self {
        use crate::rockspec::Rockspec;

        Self {
            rockspec_format: project.format().clone(),
            package: project.package().clone(),
            version: project.version().clone(),
            description: project.description().clone(),
            supported_platforms: project.supported_platforms().clone(),
            lua: project.lua().clone(),
            dependencies: project.dependencies().clone(),
            build_dependencies: project.build_dependencies().clone(),
            external_dependencies: project.external_dependencies().clone(),
            test_dependencies: project.test_dependencies().clone(),
            build: project.build().clone(),
            source: project.source().clone(),
            test: project.test().clone(),
            deploy: project.deploy().clone(),
            raw_content,
        }
    }

    /// Wraps this rockspec into a [`RemoteLuaRockspec`], overriding its source
    /// with the given `source`.
    pub fn into_remote_with_source(
        self,
        source: PerPlatform<RemoteRockSource>,
    ) -> RemoteLuaRockspec {
        RemoteLuaRockspec {
            local: self,
            source,
        }
    }

    pub fn new(rockspec_content: &str, project_root: ProjectRoot) -> Result<Self, Error> {
        let mut lua = ottavino::Lua::core();

        let rockspec = lua
            .try_enter(|ctx| {
                let closure = Closure::load(ctx, None, rockspec_content.as_bytes())?;

                let executor = Executor::start(ctx, closure.into(), ());

                let output = executor.step(ctx, &mut Fuel::with(ROCKSPEC_FUEL_LIMIT))?;

                if !output {
                    return Ok(Err(Error::FuelLimitExceeded));
                }

                let globals = ctx.globals();

                let dependencies: PerPlatform<Vec<LuaDependencySpec>> =
                    globals.get_rockspec_key(ctx, "dependencies")?;

                let lua_version_req = dependencies
                    .current_platform()
                    .iter()
                    .find(|dep| dep.name().to_string() == "lua")
                    .cloned()
                    .map(|dep| dep.version_req().clone())
                    .unwrap_or(PackageVersionReq::Any);

                fn strip_lua(
                    dependencies: PerPlatform<Vec<LuaDependencySpec>>,
                ) -> PerPlatform<Vec<LuaDependencySpec>> {
                    dependencies.map(|deps| {
                        deps.iter()
                            .filter(|dep| dep.name().to_string() != "lua")
                            .cloned()
                            .collect()
                    })
                }

                let build_dependencies: PerPlatform<Vec<LuaDependencySpec>> =
                    globals.get_rockspec_key(ctx, "build_dependencies")?;

                let test_dependencies: PerPlatform<Vec<LuaDependencySpec>> =
                    globals.get_rockspec_key(ctx, "test_dependencies")?;

                let source: PerPlatform<RemoteRockSource> = match globals.get_value(ctx, "source") {
                    ottavino::Value::Nil => {
                        PerPlatform::new(RockSourceSpec::File(project_root.to_path_buf()).into())
                    }
                    value => from_value(value).map_err(|source| {
                        ParseLuaRockspecFieldError::Deserialize {
                            field: "source".into(),
                            source: Box::new(source),
                        }
                    })?,
                };

                let rockspec = LocalLuaRockspec {
                    rockspec_format: globals.get_rockspec_key(ctx, "rockspec_format")?,
                    package: globals.get_rockspec_key(ctx, "package")?,
                    version: globals.get_rockspec_key(ctx, "version")?,
                    description: parse_lua_tbl_or_default(ctx, "description")?,
                    supported_platforms: parse_lua_tbl_or_default(ctx, "supported_platforms")?,
                    lua: lua_version_req,
                    dependencies: strip_lua(dependencies),
                    build_dependencies: strip_lua(build_dependencies),
                    test_dependencies: strip_lua(test_dependencies),
                    external_dependencies: globals
                        .get_rockspec_key(ctx, "external_dependencies")?,
                    build: globals.get_rockspec_key(ctx, "build")?,
                    test: globals.get_rockspec_key(ctx, "test")?,
                    deploy: globals.get_rockspec_key(ctx, "deploy")?,
                    raw_content: rockspec_content.into(),

                    source,
                };

                Ok(Ok(rockspec))
            })
            .map_err(|cause| Error::from_ottavino(cause, rockspec_content))??;

        let rockspec_file_name = format!("{}-{}.rockspec", rockspec.package(), rockspec.version());

        if rockspec
            .build()
            .default
            .copy_directories
            .contains(&PathBuf::from(&rockspec_file_name))
        {
            return Err(Error::CopyDirectoriesContainRockspecName(None));
        }

        for (platform, build_override) in &rockspec.build().per_platform {
            if build_override
                .copy_directories
                .contains(&PathBuf::from(&rockspec_file_name))
            {
                return Err(Error::CopyDirectoriesContainRockspecName(Some(
                    platform.to_string(),
                )));
            }
        }
        Ok(rockspec)
    }
}

impl Rockspec for LocalLuaRockspec {
    type Error = Infallible;

    fn package(&self) -> &PackageName {
        &self.package
    }

    fn version(&self) -> &PackageVersion {
        &self.version
    }

    fn description(&self) -> &RockDescription {
        &self.description
    }

    fn supported_platforms(&self) -> &PlatformSupport {
        &self.supported_platforms
    }

    fn lua(&self) -> &PackageVersionReq {
        &self.lua
    }

    fn dependencies(&self) -> &PerPlatform<Vec<LuaDependencySpec>> {
        &self.dependencies
    }

    fn build_dependencies(&self) -> &PerPlatform<Vec<LuaDependencySpec>> {
        &self.build_dependencies
    }

    fn external_dependencies(&self) -> &PerPlatform<HashMap<String, ExternalDependencySpec>> {
        &self.external_dependencies
    }

    fn test_dependencies(&self) -> &PerPlatform<Vec<LuaDependencySpec>> {
        &self.test_dependencies
    }

    fn build(&self) -> &PerPlatform<BuildSpec> {
        &self.build
    }

    fn test(&self) -> &PerPlatform<TestSpec> {
        &self.test
    }

    fn source(&self) -> &PerPlatform<RemoteRockSource> {
        &self.source
    }

    fn deploy(&self) -> &PerPlatform<DeploySpec> {
        &self.deploy
    }

    fn build_mut(&mut self) -> &mut PerPlatform<BuildSpec> {
        &mut self.build
    }

    fn test_mut(&mut self) -> &mut PerPlatform<TestSpec> {
        &mut self.test
    }

    fn source_mut(&mut self) -> &mut PerPlatform<RemoteRockSource> {
        &mut self.source
    }

    fn deploy_mut(&mut self) -> &mut PerPlatform<DeploySpec> {
        &mut self.deploy
    }

    fn format(&self) -> &Option<RockspecFormat> {
        &self.rockspec_format
    }

    fn to_lua_remote_rockspec_string(&self) -> Result<String, Self::Error> {
        Ok(self.raw_content.clone())
    }
}

impl HasIntegrity for LocalLuaRockspec {
    async fn hash(&self) -> io::Result<Integrity> {
        Ok(Integrity::from(&self.raw_content))
    }
}

/// RockSpec for a remote rock, deserialized from a `.rockspec` file
#[derive(Clone, Debug)]
#[cfg_attr(test, derive(PartialEq))]
pub struct RemoteLuaRockspec {
    local: LocalLuaRockspec,
    source: PerPlatform<RemoteRockSource>,
}

impl RemoteLuaRockspec {
    pub fn new(rockspec_content: &str) -> Result<Self, Error> {
        let mut lua = ottavino::Lua::core();

        let source = lua
            .try_enter(|ctx| {
                let closure = Closure::load(ctx, None, rockspec_content.as_bytes())?;

                let executor = Executor::start(ctx, closure.into(), ());

                let output = executor.step(ctx, &mut Fuel::with(ROCKSPEC_FUEL_LIMIT))?;

                if !output {
                    return Ok(Err(Error::FuelLimitExceeded));
                }

                let globals = ctx.globals();

                let source: PerPlatform<RemoteRockSource> =
                    globals.get_rockspec_key(ctx, "source")?;

                Ok(Ok(source))
            })
            .map_err(|cause| Error::from_ottavino(cause, rockspec_content))??;

        Ok(RemoteLuaRockspec {
            // NOTE: Calling this outside of `lua.try_enter` reduces error message repetition
            local: LocalLuaRockspec::new(rockspec_content, ProjectRoot::new())?,
            source,
        })
    }

    /// Builds a rockspec from a source spec and dependencies.
    ///
    /// With no modules, the rockspec uses the `source` build backend.
    /// Otherwise, it uses the `builtin` build backend with the given modules.
    pub(crate) fn from_source_spec(
        package_spec: PackageSpec,
        source_spec: RockSourceSpec,
        dependencies: Vec<LuaDependencySpec>,
        modules: HashMap<LuaModule, ModuleSpec>,
    ) -> Self {
        let version = package_spec.version().clone();
        let rockspec_format = RockspecFormat::default();
        let (build_backend, build_lua) = if modules.is_empty() {
            (BuildBackendSpec::Source, r#"type = "source""#.to_string())
        } else {
            let modules_lua = modules_lua(&modules);
            (
                BuildBackendSpec::Builtin(BuiltinBuildSpec { modules }),
                format!("type = \"builtin\",\n  modules = {{\n{modules_lua}\n  }}"),
            )
        };
        let build_spec = BuildSpec {
            build_backend: Some(build_backend),
            ..Default::default()
        };
        let raw_content = rockspec_content(&package_spec, &source_spec, &build_lua);
        let source: RemoteRockSource = source_spec.into();

        let local = LocalLuaRockspec {
            rockspec_format: Some(rockspec_format),
            package: package_spec.name().clone(),
            version,
            description: RockDescription::default(),
            supported_platforms: PlatformSupport::default(),
            lua: PackageVersionReq::Any,
            dependencies: PerPlatform::new(dependencies),
            build_dependencies: PerPlatform::default(),
            external_dependencies: PerPlatform::default(),
            test_dependencies: PerPlatform::default(),
            build: PerPlatform::new(build_spec),
            source: PerPlatform::new(source.clone()),
            test: PerPlatform::default(),
            deploy: PerPlatform::default(),
            raw_content,
        };
        Self {
            local,
            source: PerPlatform::new(source),
        }
    }
}

fn rockspec_content(
    package_spec: &PackageSpec,
    source_spec: &RockSourceSpec,
    build_lua: &str,
) -> String {
    format!(
        r#"
rockspec_format = "{}"
package = "{}"
version = "{}"
{}
build = {{
{}
}}"#,
        RockspecFormat::default(),
        package_spec.name(),
        package_spec.version(),
        source_spec.display_lua(),
        build_lua,
    )
}

fn modules_lua(modules: &HashMap<LuaModule, ModuleSpec>) -> String {
    let mut entries: Vec<_> = modules.iter().collect();
    entries.sort_by_key(|(module, _)| module.as_str());
    DisplayLuaValue::Table(
        entries
            .into_iter()
            .map(|(module, spec)| DisplayLuaKV {
                key: module.to_string(),
                value: spec.display_lua_value(),
            })
            .collect(),
    )
    .to_string()
}

impl Rockspec for RemoteLuaRockspec {
    type Error = Infallible;

    fn package(&self) -> &PackageName {
        self.local.package()
    }

    fn version(&self) -> &PackageVersion {
        self.local.version()
    }

    fn description(&self) -> &RockDescription {
        self.local.description()
    }

    fn supported_platforms(&self) -> &PlatformSupport {
        self.local.supported_platforms()
    }

    fn lua(&self) -> &PackageVersionReq {
        self.local.lua()
    }

    fn dependencies(&self) -> &PerPlatform<Vec<LuaDependencySpec>> {
        self.local.dependencies()
    }

    fn build_dependencies(&self) -> &PerPlatform<Vec<LuaDependencySpec>> {
        match self.format() {
            // Rockspec formats < 3.0 don't support `build_dependencies`,
            // so we have to return regular dependencies if the build backend might need to use them.
            Some(RockspecFormat::_1_0 | RockspecFormat::_2_0)
                if self
                    .build()
                    .current_platform()
                    .build_backend
                    .as_ref()
                    .is_some_and(|build_backend| build_backend.can_use_build_dependencies()) =>
            {
                self.local.dependencies()
            }
            _ => self.local.build_dependencies(),
        }
    }

    fn external_dependencies(&self) -> &PerPlatform<HashMap<String, ExternalDependencySpec>> {
        self.local.external_dependencies()
    }

    fn test_dependencies(&self) -> &PerPlatform<Vec<LuaDependencySpec>> {
        self.local.test_dependencies()
    }

    fn build(&self) -> &PerPlatform<BuildSpec> {
        self.local.build()
    }

    fn test(&self) -> &PerPlatform<TestSpec> {
        self.local.test()
    }

    fn source(&self) -> &PerPlatform<RemoteRockSource> {
        &self.source
    }

    fn deploy(&self) -> &PerPlatform<DeploySpec> {
        self.local.deploy()
    }

    fn build_mut(&mut self) -> &mut PerPlatform<BuildSpec> {
        self.local.build_mut()
    }

    fn test_mut(&mut self) -> &mut PerPlatform<TestSpec> {
        self.local.test_mut()
    }

    fn source_mut(&mut self) -> &mut PerPlatform<RemoteRockSource> {
        &mut self.source
    }

    fn deploy_mut(&mut self) -> &mut PerPlatform<DeploySpec> {
        self.local.deploy_mut()
    }

    fn format(&self) -> &Option<RockspecFormat> {
        self.local.format()
    }

    fn to_lua_remote_rockspec_string(&self) -> Result<String, Self::Error> {
        Ok(self.local.raw_content.clone())
    }
}

#[derive(Error, Debug, Diagnostic)]
#[non_exhaustive]
pub enum LuaVersionError {
    #[error(
        r#"
The lua version {0} is not supported by {1} version {2}.

HINT: If Lux has auto-detected an incompatible Lua installation,
      use `--lua-version` to specify the Lua version to use.
      Valid versions are: '5.1', '5.2', '5.3', '5.4', '5.5', 'jit', 'jit52' and 'luau'.
"#
    )]
    LuaVersionUnsupported(LuaVersion, PackageName, PackageVersion),
    #[error(transparent)]
    #[diagnostic(transparent)]
    LuaVersionUnset(#[from] LuaVersionUnset),
}

impl HasIntegrity for RemoteLuaRockspec {
    async fn hash(&self) -> io::Result<Integrity> {
        Ok(Integrity::from(&self.local.raw_content))
    }
}

/// A rock's metadata, to be displayed on the remote package server
#[derive(Clone, Deserialize, Debug, PartialEq, Default, lux_macros::DisplayAsLuaKV)]
#[cfg_attr(feature = "schema", derive(schemars::JsonSchema))]
#[display_lua(key = "description")]
pub struct RockDescription {
    /// A one-line description of the package.
    pub summary: Option<String>,
    /// A longer description of the package.
    pub detailed: Option<String>,
    /// The license used by the package.
    pub license: Option<String>,
    /// An URL for the project. This is not the URL for the tarball, but the address of a website.
    #[serde(default, deserialize_with = "deserialize_url")]
    #[cfg_attr(feature = "schema", schemars(with = "Option<String>"))]
    pub homepage: Option<Url>,
    /// An URL for the project's issue tracker.
    #[serde(default, deserialize_with = "deserialize_url")]
    #[cfg_attr(feature = "schema", schemars(with = "Option<String>"))]
    pub issues_url: Option<Url>,
    /// Contact information for the rockspec maintainer.
    pub maintainer: Option<String>,
    /// A list of short strings that specify labels for categorization of this rock.
    #[serde(default)]
    pub labels: Vec<String>,
}

fn find_key_span(rockspec_content: &str, key: &str) -> SourceSpan {
    rockspec_content
        .match_indices(key)
        .find(|&(start, _)| {
            let end = start + key.len();
            let prefix_ok =
                start == 0 || !rockspec_content.as_bytes()[start - 1].is_ascii_alphanumeric();
            let suffix_ok = rockspec_content
                .get(end..)
                .is_some_and(|s| s.starts_with('=') || s.trim_start().starts_with('='));
            prefix_ok && suffix_ok
        })
        .map(|(start, _)| (start..start + key.len()).into())
        .unwrap_or((0..0).into())
}

fn find_value_span(rockspec_content: &str, key: &str) -> SourceSpan {
    let key = find_key_span(rockspec_content, key);
    if key.is_empty() {
        return key;
    }
    let key_offset = key.offset();
    let key_end = key_offset + key.len();
    let eq_start = key_end + rockspec_content[key_end..].find('=').unwrap_or(0);
    let value_offset = eq_start + 1;
    let line_end = rockspec_content[value_offset..]
        .find('\n')
        .map(|i| value_offset + i)
        .unwrap_or(rockspec_content.len());
    let trimmed = &rockspec_content[value_offset..line_end];
    let start = value_offset + (trimmed.len() - trimmed.trim_start().len());
    let end = line_end - (trimmed.len() - trimmed.trim_end().len());
    (start..end).into()
}

fn deserialize_url<'de, D>(deserializer: D) -> Result<Option<Url>, D::Error>
where
    D: serde::Deserializer<'de>,
{
    let s = Option::<String>::deserialize(deserializer)?;
    s.map(|s| Url::parse(&s).map_err(serde::de::Error::custom))
        .transpose()
}

#[derive(Error, Debug, Diagnostic)]
#[non_exhaustive]
#[error("invalid rockspec format")]
pub struct InvalidRockspecFormat(String);

#[derive(Default, Clone, Debug, PartialEq, Eq, Serialize, Deserialize)]
#[cfg_attr(feature = "schema", derive(schemars::JsonSchema))]
pub enum RockspecFormat {
    #[serde(rename = "1.0")]
    _1_0,
    #[serde(rename = "2.0")]
    _2_0,
    #[serde(rename = "3.0")]
    #[default]
    _3_0,
}

impl FromStr for RockspecFormat {
    type Err = InvalidRockspecFormat;

    fn from_str(s: &str) -> Result<Self, Self::Err> {
        match s {
            "1.0" => Ok(Self::_1_0),
            "2.0" => Ok(Self::_2_0),
            "3.0" => Ok(Self::_3_0),
            txt => Err(InvalidRockspecFormat(txt.to_string())),
        }
    }
}

impl Display for RockspecFormat {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            Self::_1_0 => write!(f, "1.0"),
            Self::_2_0 => write!(f, "2.0"),
            Self::_3_0 => write!(f, "3.0"),
        }
    }
}

fn parse_lua_tbl_or_default<T>(
    ctx: ottavino::Context<'_>,
    field: &str,
) -> Result<T, ParseLuaRockspecFieldError>
where
    T: Default,
    T: DeserializeOwned,
{
    let ret = match ctx.globals().get_value(ctx, field.to_string()) {
        ottavino::Value::Nil => T::default(),
        value @ ottavino::Value::Table(_) => {
            from_value(value).map_err(|source| ParseLuaRockspecFieldError::DeserializeTable {
                field: field.to_string(),
                source: Box::new(source),
            })?
        }
        value => Err(ParseLuaRockspecFieldError::NotATable {
            field: field.to_string(),
            invalid_type: value.type_name().to_string(),
        })?,
    };
    Ok(ret)
}

#[cfg(test)]
mod tests {
    use std::path::PathBuf;

    use crate::git::GitSource;
    use crate::lua_rockspec::PlatformIdentifier;
    use crate::package::PackageSpec;

    use super::*;

    #[test]
    pub fn parse_rockspec() {
        let rockspec_content = "
        rockspec_format = '1.0'\n
        package = 'foo'\n
        version = '1.0.0-1'\n
        source = {\n
            url = 'https://github.com/lumen-oss/rocks.nvim/archive/1.0.0/rocks.nvim.zip',\n
        }\n
        "
        .to_string();
        let rockspec = RemoteLuaRockspec::new(&rockspec_content).unwrap();
        assert_eq!(rockspec.local.rockspec_format, Some(RockspecFormat::_1_0));
        assert_eq!(rockspec.local.package, "foo".into());
        assert_eq!(rockspec.local.version, "1.0.0-1".parse().unwrap());
        assert_eq!(rockspec.local.description, RockDescription::default());

        let rockspec_content = "
        package = 'bar'\n
        version = '2.0.0-1'\n
        description = {}\n
        source = {\n
            url = 'https://github.com/lumen-oss/rocks.nvim/archive/1.0.0/rocks.nvim.zip',\n
        }\n
        "
        .to_string();
        let rockspec = RemoteLuaRockspec::new(&rockspec_content).unwrap();
        assert_eq!(rockspec.local.rockspec_format, None);
        assert_eq!(rockspec.local.package, "bar".into());
        assert_eq!(rockspec.local.version, "2.0.0-1".parse().unwrap());
        assert_eq!(rockspec.local.description, RockDescription::default());

        let rockspec_content = "
        package = 'rocks.nvim'\n
        version = '3.0.0-1'\n
        description = {\n
            summary = 'some summary',
            detailed = 'some detailed description',
            license = 'MIT',
            homepage = 'https://github.com/lumen-oss/rocks.nvim',
            issues_url = 'https://github.com/lumen-oss/rocks.nvim/issues',
            maintainer = 'Lumen Labs',
        }\n
        source = {\n
            url = 'https://github.com/lumen-oss/rocks.nvim/archive/1.0.0/rocks.nvim.zip',\n
        }\n
        "
        .to_string();
        let rockspec = RemoteLuaRockspec::new(&rockspec_content).unwrap();
        assert_eq!(rockspec.local.rockspec_format, None);
        assert_eq!(rockspec.local.package, "rocks.nvim".into());
        assert_eq!(rockspec.local.version, "3.0.0-1".parse().unwrap());
        let expected_description = RockDescription {
            summary: Some("some summary".into()),
            detailed: Some("some detailed description".into()),
            license: Some("MIT".into()),
            homepage: Some(Url::parse("https://github.com/lumen-oss/rocks.nvim").unwrap()),
            issues_url: Some(Url::parse("https://github.com/lumen-oss/rocks.nvim/issues").unwrap()),
            maintainer: Some("Lumen Labs".into()),
            labels: Vec::new(),
        };
        assert_eq!(rockspec.local.description, expected_description);

        let rockspec_content = "
        package = 'rocks.nvim'\n
        version = '3.0.0-1'\n
        description = {\n
            summary = 'some summary',
            detailed = 'some detailed description',
            license = 'MIT',
            homepage = 'https://github.com/lumen-oss/rocks.nvim',
            issues_url = 'https://github.com/lumen-oss/rocks.nvim/issues',
            maintainer = 'Lumen Labs',
            labels = {},
        }\n
        external_dependencies = { FOO = { library = 'foo' } }\n
        source = {\n
            url = 'https://github.com/lumen-oss/rocks.nvim/archive/1.0.0/rocks.nvim.zip',\n
        }\n
        "
        .to_string();
        let rockspec = RemoteLuaRockspec::new(&rockspec_content).unwrap();
        assert_eq!(rockspec.local.rockspec_format, None);
        assert_eq!(rockspec.local.package, "rocks.nvim".into());
        assert_eq!(rockspec.local.version, "3.0.0-1".parse().unwrap());
        let expected_description = RockDescription {
            summary: Some("some summary".into()),
            detailed: Some("some detailed description".into()),
            license: Some("MIT".into()),
            homepage: Some(Url::parse("https://github.com/lumen-oss/rocks.nvim").unwrap()),
            issues_url: Some(Url::parse("https://github.com/lumen-oss/rocks.nvim/issues").unwrap()),
            maintainer: Some("Lumen Labs".into()),
            labels: Vec::new(),
        };
        assert_eq!(rockspec.local.description, expected_description);
        assert_eq!(
            *rockspec
                .local
                .external_dependencies
                .default
                .get("FOO")
                .unwrap(),
            ExternalDependencySpec {
                library: Some("foo".into()),
                header: None
            }
        );

        let rockspec_content = "
        package = 'rocks.nvim'\n
        version = '3.0.0-1'\n
        description = {\n
            summary = 'some summary',
            detailed = 'some detailed description',
            license = 'MIT',
            homepage = 'https://github.com/lumen-oss/rocks.nvim',
            issues_url = 'https://github.com/lumen-oss/rocks.nvim/issues',
            maintainer = 'Lumen Labs',
            labels = { 'package management', },
        }\n
        supported_platforms = { 'unix', '!windows' }\n
        dependencies = { 'neorg ~> 6' }\n
        build_dependencies = { 'foo' }\n
        external_dependencies = { FOO = { header = 'foo.h' } }\n
        test_dependencies = { 'busted >= 2.0.0' }\n
        source = {\n
            url = 'git+https://github.com/lumen-oss/rocks.nvim',\n
            hash = 'sha256-uU0nuZNNPgilLlLX2n2r+sSE7+N6U4DukIj3rOLvzek=',\n
        }\n
        "
        .to_string();
        let rockspec = RemoteLuaRockspec::new(&rockspec_content).unwrap();
        assert_eq!(rockspec.local.rockspec_format, None);
        assert_eq!(rockspec.local.package, "rocks.nvim".into());
        assert_eq!(rockspec.local.version, "3.0.0-1".parse().unwrap());
        let expected_description = RockDescription {
            summary: Some("some summary".into()),
            detailed: Some("some detailed description".into()),
            license: Some("MIT".into()),
            homepage: Some(Url::parse("https://github.com/lumen-oss/rocks.nvim").unwrap()),
            issues_url: Some(Url::parse("https://github.com/lumen-oss/rocks.nvim/issues").unwrap()),
            maintainer: Some("Lumen Labs".into()),
            labels: vec!["package management".into()],
        };
        assert_eq!(rockspec.local.description, expected_description);
        assert!(rockspec
            .local
            .supported_platforms
            .is_supported(&PlatformIdentifier::Unix));
        assert!(!rockspec
            .local
            .supported_platforms
            .is_supported(&PlatformIdentifier::Windows));
        let neorg = PackageSpec::parse("neorg".into(), "6.0.0".into()).unwrap();
        assert!(rockspec
            .local
            .dependencies
            .default
            .into_iter()
            .any(|dep| dep.matches(&neorg)));
        let foo = PackageSpec::parse("foo".into(), "1.0.0".into()).unwrap();
        assert!(rockspec
            .local
            .build_dependencies
            .default
            .into_iter()
            .any(|dep| dep.matches(&foo)));
        let busted = PackageSpec::parse("busted".into(), "2.2.0".into()).unwrap();
        assert_eq!(
            *rockspec
                .local
                .external_dependencies
                .default
                .get("FOO")
                .unwrap(),
            ExternalDependencySpec {
                header: Some("foo.h".into()),
                library: None
            }
        );
        assert!(rockspec
            .local
            .test_dependencies
            .default
            .into_iter()
            .any(|dep| dep.matches(&busted)));

        let rockspec_content = "
        rockspec_format = '1.0'\n
        package = 'foo'\n
        version = '1.0.0-1'\n
        source = {\n
            url = 'git+https://hub.com/owner/example-project/',\n
            branch = 'bar',\n
        }\n
        "
        .to_string();
        let rockspec = RemoteLuaRockspec::new(&rockspec_content).unwrap();
        assert_eq!(
            rockspec.local.source.default.source_spec,
            RockSourceSpec::Git(GitSource {
                url: "https://hub.com/owner/example-project/".parse().unwrap(),
                git_ref: Some(GitRef::Branch("bar".into()))
            })
        );
        assert_eq!(rockspec.local.test, PerPlatform::default());
        let rockspec_content = "
        rockspec_format = '1.0'\n
        package = 'foo'\n
        version = '1.0.0-1'\n
        source = {\n
            url = 'git+https://hub.com/owner/example-project/',\n
            tag = 'bar',\n
        }\n
        "
        .to_string();
        let rockspec = RemoteLuaRockspec::new(&rockspec_content).unwrap();
        assert_eq!(
            rockspec.local.source.default.source_spec,
            RockSourceSpec::Git(GitSource {
                url: "https://hub.com/owner/example-project/".parse().unwrap(),
                git_ref: Some(GitRef::Tag("bar".into()))
            })
        );
        let rockspec_content = "
        rockspec_format = '1.0'\n
        package = 'foo'\n
        version = '1.0.0-1'\n
        source = {\n
            url = 'git+https://hub.com/owner/example-project/',\n
            branch = 'bar',\n
            tag = 'baz',\n
        }\n
        "
        .to_string();
        let _rockspec = RemoteLuaRockspec::new(&rockspec_content).unwrap_err();
        let rockspec_content = "
        rockspec_format = '1.0'\n
        package = 'foo'\n
        version = '1.0.0-1'\n
        source = {\n
            url = 'git+https://hub.com/owner/example-project/',\n
            tag = 'bar',\n
            file = 'foo.tar.gz',\n
        }\n
        build = {\n
            install = {\n
                conf = {['foo.bar'] = 'config/bar.toml'},\n
            },\n
        }\n
        "
        .to_string();
        let rockspec = RemoteLuaRockspec::new(&rockspec_content).unwrap();
        assert_eq!(
            rockspec.local.source.default.archive_name,
            Some("foo.tar.gz".into())
        );
        let foo_bar_path = rockspec
            .local
            .build
            .default
            .install
            .conf
            .get("foo.bar")
            .unwrap();
        assert_eq!(*foo_bar_path, PathBuf::from("config/bar.toml"));
        let rockspec_content = "
        rockspec_format = '1.0'\n
        package = 'foo'\n
        version = '1.0.0-1'\n
        source = {\n
            url = 'git+https://hub.com/example-project/foo.zip',\n
        }\n
        build = {\n
            install = {\n
                lua = {\n
                    'foo.lua',\n
                    ['foo.bar'] = 'src/bar.lua',\n
                },\n
                bin = {['foo.bar'] = 'bin/bar'},\n
            },\n
        }\n
        "
        .to_string();
        let rockspec = RemoteLuaRockspec::new(&rockspec_content).unwrap();
        assert!(matches!(
            rockspec.local.build.default.build_backend,
            Some(BuildBackendSpec::Builtin { .. })
        ));
        let install_lua_spec = rockspec.local.build.default.install.lua;
        let foo_bar_path = install_lua_spec
            .get(&LuaModule::from_str("foo.bar").unwrap())
            .unwrap();
        assert_eq!(*foo_bar_path, PathBuf::from("src/bar.lua"));
        let foo_path = install_lua_spec
            .get(&LuaModule::from_str("foo").unwrap())
            .unwrap();
        assert_eq!(*foo_path, PathBuf::from("foo.lua"));
        let foo_bar_path = rockspec
            .local
            .build
            .default
            .install
            .bin
            .get("foo.bar")
            .unwrap();
        assert_eq!(*foo_bar_path, PathBuf::from("bin/bar"));
        let rockspec_content = "
        rockspec_format = '1.0'\n
        package = 'foo'\n
        version = '1.0.0-1'\n
        source = {\n
            url = 'git+https://hub.com/example-project/',\n
        }\n
        build = {\n
            copy_directories = { 'lua' },\n
        }\n
        "
        .to_string();
        let _rockspec = RemoteLuaRockspec::new(&rockspec_content).unwrap_err();
        let rockspec_content = "
        rockspec_format = '1.0'\n
        package = 'foo'\n
        version = '1.0.0-1'\n
        source = {\n
            url = 'git+https://hub.com/example-project/',\n
        }\n
        build = {\n
            copy_directories = { 'lib' },\n
        }\n
        "
        .to_string();
        let _rockspec = RemoteLuaRockspec::new(&rockspec_content).unwrap_err();
        let rockspec_content = "
        rockspec_format = '1.0'\n
        package = 'foo'\n
        version = '1.0.0-1'\n
        source = {\n
            url = 'git+https://hub.com/example-project/',\n
        }\n
        build = {\n
            copy_directories = { 'rock_manifest' },\n
        }\n
        "
        .to_string();
        let _rockspec = RemoteLuaRockspec::new(&rockspec_content).unwrap_err();
        let rockspec_content = "
        rockspec_format = '1.0'\n
        package = 'foo'\n
        version = '1.0.0-1'\n
        source = {\n
            url = 'git+https://hub.com/example-project/foo.zip',\n
            dir = 'baz',\n
        }\n
        build = {\n
            type = 'make',\n
            install = {\n
                lib = {['foo.so'] = 'lib/bar.so'},\n
            },\n
            copy_directories = {\n
                'plugin',\n
                'ftplugin',\n
            },\n
            patches = {\n
                ['lua51-support.diff'] = [[\n
                    --- before.c\n
                    +++ path/to/after.c\n
                ]],\n
            },\n
        }\n
        "
        .to_string();
        let rockspec = RemoteLuaRockspec::new(&rockspec_content).unwrap();
        assert_eq!(rockspec.local.source.default.unpack_dir, Some("baz".into()));
        assert_eq!(
            rockspec.local.build.default.build_backend,
            Some(BuildBackendSpec::Make(MakeBuildSpec::default()))
        );
        let foo_bar_path = rockspec
            .local
            .build
            .default
            .install
            .lib
            .get("foo.so")
            .unwrap();
        assert_eq!(*foo_bar_path, PathBuf::from("lib/bar.so"));
        let copy_directories = rockspec.local.build.default.copy_directories;
        assert_eq!(
            copy_directories,
            vec![PathBuf::from("plugin"), PathBuf::from("ftplugin")]
        );
        let patches = rockspec.local.build.default.patches;
        let _patch = patches.get(&PathBuf::from("lua51-support.diff")).unwrap();
        let rockspec_content = "
        rockspec_format = '1.0'\n
        package = 'foo'\n
        version = '1.0.0-1'\n
        source = {\n
            url = 'git+https://hub.com/example-project/foo.zip',\n
        }\n
        build = {\n
            type = 'cmake',\n
        }\n
        "
        .to_string();
        let rockspec = RemoteLuaRockspec::new(&rockspec_content).unwrap();
        assert_eq!(
            rockspec.local.build.default.build_backend,
            Some(BuildBackendSpec::CMake(CMakeBuildSpec::default()))
        );
        let rockspec_content = "
        rockspec_format = '1.0'\n
        package = 'foo'\n
        version = '1.0.0-1'\n
        source = {\n
            url = 'git+https://hub.com/example-project/foo.zip',\n
        }\n
        build = {\n
            type = 'command',\n
            build_command = 'foo',\n
            install_command = 'bar',\n
        }\n
        "
        .to_string();
        let rockspec = RemoteLuaRockspec::new(&rockspec_content).unwrap();
        assert!(matches!(
            rockspec.local.build.default.build_backend,
            Some(BuildBackendSpec::Command(CommandBuildSpec { .. }))
        ));
        let rockspec_content = "
        rockspec_format = '1.0'\n
        package = 'foo'\n
        version = '1.0.0-1'\n
        source = {\n
            url = 'git+https://hub.com/example-project/foo.zip',\n
        }\n
        build = {\n
            type = 'command',\n
            install_command = 'foo',\n
        }\n
        "
        .to_string();
        RemoteLuaRockspec::new(&rockspec_content).unwrap();
        let rockspec_content = "
        rockspec_format = '1.0'\n
        package = 'foo'\n
        version = '1.0.0-1'\n
        source = {\n
            url = 'git+https://hub.com/example-project/foo.zip',\n
        }\n
        build = {\n
            type = 'command',\n
            build_command = 'foo',\n
        }\n
        "
        .to_string();
        RemoteLuaRockspec::new(&rockspec_content).unwrap();
        // platform overrides
        let rockspec_content = "
        package = 'rocks'\n
        version = '3.0.0-1'\n
        dependencies = {\n
          'neorg ~> 6',\n
          'toml-edit ~> 1',\n
          platforms = {\n
            windows = {\n
              'neorg = 5.0.0',\n
              'toml = 1.0.0',\n
            },\n
            unix = {\n
              'neorg = 5.0.0',\n
            },\n
            linux = {\n
              'toml = 1.0.0',\n
            },\n
          },\n
        }\n
        source = {\n
            url = 'git+https://github.com/lumen-oss/rocks.nvim',\n
            hash = 'sha256-uU0nuZNNPgilLlLX2n2r+sSE7+N6U4DukIj3rOLvzek=',\n
        }\n
        "
        .to_string();
        let rockspec = RemoteLuaRockspec::new(&rockspec_content).unwrap();
        let neorg_override = PackageSpec::parse("neorg".into(), "5.0.0".into()).unwrap();
        let toml_edit = PackageSpec::parse("toml-edit".into(), "1.0.0".into()).unwrap();
        let toml = PackageSpec::parse("toml".into(), "1.0.0".into()).unwrap();
        assert_eq!(rockspec.local.dependencies.default.len(), 2);
        let per_platform = &rockspec.local.dependencies.per_platform;
        assert_eq!(
            per_platform
                .get(&PlatformIdentifier::Windows)
                .unwrap()
                .iter()
                .filter(|dep| dep.matches(&neorg_override)
                    || dep.matches(&toml_edit)
                    || dep.matches(&toml))
                .count(),
            3
        );
        assert_eq!(
            per_platform
                .get(&PlatformIdentifier::Unix)
                .unwrap()
                .iter()
                .filter(|dep| dep.matches(&neorg_override)
                    || dep.matches(&toml_edit)
                    || dep.matches(&toml))
                .count(),
            2
        );
        assert_eq!(
            per_platform
                .get(&PlatformIdentifier::Linux)
                .unwrap()
                .iter()
                .filter(|dep| dep.matches(&neorg_override)
                    || dep.matches(&toml_edit)
                    || dep.matches(&toml))
                .count(),
            3
        );
        let rockspec_content = "
        package = 'rocks'\n
        version = '3.0.0-1'\n
        external_dependencies = {\n
            FOO = { library = 'foo' },\n
            platforms = {\n
              windows = {\n
                FOO = { library = 'foo.dll' },\n
              },\n
              unix = {\n
                BAR = { header = 'bar.h' },\n
              },\n
              linux = {\n
                FOO = { library = 'foo.so' },\n
              },\n
            },\n
        }\n
        source = {\n
            url = 'https://github.com/lumen-oss/rocks.nvim/archive/1.0.0/rocks.nvim.zip',\n
        }\n
        "
        .to_string();
        let rockspec = RemoteLuaRockspec::new(&rockspec_content).unwrap();
        assert_eq!(
            *rockspec
                .local
                .external_dependencies
                .default
                .get("FOO")
                .unwrap(),
            ExternalDependencySpec {
                library: Some("foo".into()),
                header: None
            }
        );
        let per_platform = rockspec.local.external_dependencies.per_platform;
        assert_eq!(
            *per_platform
                .get(&PlatformIdentifier::Windows)
                .and_then(|it| it.get("FOO"))
                .unwrap(),
            ExternalDependencySpec {
                library: Some("foo.dll".into()),
                header: None
            }
        );
        assert_eq!(
            *per_platform
                .get(&PlatformIdentifier::Unix)
                .and_then(|it| it.get("FOO"))
                .unwrap(),
            ExternalDependencySpec {
                library: Some("foo".into()),
                header: None
            }
        );
        assert_eq!(
            *per_platform
                .get(&PlatformIdentifier::Unix)
                .and_then(|it| it.get("BAR"))
                .unwrap(),
            ExternalDependencySpec {
                header: Some("bar.h".into()),
                library: None
            }
        );
        assert_eq!(
            *per_platform
                .get(&PlatformIdentifier::Linux)
                .and_then(|it| it.get("BAR"))
                .unwrap(),
            ExternalDependencySpec {
                header: Some("bar.h".into()),
                library: None
            }
        );
        assert_eq!(
            *per_platform
                .get(&PlatformIdentifier::Linux)
                .and_then(|it| it.get("FOO"))
                .unwrap(),
            ExternalDependencySpec {
                library: Some("foo.so".into()),
                header: None
            }
        );
        let rockspec_content = "
        rockspec_format = '1.0'\n
        package = 'foo'\n
        version = '1.0.0-1'\n
        source = {\n
            url = 'git+https://hub.com/example-project/.git',\n
            branch = 'bar',\n
            platforms = {\n
                macosx = {\n
                    branch = 'mac',\n
                },\n
                windows = {\n
                    url = 'git+https://winhub.com/example-project/.git',\n
                    branch = 'win',\n
                },\n
            },\n
        }\n
        "
        .to_string();
        let rockspec = RemoteLuaRockspec::new(&rockspec_content).unwrap();
        assert_eq!(
            rockspec.local.source.default.source_spec,
            RockSourceSpec::Git(GitSource {
                url: "https://hub.com/example-project/.git".parse().unwrap(),
                git_ref: Some(GitRef::Branch("bar".into()))
            })
        );
        assert_eq!(
            rockspec
                .source
                .per_platform
                .get(&PlatformIdentifier::MacOSX)
                .map(|it| it.source_spec.clone())
                .unwrap(),
            RockSourceSpec::Git(GitSource {
                url: "https://hub.com/example-project/.git".parse().unwrap(),
                git_ref: Some(GitRef::Branch("mac".into()))
            })
        );
        assert_eq!(
            rockspec
                .source
                .per_platform
                .get(&PlatformIdentifier::Windows)
                .map(|it| it.source_spec.clone())
                .unwrap(),
            RockSourceSpec::Git(GitSource {
                url: "https://winhub.com/example-project/.git".parse().unwrap(),
                git_ref: Some(GitRef::Branch("win".into()))
            })
        );
        let rockspec_content = "
        rockspec_format = '1.0'\n
        package = 'foo'\n
        version = '1.0.0-1'\n
        source = { url = 'git+https://hub.com/example-project/foo.zip' }\n
        build = {\n
            type = 'make',\n
            install = {\n
                lib = {['foo.bar'] = 'lib/bar.so'},\n
            },\n
            copy_directories = { 'plugin' },\n
            platforms = {\n
                unix = {\n
                    copy_directories = { 'ftplugin' },\n
                },\n
                linux = {\n
                    copy_directories = { 'foo' },\n
                },\n
            },\n
        }\n
        "
        .to_string();
        let rockspec = RemoteLuaRockspec::new(&rockspec_content).unwrap();
        let per_platform = rockspec.local.build.per_platform;
        let unix = per_platform.get(&PlatformIdentifier::Unix).unwrap();
        assert_eq!(
            unix.copy_directories,
            vec![PathBuf::from("plugin"), PathBuf::from("ftplugin")]
        );
        let linux = per_platform.get(&PlatformIdentifier::Linux).unwrap();
        assert_eq!(
            linux.copy_directories,
            vec![
                PathBuf::from("plugin"),
                PathBuf::from("foo"),
                PathBuf::from("ftplugin")
            ]
        );
        let rockspec_content = "
        package = 'foo'\n
        version = '1.0.0-1'\n
        source = { url = 'git+https://hub.com/example-project/foo.zip' }\n
        build = {\n
            type = 'builtin',\n
            modules = {\n
                cjson = {\n
                    sources = { 'lua_cjson.c', 'strbuf.c', 'fpconv.c' },\n
                }\n
            },\n
            platforms = {\n
                win32 = { modules = { cjson = { defines = {\n
                    'DISABLE_INVALID_NUMBERS', 'USE_INTERNAL_ISINF'\n
                } } } }\n
            },\n
        }\n
        "
        .to_string();
        let rockspec = RemoteLuaRockspec::new(&rockspec_content).unwrap();
        let win32 = rockspec.local.build.get(&PlatformIdentifier::Windows);
        assert_eq!(
            win32.build_backend,
            Some(BuildBackendSpec::Builtin(BuiltinBuildSpec {
                modules: vec![(
                    LuaModule::from_str("cjson").unwrap(),
                    ModuleSpec::ModulePaths(ModulePaths {
                        sources: vec!["lua_cjson.c".into(), "strbuf.c".into(), "fpconv.c".into()],
                        libraries: Vec::default(),
                        defines: vec![
                            ("DISABLE_INVALID_NUMBERS".into(), None),
                            ("USE_INTERNAL_ISINF".into(), None)
                        ],
                        incdirs: Vec::default(),
                        libdirs: Vec::default(),
                    })
                )]
                .into_iter()
                .collect()
            }))
        );
        let rockspec_content = "
        rockspec_format = '1.0'\n
        package = 'foo'\n
        version = '1.0.0-1'\n
        deploy = {\n
            wrap_bin_scripts = false,\n
        }\n
        source = { url = 'git+https://hub.com/example-project/foo.zip' }\n
        ";
        let rockspec = RemoteLuaRockspec::new(rockspec_content).unwrap();
        let deploy_spec = &rockspec.deploy().current_platform();
        assert!(!deploy_spec.wrap_bin_scripts);
    }

    #[test]
    pub fn parse_scm_rockspec() {
        let rockspec_content = "
        package = 'foo'\n
        version = 'scm-1'\n
        source = {\n
            url = 'https://github.com/lumen-oss/rocks.nvim/archive/1.0.0/rocks.nvim.zip',\n
        }\n
        "
        .to_string();
        let rockspec = RemoteLuaRockspec::new(&rockspec_content).unwrap();
        assert_eq!(rockspec.local.package, "foo".into());
        assert_eq!(rockspec.local.version, "scm-1".parse().unwrap());
    }

    #[test]
    pub fn regression_luasystem() {
        let rockspec_content = String::from_utf8(
            std::fs::read(
                PathBuf::from(env!("CARGO_MANIFEST_DIR"))
                    .join("resources/test/luasystem-0.4.4-1.rockspec"),
            )
            .unwrap(),
        )
        .unwrap();
        let rockspec = RemoteLuaRockspec::new(&rockspec_content).unwrap();
        let build_spec = rockspec.local.build.current_platform();
        assert!(matches!(
            build_spec.build_backend,
            Some(BuildBackendSpec::Builtin { .. })
        ));
        if let Some(BuildBackendSpec::Builtin(BuiltinBuildSpec { modules })) =
            &build_spec.build_backend
        {
            assert_eq!(
                modules.get(&LuaModule::from_str("system.init").unwrap()),
                Some(&ModuleSpec::SourcePath("system/init.lua".into()))
            );
            assert_eq!(
                modules.get(&LuaModule::from_str("system.core").unwrap()),
                Some(&ModuleSpec::ModulePaths(ModulePaths {
                    sources: vec![
                        "src/core.c".into(),
                        "src/compat.c".into(),
                        "src/time.c".into(),
                        "src/environment.c".into(),
                        "src/random.c".into(),
                        "src/term.c".into(),
                        "src/bitflags.c".into(),
                        "src/wcwidth.c".into(),
                    ],
                    defines: luasystem_expected_defines(),
                    libraries: luasystem_expected_libraries(),
                    incdirs: luasystem_expected_incdirs(),
                    libdirs: luasystem_expected_libdirs(),
                }))
            );
        }
        if let Some(BuildBackendSpec::Builtin(BuiltinBuildSpec { modules })) = &rockspec
            .local
            .build
            .get(&PlatformIdentifier::Windows)
            .build_backend
        {
            if let ModuleSpec::ModulePaths(paths) = modules
                .get(&LuaModule::from_str("system.core").unwrap())
                .unwrap()
            {
                assert_eq!(paths.libraries, luasystem_expected_windows_libraries());
            };
        }
        if let Some(BuildBackendSpec::Builtin(BuiltinBuildSpec { modules })) = &rockspec
            .local
            .build
            .get(&PlatformIdentifier::Win32)
            .build_backend
        {
            if let ModuleSpec::ModulePaths(paths) = modules
                .get(&LuaModule::from_str("system.core").unwrap())
                .unwrap()
            {
                assert_eq!(paths.libraries, luasystem_expected_windows_libraries());
            };
        }
    }

    fn luasystem_expected_defines() -> Vec<(String, Option<String>)> {
        if cfg!(target_os = "windows") {
            vec![
                ("WINVER".into(), Some("0x0600".into())),
                ("_WIN32_WINNT".into(), Some("0x0600".into())),
            ]
        } else {
            Vec::default()
        }
    }

    fn luasystem_expected_windows_libraries() -> Vec<PathBuf> {
        vec!["advapi32".into(), "winmm".into()]
    }
    fn luasystem_expected_libraries() -> Vec<PathBuf> {
        if cfg!(any(target_os = "linux", target_os = "android")) {
            vec!["rt".into()]
        } else if cfg!(target_os = "windows") {
            luasystem_expected_windows_libraries()
        } else {
            Vec::default()
        }
    }

    fn luasystem_expected_incdirs() -> Vec<PathBuf> {
        Vec::default()
    }

    fn luasystem_expected_libdirs() -> Vec<PathBuf> {
        Vec::default()
    }

    #[test]
    pub fn rust_mlua_rockspec() {
        let rockspec_content = "
    package = 'foo'\n
    version = 'scm-1'\n
    source = {\n
        url = 'https://github.com/lumen-oss/rocks.nvim/archive/1.0.0/rocks.nvim.zip',\n
    }\n
    build = {
        type = 'rust-mlua',
        modules = {
            'foo',
            bar = 'baz',
        },
        target_path = 'path/to/cargo/target/directory',
        default_features = false,
        include = {
            'file.lua',
            ['path/to/another/file.lua'] = 'another-file.lua',
        },
        features = {'extra', 'features'},
    }
            ";
        let rockspec = RemoteLuaRockspec::new(rockspec_content).unwrap();
        let build_spec = rockspec.local.build.current_platform();
        if let Some(BuildBackendSpec::RustMlua(build_spec)) = build_spec.build_backend.to_owned() {
            assert_eq!(
                build_spec.modules.get("foo").unwrap(),
                &PathBuf::from(format!("libfoo.{}", std::env::consts::DLL_EXTENSION))
            );
            assert_eq!(
                build_spec.modules.get("bar").unwrap(),
                &PathBuf::from(format!("libbaz.{}", std::env::consts::DLL_EXTENSION))
            );
            assert_eq!(
                build_spec.include.get(&PathBuf::from("file.lua")).unwrap(),
                &PathBuf::from("file.lua")
            );
            assert_eq!(
                build_spec
                    .include
                    .get(&PathBuf::from("path/to/another/file.lua"))
                    .unwrap(),
                &PathBuf::from("another-file.lua")
            );
        } else {
            panic!("Expected RustMlua build backend");
        }
    }

    #[test]
    pub fn rust_binary_rockspec() {
        let rockspec_content = "
    package = 'foo'\n
    version = 'scm-1'\n
    source = {\n
        url = 'https://github.com/lumen-oss/rocks.nvim/archive/1.0.0/rocks.nvim.zip',\n
    }\n
    build = {\n
        type = 'rust-binary',\n
        package = 'foo',\n
        features = {'extra', 'features'},\n
    }\n
            ";
        let rockspec = RemoteLuaRockspec::new(rockspec_content).unwrap();
        let build_spec = rockspec.local.build.current_platform();
        if let Some(BuildBackendSpec::RustBinary(build_spec)) = build_spec.build_backend.to_owned()
        {
            assert_eq!(build_spec.package, Some("foo".to_string()));
            assert_eq!(build_spec.features, vec!["extra", "features"]);
        } else {
            panic!("Expected RustBinary build backend");
        }
    }

    #[tokio::test]
    pub async fn regression_ltui() {
        let content = String::from_utf8(
            std::fs::read(
                PathBuf::from(env!("CARGO_MANIFEST_DIR"))
                    .join("resources/test/ltui-2.8-2.rockspec"),
            )
            .unwrap(),
        )
        .unwrap();
        RemoteLuaRockspec::new(&content).unwrap();
    }

    // Luarocks allows the `install.bin` field to be a list, even though it
    // should only allow a table.
    #[test]
    pub fn regression_off_spec_install_binaries() {
        let rockspec_content = r#"
            package = "WSAPI"
            version = "1.7-1"

            source = {
              url = "git://github.com/keplerproject/wsapi",
              tag = "v1.7",
            }

            build = {
              type = "builtin",
              modules = {
                ["wsapi"] = "src/wsapi.lua",
              },
              -- Offending Line
              install = { bin = { "src/launcher/wsapi.cgi" } }
            }
        "#;

        let rockspec = RemoteLuaRockspec::new(rockspec_content).unwrap();

        assert_eq!(
            rockspec.build().current_platform().install.bin,
            HashMap::from([("wsapi.cgi".into(), PathBuf::from("src/launcher/wsapi.cgi"))])
        );
    }

    #[test]
    pub fn regression_external_dependencies() {
        let content = String::from_utf8(
            std::fs::read(
                PathBuf::from(env!("CARGO_MANIFEST_DIR"))
                    .join("resources/test/luaossl-20220711-0.rockspec"),
            )
            .unwrap(),
        )
        .unwrap();
        let rockspec = RemoteLuaRockspec::new(&content).unwrap();
        if cfg!(target_family = "unix") {
            assert_eq!(
                rockspec
                    .local
                    .external_dependencies
                    .current_platform()
                    .get("OPENSSL")
                    .unwrap(),
                &ExternalDependencySpec {
                    library: Some("ssl".into()),
                    header: Some("openssl/ssl.h".into()),
                }
            );
        }
        let per_platform = rockspec.local.external_dependencies.per_platform;
        assert_eq!(
            *per_platform
                .get(&PlatformIdentifier::Windows)
                .and_then(|it| it.get("OPENSSL"))
                .unwrap(),
            ExternalDependencySpec {
                library: Some("libeay32".into()),
                header: Some("openssl/ssl.h".into()),
            }
        );
    }

    #[test]
    pub fn remote_lua_rockspec_from_package_and_source_spec() {
        let package_req = "foo@1.0.5".parse().unwrap();
        let source = GitSource {
            url: "https://hub.com/owner/example-project.git".parse().unwrap(),
            git_ref: Some(GitRef::Tag("1.0.5".into())),
        };
        let source_spec = RockSourceSpec::Git(source);
        let rockspec = RemoteLuaRockspec::from_source_spec(
            package_req,
            source_spec.clone(),
            Vec::new(),
            HashMap::new(),
        );
        let generated_rockspec_str = rockspec.local.raw_content;
        let rockspec2 = RemoteLuaRockspec::new(&generated_rockspec_str).unwrap();
        assert_eq!(rockspec2.local.package, "foo".into());
        assert_eq!(rockspec2.local.version, "1.0.5".parse().unwrap());
        assert_eq!(rockspec2.local.source, PerPlatform::new(source_spec.into()));
    }

    #[test]
    pub fn regression_complex_source_field() {
        let rockspec_content = r#"
            package = "say"
            local rock_version = "1.4.1"
            local rock_release = "3"
            local namespace = "lunarmodules"
            local repository = package

            version = ("%s-%s"):format(rock_version, rock_release)

            source = {
              url = ("git+https://github.com/%s/%s.git"):format(namespace, repository),
              branch = rock_version == "scm" and "master" or nil,
              tag = rock_version ~= "scm" and "v"..rock_version or nil,
            }

            description = {
              summary = "Lua string hashing/indexing library",
            }

            dependencies = {
              "lua >= 5.1",
            }

            build = {
              type = "builtin",
            }
        "#
        .to_string();
        RemoteLuaRockspec::new(&rockspec_content).unwrap();
    }

    fn eval_lua_global<T: serde::de::DeserializeOwned>(code: &str, key: &'static str) -> T {
        use ottavino::{Closure, Executor, Fuel, Lua};
        use ottavino_util::serde::from_value;
        Lua::core()
            .try_enter(|ctx| {
                let closure = Closure::load(ctx, None, code.as_bytes())?;
                let executor = Executor::start(ctx, closure.into(), ());
                executor.step(ctx, &mut Fuel::with(i32::MAX))?;
                from_value(ctx.globals().get_value(ctx, key)).map_err(ottavino::Error::from)
            })
            .unwrap()
    }

    #[test]
    pub fn rock_description_roundtrip() {
        let desc = RockDescription {
            summary: Some("A great package".into()),
            detailed: Some("Detailed description here".into()),
            license: Some("MIT".into()),
            homepage: Some("https://example.com".parse().unwrap()),
            issues_url: Some("https://example.com/issues".parse().unwrap()),
            maintainer: Some("Some Maintainer <maintainer@example.com>".into()),
            labels: vec!["neovim".into(), "lua".into()],
        };
        let lua = desc.display_lua().to_string();
        let restored: RockDescription = eval_lua_global(&lua, "description");
        assert_eq!(desc, restored);
    }

    #[test]
    pub fn rock_description_empty_roundtrip() {
        let desc = RockDescription::default();
        let lua = desc.display_lua().to_string();
        // An empty description produces an empty assignment; restore from empty table.
        let lua = if lua.trim().is_empty() {
            "description = {}".to_string()
        } else {
            lua
        };
        let restored: RockDescription = eval_lua_global(&lua, "description");
        assert_eq!(desc, restored);
    }

    #[test]
    pub fn regression_luasec() {
        let rockspec_content = r#"
package = "LuaSec"
version = "1.3.2-1"
source = {
  url = "git+https://github.com/brunoos/luasec",
  tag = "v1.3.2",
}
external_dependencies = {
   platforms = {
      unix = {
         OPENSSL = {
            header = "openssl/ssl.h",
            library = "ssl"
         }
      },
      windows = {
         OPENSSL = {
            header = "openssl/ssl.h",
         }
      },
   }
}
"#;
        let rockspec = RemoteLuaRockspec::new(rockspec_content).unwrap();
        let linux_external_deps = rockspec
            .external_dependencies()
            .get(&PlatformIdentifier::Linux);
        assert!(linux_external_deps.get("OPENSSL").is_some());
        let windows_external_deps = rockspec
            .external_dependencies()
            .get(&PlatformIdentifier::Windows);
        assert!(windows_external_deps.get("OPENSSL").is_some());
    }

    #[test]
    pub fn find_value_span_spaces() {
        let source = "package = 'foo'\nversion = '1.0'";
        let span = find_value_span(source, "package");
        assert_eq!(&source[span.offset()..span.offset() + span.len()], "'foo'");
    }

    #[test]
    pub fn find_value_span_no_spaces() {
        let source = "package='foo'";
        let span = find_value_span(source, "package");
        assert_eq!(&source[span.offset()..span.offset() + span.len()], "'foo'");
    }

    #[test]
    pub fn find_value_span_returns_empty_when_not_found() {
        let source = "version = '1.0'";
        let span = find_value_span(source, "package");
        assert_eq!(span.len(), 0);
    }

    #[test]
    pub fn find_value_span_skips_url_containing_field() {
        let source = "homepage = 'https://luapackager.com'\npackage = 'foo'";
        let span = find_value_span(source, "package");
        assert_eq!(&source[span.offset()..span.offset() + span.len()], "'foo'");
    }

    #[test]
    pub fn find_value_span_double_quoted() {
        let source = r#"package = "foo""#;
        let span = find_value_span(source, "package");
        assert_eq!(
            &source[span.offset()..span.offset() + span.len()],
            r#""foo""#
        );
    }

    #[test]
    pub fn find_value_span_table() {
        let source = "description = {\n    summary = 'bar'\n}";
        let span = find_value_span(source, "description");
        assert_eq!(&source[span.offset()..span.offset() + span.len()], "{");
    }

    #[test]
    pub fn parse_lua_rockspec_field_error() {
        let rockspec_content = "package = 'my-lua-lib'\nfoo = 123";
        let extern_err = ottavino::Lua::core()
            .try_enter(|_ctx| {
                Err(ParseLuaRockspecFieldError::Deserialize {
                    field: "foo".into(),
                    source: Box::new(ottavino_util::serde::de::Error::Message(
                        "expected bar".into(),
                    )),
                })?;
                Ok(())
            })
            .unwrap_err();
        let err = Error::from_ottavino(extern_err, rockspec_content);
        match err {
            Error::ExecutionError {
                cause,
                err_span: Some(span),
                ..
            } => {
                assert!(cause.contains("invalid rockspec field 'foo'"));
                assert_eq!(
                    &rockspec_content[span.offset()..span.offset() + span.len()],
                    "foo"
                );
            }
            other => panic!("expected ExecutionError with err_span, got: {other:?}"),
        }
        let extern_err = ottavino::Lua::core()
            .try_enter(|_ctx| {
                Err(ParseLuaRockspecFieldError::DeserializeTable {
                    field: "foo".into(),
                    source: Box::new(ottavino_util::serde::de::Error::Message(
                        "expected bar".into(),
                    )),
                })?;
                Ok(())
            })
            .unwrap_err();
        let err = Error::from_ottavino(extern_err, rockspec_content);
        match err {
            Error::ExecutionError {
                cause,
                err_span: Some(span),
                ..
            } => {
                assert!(cause.contains("invalid rockspec field 'foo'"));
                assert_eq!(
                    &rockspec_content[span.offset()..span.offset() + span.len()],
                    "foo"
                );
            }
            other => panic!("expected ExecutionError with err_span, got: {other:?}"),
        }
        let extern_err = ottavino::Lua::core()
            .try_enter(|_ctx| {
                Err(ParseLuaRockspecFieldError::NotATable {
                    field: "foo".into(),
                    invalid_type: "integer".into(),
                })?;
                Ok(())
            })
            .unwrap_err();
        let err = Error::from_ottavino(extern_err, rockspec_content);
        match err {
            Error::ExecutionError {
                cause,
                err_span: Some(span),
                ..
            } => {
                assert!(cause.contains("invalid rockspec field 'foo'"));
                assert_eq!(
                    &rockspec_content[span.offset()..span.offset() + span.len()],
                    "foo"
                );
            }
            other => panic!("expected ExecutionError with err_span, got: {other:?}"),
        }
    }
}