use super::*;
pub type Generator = trans_gen::gens::python::Generator;
#[derive(Debug, Serialize, Deserialize, Clone, PartialEq, Eq)]
pub enum Compiler {
CPython,
PyPy,
}
impl Default for Compiler {
fn default() -> Self {
Self::CPython
}
}
#[derive(Default, Debug, Serialize, Deserialize, Clone)]
pub struct GenOptions {
#[serde(default)]
compiler: Compiler,
#[serde(default)]
trans: <Generator as trans_gen::Generator>::Options,
}
impl<G: Game> ClientGen<G> for trans_gen::GeneratorImpl<Generator> {
const NAME: &'static str = "Python";
const RUNNABLE: bool = true;
type GenOptions = GenOptions;
fn gen(options: &Options, gen_options: Self::GenOptions) -> anyhow::Result<()> {
let mut gen = Self::new(options.name, options.version, gen_options.trans);
gen.add(&trans::Schema::of::<ClientMessage<G>>());
gen.add(&trans::Schema::of::<ServerMessage<G>>());
gen.add(&trans::Schema::of::<G::DebugState>());
gen.result().write_to(options.target_dir)?;
match gen_options.compiler {
Compiler::CPython => {
write_file(
options.target_dir.join("Dockerfile"),
project_file!(options, "cpython/Dockerfile"),
)?;
write_file(
options.target_dir.join("compile.sh"),
project_file!(options, "cpython/compile.sh"),
)?;
write_file(
options.target_dir.join("run.sh"),
project_file!(options, "cpython/run.sh"),
)?;
}
Compiler::PyPy => {
write_file(
options.target_dir.join("Dockerfile"),
project_file!(options, "pypy/Dockerfile"),
)?;
write_file(
options.target_dir.join("compile.sh"),
project_file!(options, "pypy/compile.sh"),
)?;
write_file(
options.target_dir.join("run.sh"),
project_file!(options, "pypy/run.sh"),
)?;
}
}
write_file(
options.target_dir.join("main.py"),
project_file!(options, "main.py"),
)?;
write_file(
options.target_dir.join("debug_interface.py"),
project_file!(options, "debug_interface.py"),
)?;
write_file(
options.target_dir.join("my_strategy.py"),
project_file!(options, "my_strategy.py"),
)?;
Ok(())
}
fn build_local(_options: &Options) -> anyhow::Result<()> {
Ok(())
}
fn run_local(options: &Options) -> anyhow::Result<Command> {
let mut command = command(if cfg!(windows) { "py -3" } else { "python3" });
command.arg("main.py");
command.current_dir(options.target_dir);
Ok(command)
}
}