Refactor: Extract new crate binstalk-fetchers (#1291)

To reduce `binstalk` codegen and enable better reuse. Signed-off-by: Jiahao XU <Jiahao_XU@outlook.com>
2025-05-06 12:10:02 +00:00 · 2023-08-14 13:20:34 +10:00 · 2023-08-14 13:20:34 +10:00 · 76c72469eb
commit 76c72469eb
parent 623f7ff4ed
19 changed files with 1008 additions and 122 deletions
--- a/crates/binstalk-fetchers/src/gh_crate_meta/hosting.rs
+++ b/crates/binstalk-fetchers/src/gh_crate_meta/hosting.rs
@ -0,0 +1,117 @@
+use itertools::Itertools;
+use leon::{Item, Template};
+use leon_macros::template;
+use url::Url;
+
+#[derive(Copy, Clone, Debug, Eq, PartialEq)]
+pub enum RepositoryHost {
+    GitHub,
+    GitLab,
+    BitBucket,
+    SourceForge,
+    Unknown,
+}
+
+/// Make sure to update possible_dirs in `bins::infer_bin_dir_template`
+/// if you modified FULL_FILENAMES or NOVERSION_FILENAMES.
+pub const FULL_FILENAMES: &[Template<'_>] = &[
+    template!("/{ name }-{ target }-v{ version }{ archive-suffix }"),
+    template!("/{ name }-{ target }-{ version }{ archive-suffix }"),
+    template!("/{ name }-{ version }-{ target }{ archive-suffix }"),
+    template!("/{ name }-v{ version }-{ target }{ archive-suffix }"),
+    template!("/{ name }_{ target }_v{ version }{ archive-suffix }"),
+    template!("/{ name }_{ target }_{ version }{ archive-suffix }"),
+    template!("/{ name }_{ version }_{ target }{ archive-suffix }"),
+    template!("/{ name }_v{ version }_{ target }{ archive-suffix }"),
+];
+
+pub const NOVERSION_FILENAMES: &[Template<'_>] = &[
+    template!("/{ name }-{ target }{ archive-suffix }"),
+    template!("/{ name }_{ target }{ archive-suffix }"),
+];
+
+const GITHUB_RELEASE_PATHS: &[Template<'_>] = &[
+    template!("{ repo }/releases/download/{ version }"),
+    template!("{ repo }/releases/download/v{ version }"),
+    // %2F is escaped form of '/'
+    template!("{ repo }/releases/download/{ subcrate }%2F{ version }"),
+    template!("{ repo }/releases/download/{ subcrate }%2Fv{ version }"),
+];
+
+const GITLAB_RELEASE_PATHS: &[Template<'_>] = &[
+    template!("{ repo }/-/releases/{ version }/downloads/binaries"),
+    template!("{ repo }/-/releases/v{ version }/downloads/binaries"),
+    // %2F is escaped form of '/'
+    template!("{ repo }/-/releases/{ subcrate }%2F{ version }/downloads/binaries"),
+    template!("{ repo }/-/releases/{ subcrate }%2Fv{ version }/downloads/binaries"),
+];
+
+const BITBUCKET_RELEASE_PATHS: &[Template<'_>] = &[template!("{ repo }/downloads")];
+
+const SOURCEFORGE_RELEASE_PATHS: &[Template<'_>] = &[
+    template!("{ repo }/files/binaries/{  version }"),
+    template!("{ repo }/files/binaries/v{ version }"),
+    // %2F is escaped form of '/'
+    template!("{ repo }/files/binaries/{ subcrate }%2F{  version }"),
+    template!("{ repo }/files/binaries/{ subcrate }%2Fv{ version }"),
+];
+
+impl RepositoryHost {
+    pub fn guess_git_hosting_services(repo: &Url) -> Self {
+        use RepositoryHost::*;
+
+        match repo.domain() {
+            Some(domain) if domain.starts_with("github") => GitHub,
+            Some(domain) if domain.starts_with("gitlab") => GitLab,
+            Some(domain) if domain == "bitbucket.org" => BitBucket,
+            Some(domain) if domain == "sourceforge.net" => SourceForge,
+            _ => Unknown,
+        }
+    }
+
+    pub fn get_default_pkg_url_template(
+        self,
+    ) -> Option<impl Iterator<Item = Template<'static>> + Clone + 'static> {
+        use RepositoryHost::*;
+
+        match self {
+            GitHub => Some(apply_filenames_to_paths(
+                GITHUB_RELEASE_PATHS,
+                &[FULL_FILENAMES, NOVERSION_FILENAMES],
+                "",
+            )),
+            GitLab => Some(apply_filenames_to_paths(
+                GITLAB_RELEASE_PATHS,
+                &[FULL_FILENAMES, NOVERSION_FILENAMES],
+                "",
+            )),
+            BitBucket => Some(apply_filenames_to_paths(
+                BITBUCKET_RELEASE_PATHS,
+                &[FULL_FILENAMES],
+                "",
+            )),
+            SourceForge => Some(apply_filenames_to_paths(
+                SOURCEFORGE_RELEASE_PATHS,
+                &[FULL_FILENAMES, NOVERSION_FILENAMES],
+                "/download",
+            )),
+            Unknown => None,
+        }
+    }
+}
+
+fn apply_filenames_to_paths(
+    paths: &'static [Template<'static>],
+    filenames: &'static [&'static [Template<'static>]],
+    suffix: &'static str,
+) -> impl Iterator<Item = Template<'static>> + Clone + 'static {
+    filenames
+        .iter()
+        .flat_map(|fs| fs.iter())
+        .cartesian_product(paths.iter())
+        .map(move |(filename, path)| {
+            let mut template = path.clone() + filename;
+            template += Item::Text(suffix);
+            template
+        })
+}