Skip to content

Commit

Permalink
Format code
Browse files Browse the repository at this point in the history
  • Loading branch information
@ShadowMitia
ShadowMitia committed May 29, 2022
1 parent 7e75a9e commit c9e2582
Showing 1 changed file with 123 additions and 122 deletions.
245 changes: 123 additions & 122 deletions src/main.rs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -25,29 +25,29 @@ fn generate_steam_rungame(id: &str) -> String {
/// Detect if app is a Proton runtime
fn is_proton(app_name: &str) -> bool {
if app_name.starts_with("Proton") {
let number = app_name.split(' ').collect::<Vec<&str>>();
if number.len() > 1 {
let number = number[1].parse::<f32>();
if number.unwrap_or(0.0) != 0.0 {
return true;
}
}
let number = app_name.split(' ').collect::<Vec<&str>>();
if number.len() > 1 {
let number = number[1].parse::<f32>();
if number.unwrap_or(0.0) != 0.0 {
return true;
}
}
}
false
}

/// List of names of applications/games we don't want to launch.
fn is_blacklisted(app_name: &str) -> bool {
let steam_libs = [
"Steamworks Common Redistributables",
"SteamVR",
"Proton Experimental",
"Steamworks Common Redistributables",
"SteamVR",
"Proton Experimental",
];

steam_libs.iter().any(|&b| b == app_name)
|| app_name.ends_with("Soundtrack") // This **should** deal with downloaded albums, and ignore them
|| is_proton(app_name)
|| app_name.starts_with("Steam Linux Runtime")
|| app_name.ends_with("Soundtrack") // This **should** deal with downloaded albums, and ignore them
|| is_proton(app_name)
|| app_name.starts_with("Steam Linux Runtime")
}

// fn parse_vdf(path_to_vdf: &Path) -> HashMap<String, String> {
Expand All @@ -68,7 +68,7 @@ fn is_blacklisted(app_name: &str) -> bool {
// }
// println!("splitted {:?}", key_value);
// res.insert(key_value[0].to_string(), key_value[1].to_string());

// }
// }

Expand All @@ -89,10 +89,10 @@ fn get_other_install_dirs(path: &Path) -> Vec<String> {

let lines = lines.lines();
for line in lines {
if line.contains("path") {
let splitted:Vec<&str> = line.split_whitespace().collect();
libs.push(splitted[1][1..splitted[1].len()-1].to_string());
}
if line.contains("path") {
let splitted:Vec<&str> = line.split_whitespace().collect();
libs.push(splitted[1][1..splitted[1].len()-1].to_string());
}
}
libs
}
Expand All @@ -102,44 +102,44 @@ fn get_games_from_manifest_in_path(path: &Path) -> Vec<(String, String)> {
let dir = std::fs::read_dir(path).unwrap();

let manifest_files = dir
.map(|e| e.unwrap())
.filter(|file| {
let file_name = file.file_name();
let file_name = file_name.to_str().unwrap();
file_name.starts_with("appmanifest")
})
.collect::<Vec<DirEntry>>();
.map(|e| e.unwrap())
.filter(|file| {
let file_name = file.file_name();
let file_name = file_name.to_str().unwrap();
file_name.starts_with("appmanifest")
})
.collect::<Vec<DirEntry>>();

let mut games = Vec::new();

for file in manifest_files {
let file_path = file.path();
let contents = std::fs::read_to_string(file_path);
let lines = contents.unwrap();
let lines = lines.lines().skip(2).collect::<Vec<&str>>();

let mut game = "".to_string();
let mut id = "".to_string();

for line in lines.iter().take(lines.len() - 1) {
let line = line
.split('\t')
.filter(|s| !s.is_empty())
.collect::<Vec<&str>>();
if line[0].contains("name") {
let app_name = line[1].replace("\"", "");

game = app_name.clone();
}

if line[0].contains("appid") {
let app_id = line[1].replace("\"", "");
id = app_id.clone();
}
}
if !is_blacklisted(&game) {
games.push((game, id));
}
let file_path = file.path();
let contents = std::fs::read_to_string(file_path);
let lines = contents.unwrap();
let lines = lines.lines().skip(2).collect::<Vec<&str>>();

let mut game = "".to_string();
let mut id = "".to_string();

for line in lines.iter().take(lines.len() - 1) {
let line = line
.split('\t')
.filter(|s| !s.is_empty())
.collect::<Vec<&str>>();
if line[0].contains("name") {
let app_name = line[1].replace("\"", "");

game = app_name.clone();
}

if line[0].contains("appid") {
let app_id = line[1].replace("\"", "");
id = app_id.clone();
}
}
if !is_blacklisted(&game) {
games.push((game, id));
}
}

games
Expand All @@ -157,26 +157,26 @@ enum SteamKind {
fn detect_steam() -> SteamKind {
let has_steam_vanilla = which::which("steam").is_ok();
let flatpak_list = Command::new("flatpak")
.arg("list")
.stdout(Stdio::piped())
.spawn()
.unwrap();
.arg("list")
.stdout(Stdio::piped())
.spawn()
.unwrap();
let mut flatpak_steam_output = Command::new("grep")
.stdin(flatpak_list.stdout.unwrap())
.arg("-c")
.arg("Steam")
.output()
.unwrap()
.stdout;
.stdin(flatpak_list.stdout.unwrap())
.arg("-c")
.arg("Steam")
.output()
.unwrap()
.stdout;
flatpak_steam_output.remove(flatpak_steam_output.len() - 1);
let has_flatpak_steam = String::from_utf8(flatpak_steam_output).unwrap();

let has_flatpak_steam = has_flatpak_steam.parse::<u32>().unwrap() > 0;

match (has_steam_vanilla, has_flatpak_steam) {
(true, _) => SteamKind::Vanilla,
(_, true) => SteamKind::Flatpak,
_ => SteamKind::NotFound,
(true, _) => SteamKind::Vanilla,
(_, true) => SteamKind::Flatpak,
_ => SteamKind::NotFound,
}
}

Expand All @@ -185,8 +185,8 @@ fn detect_steam() -> SteamKind {
fn detect_steam() -> SteamKind {
let has_steam_vanilla = which::which(r#"C:\Program Files (x86)\Steam\steam.exe"#).is_ok();
match has_steam_vanilla {
true => SteamKind::Vanilla,
_ => SteamKind::NotFound,
true => SteamKind::Vanilla,
_ => SteamKind::NotFound,
}
}

Expand All @@ -195,46 +195,47 @@ fn detect_steam() -> SteamKind {
fn detect_steam() -> SteamKind {
let has_steam_vanilla = which::which("steam").is_ok();
match has_steam_vanilla {
true => SteamKind::Vanilla,
_ => SteamKind::NotFound,
true => SteamKind::Vanilla,
_ => SteamKind::NotFound,
}
}

/// Launche the game from its id using the appropriate Steam environment
#[cfg(target_os = "linux")]
fn run(steam_type: SteamKind, id: &str) -> std::io::Result<Child> {
let child = match steam_type {
SteamKind::Flatpak => std::process::Command::new("flatpak")
.args(&[
"run",
"com.valvesoftware.Steam",
&generate_steam_rungame(id),
])
.stdout(Stdio::null())
.stderr(Stdio::null())
.spawn()?,
SteamKind::Vanilla => std::process::Command::new("steam")
.arg(&generate_steam_rungame(id))
.stdout(Stdio::null())
.stderr(Stdio::null())
.spawn()?,
SteamKind::NotFound => panic!("Couldn't find steam!"),
SteamKind::Flatpak => std::process::Command::new("flatpak")
.args(&[
"run",
"com.valvesoftware.Steam",
&generate_steam_rungame(id),
])
.stdout(Stdio::null())
.stderr(Stdio::null())
.spawn()?,
SteamKind::Vanilla => std::process::Command::new("steam")
.arg(&generate_steam_rungame(id))
.stdout(Stdio::null())
.stderr(Stdio::null())
.spawn()?,
SteamKind::NotFound => panic!("Couldn't find steam!"),
};
println!("{:?} {} {:?}", steam_type, id, child);
Ok(child)
}

/// Launche the game from its id using the appropriate Steam environment
#[cfg(target_os = "windows")]
fn run(steam_type: SteamKind, id: &str) -> std::io::Result<Child> {
let child = match steam_type {
SteamKind::Vanilla => {
std::process::Command::new(r#"C:\Program Files (x86)\Steam\steam.exe"#)
.arg(&generate_steam_rungame(id))
.stdout(Stdio::null())
.stderr(Stdio::null())
.spawn()?
}
_ => panic!("Couldn't find steam!"),
SteamKind::Vanilla => {
std::process::Command::new(r#"C:\Program Files (x86)\Steam\steam.exe"#)
.arg(&generate_steam_rungame(id))
.stdout(Stdio::null())
.stderr(Stdio::null())
.spawn()?
}
_ => panic!("Couldn't find steam!"),
};
Ok(child)
}
Expand All @@ -243,20 +244,20 @@ fn run(steam_type: SteamKind, id: &str) -> std::io::Result<Child> {
#[cfg(target_os = "macos")]
fn run(steam_type: SteamKind, id: &str) -> std::io::Result<Child> {
let child = match steam_type {
SteamKind::Vanilla => std::process::Command::new("steam")
.arg(&generate_steam_rungame(id))
.stdout(Stdio::null())
.stderr(Stdio::null())
.spawn()?,
SteamKind::NotFound => panic!("Couldn't find steam!"),
SteamKind::Vanilla => std::process::Command::new("steam")
.arg(&generate_steam_rungame(id))
.stdout(Stdio::null())
.stderr(Stdio::null())
.spawn()?,
SteamKind::NotFound => panic!("Couldn't find steam!"),
};
Ok(child)
}

/// Randomly picks an installed game from your Steam library and launches it.
#[derive(Parser)]
#[clap(
version = VERSION
version = VERSION
)]
struct Opts {
/// Show short message telling which game is being launched
Expand All @@ -273,25 +274,25 @@ fn main() {
let steam_type = detect_steam();

if steam_type == SteamKind::NotFound {
eprintln!("Couldn't find Steam. Please make sure it is installed.");
return;
eprintln!("Couldn't find Steam. Please make sure it is installed.");
return;
}

let mut path = {
let mut home = dirs::home_dir().unwrap();
match steam_type {
SteamKind::Flatpak => home.push(FLATPAK_APPLICATIONS_PATH),
#[cfg(target_os = "linux")]
SteamKind::Vanilla => home.push(VANILLA_APPLICATIONS_PATHS.iter().find(|&&p| {
let mut test_path = home.to_path_buf();
test_path.push(p);
test_path.exists() && test_path.is_dir()
}).unwrap()),
#[cfg(not(target_os = "linux"))]
SteamKind::Vanilla => home.push(VANILLA_APPLICATIONS_PATH),
_ => {}
}
home
let mut home = dirs::home_dir().unwrap();
match steam_type {
SteamKind::Flatpak => home.push(FLATPAK_APPLICATIONS_PATH),
#[cfg(target_os = "linux")]
SteamKind::Vanilla => home.push(VANILLA_APPLICATIONS_PATHS.iter().find(|&&p| {
let mut test_path = home.to_path_buf();
test_path.push(p);
test_path.exists() && test_path.is_dir()
}).unwrap()),
#[cfg(not(target_os = "linux"))]
SteamKind::Vanilla => home.push(VANILLA_APPLICATIONS_PATH),
_ => {}
}
home
};

path.push(MANIFEST_DIR);
Expand All @@ -301,19 +302,19 @@ fn main() {
let mut games = get_games_from_manifest_in_path(&path);

for other_dir in install_dirs {
let mut path = PathBuf::new();
path.push(other_dir);
path.push(MANIFEST_DIR);
games.extend(get_games_from_manifest_in_path(&path));
let mut path = PathBuf::new();
path.push(other_dir);
path.push(MANIFEST_DIR);
games.extend(get_games_from_manifest_in_path(&path));
}

let (game, id) = games.choose(&mut rand::thread_rng()).unwrap();

if opts.verbose > 0 {
println!("Randomly launching \"{}\"! Have fun!", game);
println!("Randomly launching \"{}\"! Have fun!", game);
}

if !opts.dry_run {
let _ = run(steam_type, id).unwrap();
let _ = run(steam_type, id).unwrap();
}
}

0 comments on commit c9e2582

Please sign in to comment.