87 lines
2.4 KiB
Rust
87 lines
2.4 KiB
Rust
// Copyright (C) 2021 "Maxime “pep” Buquet <pep@bouah.net>"
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify it
|
|
// under the terms of the GNU Affero General Public License as published by the
|
|
// Free Software Foundation, either version 3 of the License, or (at your
|
|
// option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful, but WITHOUT
|
|
// ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
// FITNESS FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License
|
|
// for more details.
|
|
//
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
use std::env;
|
|
use std::fmt;
|
|
use std::fs::{self, File};
|
|
use std::io::{BufWriter, Write};
|
|
use std::error::Error as StdError;
|
|
use std::path::Path;
|
|
|
|
use quote::{format_ident, quote};
|
|
|
|
#[derive(Debug, Eq, PartialEq)]
|
|
pub enum Error {
|
|
EnvVariableError(env::VarError),
|
|
ROMDoesntExist,
|
|
UnableToOpenROM,
|
|
}
|
|
|
|
impl StdError for Error {}
|
|
|
|
impl fmt::Display for Error {
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> Result<(), fmt::Error> {
|
|
match self {
|
|
Error::EnvVariableError(e) => write!(f, "POKEMON_ROM env var error: {}", e),
|
|
Error::ROMDoesntExist => write!(f, "ROM doesn't exist"),
|
|
Error::UnableToOpenROM => write!(f, "Unable to open ROM"),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl From<env::VarError> for Error {
|
|
fn from(err: env::VarError) -> Error {
|
|
Error::EnvVariableError(err)
|
|
}
|
|
}
|
|
|
|
fn get_rom() -> Result<Vec<u8>, Error> {
|
|
let env_romp = env::var("POKEMON_ROM")?;
|
|
let romp = Path::new(&env_romp);
|
|
|
|
if ! romp.exists() {
|
|
return Err(Error::ROMDoesntExist);
|
|
}
|
|
|
|
Ok(match fs::read(romp) {
|
|
Ok(v) => v,
|
|
Err(_) => return Err(Error::UnableToOpenROM),
|
|
})
|
|
}
|
|
|
|
fn main() -> Result<(), Error> {
|
|
let output = Path::new(&env::var("OUT_DIR").unwrap()).join("output.rs");
|
|
let mut output = BufWriter::new(File::create(&output).unwrap());
|
|
let _rom: Vec<u8> = get_rom()?;
|
|
|
|
let pkmn_names: Vec<_> = [
|
|
"Rhydon",
|
|
"Kangashkan",
|
|
"NidoranM",
|
|
].iter()
|
|
.map(|n| format_ident!("{}", n))
|
|
.collect();
|
|
|
|
let tokens = quote! {
|
|
/// Generated file containing various resources extracted from the provided ROM.
|
|
|
|
pub enum PokemonNames {
|
|
#(#pkmn_names),*
|
|
}
|
|
};
|
|
|
|
write!(output, "{}", tokens).unwrap();
|
|
Ok(())
|
|
}
|