/*
 * This file is part of espanso.
 *
 * Copyright (C) 2019-2021 Federico Terzi
 *
 * espanso is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * espanso is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with espanso.  If not, see <https://www.gnu.org/licenses/>.
 */

use std::path::Path;

use anyhow::{bail, Result};

mod archive;
#[macro_use]
mod logging;
mod manifest;
mod package;
mod provider;
mod resolver;
mod util;

pub use archive::{ArchivedPackage, Archiver, SaveOptions, StoredPackage};
pub use package::Package;
pub use provider::{PackageProvider, PackageSpecifier, ProviderOptions};

pub fn get_provider(
    package: &PackageSpecifier,
    runtime_dir: &Path,
    options: &ProviderOptions,
) -> Result<Box<dyn PackageProvider>> {
    if let Some(git_repo_url) = package.git_repo_url.as_deref() {
        if !package.use_native_git {
            let matches_known_hosts = if let Some(github_parts) =
                util::github::extract_github_url_parts(git_repo_url)
            {
                if let Some(repo_scheme) =
                    util::github::resolve_repo_scheme(github_parts, package.git_branch.as_deref())?
                {
                    return Ok(Box::new(provider::github::GitHubPackageProvider::new(
                        repo_scheme.author,
                        repo_scheme.name,
                        repo_scheme.branch,
                    )));
                }

                true
            } else if let Some(gitlab_parts) = util::gitlab::extract_gitlab_url_parts(git_repo_url)
            {
                if let Some(repo_scheme) =
                    util::gitlab::resolve_repo_scheme(gitlab_parts, package.git_branch.as_deref())?
                {
                    return Ok(Box::new(provider::gitlab::GitLabPackageProvider::new(
                        repo_scheme.author,
                        repo_scheme.name,
                        repo_scheme.branch,
                    )));
                }

                true
            } else {
                false
            };

            // Git repository seems to be in one of the known hosts, but the direct methods
            // couldn't retrieve its content. This might happen with private repos (as they are not
            // available to non-authenticated requests), so we check if a "git ls-remote" command
            // is able to access it.
            if matches_known_hosts && !util::git::is_private_repo(git_repo_url) {
                bail!("could not access repository: {}, make sure it exists and that you have the necessary access rights.", git_repo_url);
            }
        }

        // Git repository is neither on Github or Gitlab
        // OR it's a private repository, which means we can't use the direct method
        // (because it's not authenticated)
        Ok(Box::new(provider::git::GitPackageProvider::new()))
    } else {
        // Download from the official espanso hub
        Ok(Box::new(provider::hub::EspansoHubPackageProvider::new(
            runtime_dir,
            options.force_index_update,
        )))
    }
}

pub fn get_archiver(package_dir: &Path) -> Result<Box<dyn Archiver>> {
    Ok(Box::new(archive::default::DefaultArchiver::new(
        package_dir,
    )))
}

#[cfg(test)]
pub(crate) mod tests {
    use super::*;
    use tempdir::TempDir;

    pub fn run_with_temp_dir(action: impl FnOnce(&Path)) {
        let tmp_dir = TempDir::new("espanso-package").unwrap();
        let tmp_path = tmp_dir.path();

        action(tmp_path);
    }
}
