AstralRinth/theseus_playground/src/main.rs

#![cfg_attr(
    all(not(debug_assertions), target_os = "windows"),
    windows_subsystem = "windows"
)]

use dunce::canonicalize;
use theseus::prelude::*;
use theseus::profile_create::profile_create;
use tokio::time::{sleep, Duration};

// A simple Rust implementation of the authentication run
// 1) call the authenticate_begin_flow() function to get the URL to open (like you would in the frontend)
// 2) open the URL in a browser
// 3) call the authenticate_await_complete_flow() function to get the credentials (like you would in the frontend)
pub async fn authenticate_run() -> theseus::Result<Credentials> {
    println!("A browser window will now open, follow the login flow there.");
    let url = auth::authenticate_begin_flow().await?;

    println!("URL {}", url.as_str());
    webbrowser::open(url.as_str())?;

    let credentials = auth::authenticate_await_complete_flow().await?;
    State::sync().await?;

    println!("Logged in user {}.", credentials.username);
    Ok(credentials)
}

#[tokio::main]
async fn main() -> theseus::Result<()> {
    println!("Starting.");

    // Initialize state
    let st = State::get().await?;
    st.settings.write().await.max_concurrent_downloads = 5;
    // Changed the settings, so need to reset the semaphore
    st.reset_semaphore().await;

    // Clear profiles
    println!("Clearing profiles.");
    {
        let h = profile::list().await?;
        for (path, _) in h.into_iter() {
            profile::remove(&path).await?;
        }
    }

    println!("Creating/adding profile.");

    let name = "Example".to_string();
    let game_version = "1.19.2".to_string();
    let modloader = ModLoader::Fabric;
    let loader_version = "stable".to_string();

    let profile_path = profile_create(
        name.clone(),
        game_version,
        modloader,
        Some(loader_version),
        None,
        None,
    )
    .await?;

    println!("Adding sodium");
    let sodium_path = profile::add_project_from_version(
        &profile_path,
        "rAfhHfow".to_string(),
    )
    .await?;

    let mod_menu_path = profile::add_project_from_version(
        &profile_path,
        "gSoPJyVn".to_string(),
    )
    .await?;

    println!("Disabling sodium");
    profile::toggle_disable_project(&profile_path, &sodium_path).await?;

    profile::remove_project(&profile_path, &mod_menu_path).await?;
    // let profile_path =
    //     pack::install_pack_from_version_id("KxUUUFh5".to_string())
    //         .await
    //         .unwrap();

    //  async closure for testing any desired edits
    // (ie: changing the java runtime of an added profile)
    println!("Editing.");
    profile::edit(&profile_path, |_profile| {
        // Eg: Java- this would let you change the java runtime of the profile instead of using the default
        // use theseus::prelude::jre::JAVA__KEY;
        // profile.java = Some(JavaSettings {
        // jre_key: Some(JAVA_17_KEY.to_string()),
        //     extra_arguments: None,
        // });
        async { Ok(()) }
    })
    .await?;
    State::sync().await?;

    // Attempt to run game
    if auth::users().await?.len() == 0 {
        println!("No users found, authenticating.");
        authenticate_run().await?; // could take credentials from here direct, but also deposited in state users
    }

    println!("running");
    // Run a profile, running minecraft and store the RwLock to the process
    let proc_lock = profile::run(&canonicalize(&profile_path)?).await?;

    let pid = proc_lock
        .read()
        .await
        .child
        .id()
        .expect("Could not get PID from process.");
    println!("Minecraft PID: {}", pid);

    // Wait 5 seconds
    println!("Waiting 5 seconds to gather logs...");
    sleep(Duration::from_secs(5)).await;
    let _stdout = process::get_stdout_by_pid(pid).await?;
    let _stderr = process::get_stderr_by_pid(pid).await?;
    // println!("Logs after 5sec <<< {stdout} >>> end stdout");

    println!(
        "All running process paths {:?}",
        process::get_all_running_profile_paths().await?
    );

    // hold the lock to the process until it ends
    println!("Waiting for process to end...");
    let mut proc = proc_lock.write().await;
    process::wait_for(&mut proc).await?;

    Ok(())
}