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

use std::path::PathBuf;

use crate::Error;
use anyhow::Context;
use clap::Parser;

#[derive(Debug, Parser)]
pub struct Codegen {
    /// What to name the generated bindings.
    #[clap(short, long)]
    pub name: Option<String>,
    /// Where to save the generated bindings.
    #[clap(short, long)]
    pub out_dir: Option<PathBuf>,
    /// The Pirita file to read.
    pub input: PathBuf,
}

impl Codegen {
    pub fn run(self, language: Language) -> Result<(), Error> {
        let Codegen {
            name,
            out_dir,
            input,
        } = self;
        let pkg = crate::utils::load(&input)?;
        let options = wasmer_pack::BindingsOptions { name };
        let files = match language {
            Language::JavaScript => wasmer_pack::generate_javascript(&pkg, &options)?,
            Language::Python => wasmer_pack::generate_python(&pkg, &options)?,
        };

        let metadata = pkg.metadata();

        let out_dir = out_dir.unwrap_or_else(|| {
            // If no output directory was specified, let's save to something
            // like "namespace/name/"
            let pkg_name = &metadata.package_name;
            match pkg_name.namespace().as_str() {
                Some(ns) => PathBuf::from(ns).join(pkg_name.name()),
                None => PathBuf::from(pkg_name.name()),
            }
        });
        files
            .save_to_disk(&out_dir)
            .with_context(|| format!("Unable to save to \"{}\"", out_dir.display()))?;

        Ok(())
    }
}

#[derive(Debug, Copy, Clone, PartialEq, Eq, Hash)]
pub enum Language {
    JavaScript,
    Python,
}

impl Language {
    pub fn name(self) -> &'static str {
        match self {
            Language::JavaScript => "JavaScript",
            Language::Python => "Python",
        }
    }
}