cargo-skyline 1.16.0

A cargo subcommand for working with Skyline plugins written in Rust
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87

use owo_colors::OwoColorize;
use crate::game_paths::{get_subsdk_path, get_npdm_path, get_plugin_nro_path};
use crate::error::{Error, Result};
use crate::cargo_info;
use crate::build;
use std::io::{Read, Cursor, Write};
use std::result::Result as StdResult;
use zip::{ZipArchive, ZipWriter};
use std::fs;

pub struct Exefs {
    pub main_npdm: Vec<u8>,
    pub subsdk1: Vec<u8>,
}

// TODO: Cache exefs to disk, figure out some strategy for cache invalidation?
pub fn get_exefs(url: &str) -> Result<Exefs> {
    let zip_reader = Cursor::new(
        attohttpc::get(url).send()
            .map_err(|_| Error::DownloadError)?
            .bytes().map_err(|_| Error::DownloadError)?
    );

    let mut zip = ZipArchive::new(zip_reader).unwrap();

    let subsdk1 = zip.by_name("exefs/subsdk9")?.bytes().collect::<StdResult<_, _>>()?;
    let main_npdm = zip.by_name("exefs/main.npdm")?.bytes().collect::<StdResult<_, _>>()?;

    Ok(Exefs {
        main_npdm,
        subsdk1,
    })
}

pub fn package(skyline_url: &str, title_id: Option<&str>, out_path: &str) -> Result<()> {
    let args = vec![String::from("--release")];
    let nro_path = build::build_get_nro(args)?;
    let plugin_name = nro_path.file_name().unwrap().to_string_lossy();
    println!("Built {:?}!", plugin_name);

    let metadata = cargo_info::get_metadata()?;

    let title_id =
            title_id.or_else(|| metadata.title_id.as_deref())
                    .ok_or(Error::NoTitleId)?;

    println!("Downloading latest Skyline release...");
    let exefs = get_exefs(skyline_url)?;
    
    println!("Building Zip File...");
    let plugin_data = fs::read(&nro_path)?;
    
    let mut zip = ZipWriter::new(fs::File::create(out_path)?);

    zip.start_file(get_plugin_nro_path(title_id, plugin_name.as_ref()), Default::default())?;
    zip.write_all(&plugin_data)?;

    // main.npdm
    let main_npdm =
        metadata.npdm_path
            .as_ref()
            .map(|path| fs::read(path))
            .transpose()
            .map_err(|_| Error::NoNpdmFileFound)?;
    let generated_npdm = crate::installer::generate_npdm(&title_id);
    let main_npdm =
        main_npdm
            .as_ref()
            .unwrap_or_else(|| {
                eprintln!("\n{}: defaulting to a generated NPDM.", "Warning".yellow());
                eprintln!("{}: To specify a custom npdm add the following to your Cargo.toml:", "NOTE".bright_blue());
                eprintln!("\n{}\n", "[package.metadata.skyline]".bright_blue());
                eprintln!("{}\n", "custom-npdm = \"path/to/your.npdm\"".bright_blue());
                &generated_npdm
            });
    zip.start_file(get_npdm_path(title_id), Default::default())?;
    zip.write_all(main_npdm)?;

    // subsdk
    let subsdk_name = metadata.subsdk_name.as_deref().unwrap_or("subsdk9");
    zip.start_file(get_subsdk_path(title_id, subsdk_name), Default::default())?;
    zip.write_all(&exefs.subsdk1)?;

    println!("Finished building zip at '{}'", out_path);

    Ok(())
}