Add option to search for icons (#120)

2026-09-01 05:53:41 +02:00 · 2022-05-13 00:31:35 +02:00
parent 21a696fdbb
commit 9f59957edc
8 changed files with 1116 additions and 787 deletions
@@ -2747,9 +2747,9 @@ dependencies = [

 [[package]]
 name = "steamgriddb_api"
-version = "0.2.0"
+version = "0.3.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "49989dd0a9cb16bc02ea1939b513e310fcdbc31d52981d8c70bcb86a23b12d5f"
+checksum = "14968a884ba517e194649a482617c37e908ac9cb4dbcb55582a838705ee705c7"
 dependencies = [
 "reqwest",
 "serde 1.0.136",
@@ -6,7 +6,7 @@ edition = "2021"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 [dependencies]
 steam_shortcuts_util = "1.1.7"
-steamgriddb_api = "^0.2,0"
+steamgriddb_api = "^0.3.0"
 serde = { version = "^1.0.136", features = ["derive"] }
 serde_json = "^1.0.79"
 tokio = { version = "^1.17.0", features = ["full"] }
@@ -0,0 +1,338 @@
+#!/usr/bin/env python3
+
+#Source: https://github.com/flatpak/flatpak-builder-tools/blob/master/cargo/flatpak-cargo-generator.py
+
+__license__ = 'MIT'
+import json
+from urllib.parse import urlparse, ParseResult, parse_qs
+import os
+import glob
+import subprocess
+import argparse
+import logging
+import hashlib
+import asyncio
+import aiohttp
+import toml
+
+CRATES_IO = 'https://static.crates.io/crates'
+CARGO_HOME = 'cargo'
+CARGO_CRATES = f'{CARGO_HOME}/vendor'
+VENDORED_SOURCES = 'vendored-sources'
+GIT_CACHE = 'flatpak-cargo/git'
+COMMIT_LEN = 7
+
+
+def canonical_url(url):
+    'Converts a string to a Cargo Canonical URL, as per https://github.com/rust-lang/cargo/blob/35c55a93200c84a4de4627f1770f76a8ad268a39/src/cargo/util/canonical_url.rs#L19'
+    # Hrm. The upstream cargo does not replace those URLs, but if we don't then it doesn't work too well :(
+    url = url.replace('git+https://', 'https://')
+    u = urlparse(url)
+    # It seems cargo drops query and fragment
+    u = ParseResult(u.scheme, u.netloc, u.path, None, None, None)
+    u = u._replace(path = u.path.rstrip('/'))
+
+    if u.netloc == 'github.com':
+        u = u._replace(scheme = 'https')
+        u = u._replace(path = u.path.lower())
+
+    if u.path.endswith('.git'):
+        u = u._replace(path = u.path[:-len('.git')])
+
+    return u
+
+
+def get_git_tarball(repo_url, commit):
+    url = canonical_url(repo_url)
+    path = url.path.split('/')[1:]
+
+    assert len(path) == 2
+    owner = path[0]
+    if path[1].endswith('.git'):
+        repo = path[1].replace('.git', '')
+    else:
+        repo = path[1]
+    if url.hostname == 'github.com':
+        return f'https://codeload.{url.hostname}/{owner}/{repo}/tar.gz/{commit}'
+    elif url.hostname.split('.')[0] == 'gitlab':
+        return f'https://{url.hostname}/{owner}/{repo}/-/archive/{commit}/{repo}-{commit}.tar.gz'
+    elif url.hostname == 'bitbucket.org':
+        return f'https://{url.hostname}/{owner}/{repo}/get/{commit}.tar.gz'
+    else:
+        raise ValueError(f'Don\'t know how to get tarball for {repo_url}')
+
+
+async def get_remote_sha256(url):
+    logging.info(f"started sha256({url})")
+    sha256 = hashlib.sha256()
+    async with aiohttp.ClientSession(raise_for_status=True) as http_session:
+        async with http_session.get(url) as response:
+            while True:
+                data = await response.content.read(4096)
+                if not data:
+                    break
+                sha256.update(data)
+    logging.info(f"done sha256({url})")
+    return sha256.hexdigest()
+
+
+def load_toml(tomlfile='Cargo.lock'):
+    with open(tomlfile, 'r') as f:
+        toml_data = toml.load(f)
+    return toml_data
+
+
+def git_repo_name(git_url, commit):
+    name = canonical_url(git_url).path.split('/')[-1]
+    return f'{name}-{commit[:COMMIT_LEN]}'
+
+
+def fetch_git_repo(git_url, commit):
+    repo_dir = git_url.replace('://', '_').replace('/', '_')
+    cache_dir = os.environ.get('XDG_CACHE_HOME', os.path.expanduser('~/.cache'))
+    clone_dir = os.path.join(cache_dir, 'flatpak-cargo', repo_dir)
+    if not os.path.isdir(os.path.join(clone_dir, '.git')):
+        subprocess.run(['git', 'clone', git_url, clone_dir], check=True)
+    rev_parse_proc = subprocess.run(['git', 'rev-parse', 'HEAD'], cwd=clone_dir, check=True,
+                                    stdout=subprocess.PIPE)
+    head = rev_parse_proc.stdout.decode().strip()
+    if head[:COMMIT_LEN] != commit[:COMMIT_LEN]:
+        subprocess.run(['git', 'fetch', 'origin', commit], cwd=clone_dir, check=True)
+        subprocess.run(['git', 'checkout', commit], cwd=clone_dir, check=True)
+    return clone_dir
+
+
+async def get_git_repo_packages(git_url, commit):
+    logging.info('Loading packages from %s', git_url)
+    git_repo_dir = fetch_git_repo(git_url, commit)
+    root_toml = load_toml(os.path.join(git_repo_dir, 'Cargo.toml'))
+    assert 'package' in root_toml or 'workspace' in root_toml
+    packages = {}
+
+    async def get_dep_packages(entry, toml_dir):
+        # https://doc.rust-lang.org/cargo/reference/specifying-dependencies.html
+        if 'dependencies' in entry:
+            for dep_name, dep in entry['dependencies'].items():
+                if 'package' in dep:
+                    dep_name = dep['package']
+                if 'path' not in dep:
+                    continue
+                if dep_name in packages:
+                    continue
+                dep_dir = os.path.normpath(os.path.join(toml_dir, dep['path']))
+                logging.debug("Loading dependency %s from %s in %s", dep_name, dep_dir, git_url)
+                dep_toml = load_toml(os.path.join(git_repo_dir, dep_dir, 'Cargo.toml'))
+                assert dep_toml['package']['name'] == dep_name, (git_url, toml_dir)
+                await get_dep_packages(dep_toml, dep_dir)
+                packages[dep_name] = dep_dir
+        if 'target' in entry:
+            for _, target in entry['target'].items():
+                await get_dep_packages(target, toml_dir)
+
+    if 'package' in root_toml:
+        await get_dep_packages(root_toml, '.')
+        packages[root_toml['package']['name']] = '.'
+
+    if 'workspace' in root_toml:
+        for member in root_toml['workspace']['members']:
+            for subpkg_toml in glob.glob(os.path.join(git_repo_dir, member, 'Cargo.toml')):
+                subpkg = os.path.relpath(os.path.dirname(subpkg_toml), git_repo_dir)
+                logging.debug("Loading workspace member %s in %s", member, git_url)
+                pkg_toml = load_toml(subpkg_toml)
+                await get_dep_packages(pkg_toml, subpkg)
+                packages[pkg_toml['package']['name']] = subpkg
+
+    logging.debug('Packages in %s:\n%s', git_url, json.dumps(packages, indent=4))
+    return packages
+
+
+async def get_git_repo_sources(url, commit, tarball=False):
+    name = git_repo_name(url, commit)
+    if tarball:
+        tarball_url = get_git_tarball(url, commit)
+        git_repo_sources = [{
+            'type': 'archive',
+            'archive-type': 'tar-gzip',
+            'url': tarball_url,
+            'sha256': await get_remote_sha256(tarball_url),
+            'dest': f'{GIT_CACHE}/{name}',
+        }]
+    else:
+        git_repo_sources = [{
+            'type': 'git',
+            'url': url,
+            'commit': commit,
+            'dest': f'{GIT_CACHE}/{name}',
+        }]
+    return git_repo_sources
+
+
+async def get_git_package_sources(package, git_repos):
+    name = package['name']
+    source = package['source']
+    commit = urlparse(source).fragment
+    assert commit, 'The commit needs to be indicated in the fragement part'
+    canonical = canonical_url(source)
+    repo_url = canonical.geturl()
+
+    git_repo = git_repos.setdefault(repo_url, {
+        'commits': {},
+        'lock': asyncio.Lock(),
+    })
+    async with git_repo['lock']:
+        if commit not in git_repo['commits']:
+            git_repo['commits'][commit] = await get_git_repo_packages(repo_url, commit)
+
+    cargo_vendored_entry = {
+        repo_url: {
+            'git': repo_url,
+            'replace-with': VENDORED_SOURCES,
+        }
+    }
+    rev = parse_qs(urlparse(source).query).get('rev')
+    tag = parse_qs(urlparse(source).query).get('tag')
+    branch = parse_qs(urlparse(source).query).get('branch')
+    if rev:
+        assert len(rev) == 1
+        cargo_vendored_entry[repo_url]['rev'] = rev[0]
+    elif tag:
+        assert len(tag) == 1
+        cargo_vendored_entry[repo_url]['tag'] = tag[0]
+    elif branch:
+        assert len(branch) == 1
+        cargo_vendored_entry[repo_url]['branch'] = branch[0]
+
+    logging.info("Adding package %s from %s", name, repo_url)
+    pkg_subpath = git_repo['commits'][commit][name]
+    pkg_repo_dir = os.path.join(GIT_CACHE, git_repo_name(repo_url, commit), pkg_subpath)
+    git_sources = [
+        {
+            'type': 'shell',
+            'commands': [
+                f'cp -r --reflink=auto "{pkg_repo_dir}" "{CARGO_CRATES}/{name}"'
+            ],
+        },
+        {
+            'type': 'inline',
+            'contents': json.dumps({'package': None, 'files': {}}),
+            'dest': f'{CARGO_CRATES}/{name}', #-{version}',
+            'dest-filename': '.cargo-checksum.json',
+        }
+    ]
+
+    return (git_sources, cargo_vendored_entry)
+
+
+async def get_package_sources(package, cargo_lock, git_repos):
+    metadata = cargo_lock.get('metadata')
+    name = package['name']
+    version = package['version']
+
+    if 'source' not in package:
+        logging.debug('%s has no source', name)
+        return
+    source = package['source']
+
+    if source.startswith('git+'):
+        return await get_git_package_sources(package, git_repos)
+
+    key = f'checksum {name} {version} ({source})'
+    if metadata is not None and key in metadata:
+        checksum = metadata[key]
+    elif 'checksum' in package:
+        checksum = package['checksum']
+    else:
+        logging.warning(f'{name} doesn\'t have checksum')
+        return
+    crate_sources = [
+        {
+            'type': 'archive',
+            'archive-type': 'tar-gzip',
+            'url': f'{CRATES_IO}/{name}/{name}-{version}.crate',
+            'sha256': checksum,
+            'dest': f'{CARGO_CRATES}/{name}-{version}',
+        },
+        {
+            'type': 'inline',
+            'contents': json.dumps({'package': checksum, 'files': {}}),
+            'dest': f'{CARGO_CRATES}/{name}-{version}',
+            'dest-filename': '.cargo-checksum.json',
+        },
+    ]
+    return (crate_sources, {'crates-io': {'replace-with': VENDORED_SOURCES}})
+
+
+async def generate_sources(cargo_lock, git_tarballs=False):
+    # {
+    #     "git-repo-url": {
+    #         "lock": asyncio.Lock(),
+    #         "commits": {
+    #             "commit-hash": {
+    #                 "package-name": "./relative/package/path"
+    #             }
+    #         }
+    #     }
+    # }
+    git_repos = {}
+    sources = []
+    package_sources = []
+    cargo_vendored_sources = {
+        VENDORED_SOURCES: {'directory': f'{CARGO_CRATES}'},
+    }
+
+    pkg_coros = [get_package_sources(p, cargo_lock, git_repos) for p in cargo_lock['package']]
+    for pkg in await asyncio.gather(*pkg_coros):
+        if pkg is None:
+            continue
+        else:
+            pkg_sources, cargo_vendored_entry = pkg
+        package_sources.extend(pkg_sources)
+        cargo_vendored_sources.update(cargo_vendored_entry)
+
+    logging.debug('Adding collected git repos:\n%s', json.dumps(list(git_repos), indent=4))
+    git_repo_coros = []
+    for git_url, git_repo in git_repos.items():
+        for git_commit in git_repo['commits']:
+            git_repo_coros.append(get_git_repo_sources(git_url, git_commit, git_tarballs))
+    sources.extend(sum(await asyncio.gather(*git_repo_coros), []))
+
+    sources.extend(package_sources)
+
+    logging.debug('Vendored sources:\n%s', json.dumps(cargo_vendored_sources, indent=4))
+    sources.append({
+        'type': 'inline',
+        'contents': toml.dumps({
+            'source': cargo_vendored_sources,
+        }),
+        'dest': CARGO_HOME,
+        'dest-filename': 'config'
+    })
+    return sources
+
+
+def main():
+    parser = argparse.ArgumentParser()
+    parser.add_argument('cargo_lock', help='Path to the Cargo.lock file')
+    parser.add_argument('-o', '--output', required=False, help='Where to write generated sources')
+    parser.add_argument('-t', '--git-tarballs', action='store_true', help='Download git repos as tarballs')
+    parser.add_argument('-d', '--debug', action='store_true')
+    args = parser.parse_args()
+    if args.output is not None:
+        outfile = args.output
+    else:
+        outfile = 'generated-sources.json'
+    if args.debug:
+        loglevel = logging.DEBUG
+    else:
+        loglevel = logging.INFO
+    logging.basicConfig(level=loglevel)
+
+    generated_sources = asyncio.run(generate_sources(load_toml(args.cargo_lock),
+                                    git_tarballs=args.git_tarballs))
+    with open(outfile, 'w') as out:
+        json.dump(generated_sources, out, indent=4, sort_keys=False)
+
+
+if __name__ == '__main__':
+    main()
+
@@ -25,6 +25,11 @@ https://hughsie.github.io/oars/index.html
 -->
  <content_rating type="oars-1.1" />
  <releases>
+    <release version="1.3.1" date="2022-05-13">
+      <description>
+        <p>It is now possible to download icons for games</p>
+      </description>
+    </release>
    <release version="1.3.0" date="2022-05-05">
      <description>
        <p>First Flatpak Release of BoilR</p>
@@ -0,0 +1,2 @@
+cd ..
+python3 ./flatpak/flatpak-cargo-generator.py ./Cargo.lock -o ./flatpak/cargo-lock.json
@@ -200,52 +200,37 @@ async fn search_for_images_to_download(

        let shortcuts: Vec<&ShortcutOwned> = images_needed.collect();

-        if let ImageType::Icon = image_type {
-            for (index, image_id) in image_ids.iter().enumerate() {
-                let shortcut = shortcuts[index];
-                if let Some(url) = get_steam_icon_url(*image_id).await {
-                    let path = grid_folder.join(image_type.file_name(shortcut.app_id));
-                    to_download.push(ToDownload {
-                        path,
-                        url,
-                        app_name: shortcut.app_name.clone(),
-                        image_type,
-                    });
-                }
-            }
-        } else {
-            for image_ids in image_ids.chunks(99) {
-                let image_search_result =
-                    get_images_for_ids(client, image_ids, &image_type, download_animated).await;
-                match image_search_result {
-                    Ok(images) => {
-                        let images = images
-                            .iter()
-                            .enumerate()
-                            .map(|(index, image)| (image, shortcuts[index], image_ids[index]));
-                        let download_for_this_type = stream::iter(images)
-                            .filter_map(|(image, shortcut, game_id)| {
-                                let path = grid_folder.join(image_type.file_name(shortcut.app_id));
-                                async move {
-                                    let image_url = match image {
-                                        Ok(img) => Some(img.url.clone()),
-                                        Err(_) => get_steam_image_url(game_id, &image_type).await,
-                                    };
-                                    image_url.map(|url| ToDownload {
-                                        path,
-                                        url,
-                                        app_name: shortcut.app_name.clone(),
-                                        image_type,
-                                    })
-                                }
-                            })
-                            .collect::<Vec<ToDownload>>()
-                            .await;
+        for image_ids in image_ids.chunks(99) {
+            let image_search_result =
+                get_images_for_ids(client, image_ids, &image_type, download_animated).await;
+            match image_search_result {
+                Ok(images) => {
+                    let images = images
+                        .iter()
+                        .enumerate()
+                        .map(|(index, image)| (image, shortcuts[index], image_ids[index]));
+                    let download_for_this_type = stream::iter(images)
+                        .filter_map(|(image, shortcut, game_id)| {
+                            let path = grid_folder.join(image_type.file_name(shortcut.app_id));
+                            async move {
+                                let image_url = match image {
+                                    Ok(img) => Some(img.url.clone()),
+                                    Err(_) => get_steam_image_url(game_id, &image_type).await,
+                                };
+                                image_url.map(|url| ToDownload {
+                                    path,
+                                    url,
+                                    app_name: shortcut.app_name.clone(),
+                                    image_type,
+                                })
+                            }
+                        })
+                        .collect::<Vec<ToDownload>>()
+                        .await;

-                        to_download.extend(download_for_this_type);
-                    }
-                    Err(err) => println!("Error getting images: {}", err),
+                    to_download.extend(download_for_this_type);
                }
+                Err(err) => println!("Error getting images: {}", err),
            }
        }
    }
@@ -301,18 +286,29 @@ pub fn get_query_type(
        nsfw: Some(&Nsfw::False),
        ..Default::default()
    };
+
+    use steamgriddb_api::query_parameters::IconQueryParameters;
+    let icon_parameters = IconQueryParameters {
+        nsfw: Some(&Nsfw::False),
+        ..Default::default()
+    };
    let query_type = match image_type {
        ImageType::Hero => steamgriddb_api::QueryType::Hero(Some(hero_parameters)),
        ImageType::BigPicture => steamgriddb_api::QueryType::Grid(Some(big_picture_parameters)),
        ImageType::Grid => steamgriddb_api::QueryType::Grid(Some(grid_parameters)),
        ImageType::WideGrid => steamgriddb_api::QueryType::Grid(Some(big_picture_parameters)),
        ImageType::Logo => steamgriddb_api::QueryType::Logo(Some(logo_parameters)),
-        _ => panic!("Unsupported image type"),
+        ImageType::Icon => steamgriddb_api::QueryType::Icon(Some(icon_parameters)),
    };
    query_type
 }

 async fn get_steam_image_url(game_id: usize, image_type: &ImageType) -> Option<String> {
+    if let ImageType::Icon = image_type {
+        if let Some(url) = get_steam_icon_url(game_id).await {
+            return Some(url);
+        }
+    }
    let steamgriddb_page_url = format!("https://www.steamgriddb.com/api/public/game/{}/", game_id);
    let response = reqwest::get(steamgriddb_page_url).await;
    if let Ok(response) = response {
@@ -575,7 +575,8 @@ fn render_shortcut_images(ui: &mut egui::Ui, state: &ImageSelectState) -> Option
                _ => None,
            })
            .flatten();
-        if render_thumbnail(ui, texture, &image_type) {
+        let clicked = render_thumbnail(ui, texture);
+        if clicked {
            return Some(UserAction::ImageTypeSelected(*image_type));
        }
    }
@@ -597,28 +598,15 @@ fn render_user_select(state: &ImageSelectState, ui: &mut egui::Ui) -> UserAction

 const MAX_WIDTH: f32 = 300.;

-fn render_thumbnail(
-    ui: &mut egui::Ui,
-    image: Option<egui::TextureHandle>,
-    image_type: &ImageType,
-) -> bool {
+fn render_thumbnail(ui: &mut egui::Ui, image: Option<egui::TextureHandle>) -> bool {
    if let Some(texture) = image {
        let mut size = texture.size_vec2();
        clamp_to_width(&mut size, MAX_WIDTH);
        let image_button = ImageButton::new(&texture, size);
        let added = ui.add(image_button);
-        match image_type {
-            ImageType::Icon => false,
-            _ => added.on_hover_text("Click to change image").clicked(),
-        }
+        added.on_hover_text("Click to change image").clicked()
    } else {
-        match image_type {
-            ImageType::Icon => {
-                ui.label("No icon found");
-                false
-            }
-            _ => ui.button("Pick an image").clicked(),
-        }
+        ui.button("Pick an image").clicked()
    }
 }