Support '--infer steam' for wrap command

CHANGELOG.md

@@ -1,3 +1,9 @@
+## Unreleased
+
+* Added:
+  * CLI: The `wrap` command now supports `--infer steam`,
+    which will check for the `STEAMAPPID` environment variable.
+
 ## v0.22.0 (2023-12-26)
 
 * Added:

src/cli.rs

@@ -31,7 +31,7 @@ use crate::{
         layout::BackupLayout, prepare_backup_target, scan_game_for_backup, BackupId, DuplicateDetector, Launchers,
         OperationStepDecision, SteamShortcuts, TitleFinder,
     },
-    wrap::{heroic::infer_game_from_heroic, WrapGameInfo},
+    wrap::{heroic::infer_game_from_heroic, infer_game_from_steam, WrapGameInfo},
 };
 
 #[derive(Clone, Debug, Default)]
@@ -792,6 +792,10 @@ pub fn run(sub: Subcommand, no_manifest_update: bool, try_manifest_update: bool)
             gui,
             commands,
         } => {
+            let manifest = load_manifest(&config, &mut cache, no_manifest_update, try_manifest_update)?;
+            let layout = BackupLayout::new(config.restore.path.clone(), config.backup.retention.clone());
+            let title_finder = TitleFinder::new(&manifest, &layout);
+
             // Determine raw game identifiers
             let wrap_game_info = if let Some(name) = name_source.name.as_ref() {
                 Some(WrapGameInfo {
@@ -802,6 +806,7 @@ pub fn run(sub: Subcommand, no_manifest_update: bool, try_manifest_update: bool)
                 let roots = config.expanded_roots();
                 match infer {
                     parse::LauncherTypes::Heroic => infer_game_from_heroic(&roots, &commands),
+                    parse::LauncherTypes::Steam => infer_game_from_steam(&title_finder),
                 }
             } else {
                 unreachable!();
@@ -812,9 +817,6 @@ pub fn run(sub: Subcommand, no_manifest_update: bool, try_manifest_update: bool)
             //
             // e.g. "Slain: Back From Hell" from legendary to "Slain: Back from
             // Hell" as known to ludusavi
-            let manifest = load_manifest(&config, &mut cache, no_manifest_update, try_manifest_update)?;
-            let layout = BackupLayout::new(config.restore.path.clone(), config.backup.retention.clone());
-            let title_finder = TitleFinder::new(&manifest, &layout);
             let game_name = wrap_game_info.as_ref().and_then(|wrap_game_info| {
                 title_finder.maybe_find_one(wrap_game_info.name.as_ref(), None, wrap_game_info.gog_id, true)
             });

src/cli/parse.rs

@@ -115,6 +115,7 @@ impl From<CliSort> for Sort {
 #[derive(Debug, Copy, Clone, PartialEq, Eq, PartialOrd, Ord, ValueEnum)]
 pub enum LauncherTypes {
     Heroic,
+    Steam,
 }
 
 #[derive(clap::Subcommand, Clone, Debug, PartialEq, Eq)]

src/wrap.rs

@@ -1,3 +1,5 @@
+use crate::scan::TitleFinder;
+
 pub mod heroic;
 
 /// Returned game information with whatever we could find
@@ -12,3 +14,20 @@ impl WrapGameInfo {
         self.name.is_none() && self.gog_id.is_none()
     }
 }
+
+pub fn infer_game_from_steam(title_finder: &TitleFinder) -> Option<WrapGameInfo> {
+    let app_id = std::env::var("STEAMAPPID").ok()?.parse::<u32>().ok()?;
+
+    log::debug!("Found Steam environment variable: STEAMAPPID={}", app_id,);
+
+    let result = WrapGameInfo {
+        name: title_finder.find_one(&[], &Some(app_id), &None, false),
+        gog_id: None,
+    };
+
+    if result.is_empty() {
+        None
+    } else {
+        Some(result)
+    }
+}