221 lines
6.6 KiB
Python
221 lines
6.6 KiB
Python
|
import sys
|
||
|
|
import os
|
||
|
|
import tomllib
|
||
|
|
import subprocess
|
||
|
|
import platform
|
||
|
|
|
||
|
|
# Global Params
|
||
|
|
param1 = ""
|
||
|
|
param2 = ""
|
||
|
|
fx_mode=False
|
||
|
|
project_list=[]
|
||
|
|
parameter_validation_check=False
|
||
|
|
project=[]
|
||
|
|
|
||
|
|
# Functions
|
||
|
|
def get_args():
|
||
|
|
global param1
|
||
|
|
global param2
|
||
|
|
global fx_mode
|
||
|
|
if len(sys.argv)==1:
|
||
|
|
help()
|
||
|
|
if len(sys.argv)>1:
|
||
|
|
param1 = sys.argv[1]
|
||
|
|
if len(sys.argv)>2:
|
||
|
|
param2 = sys.argv[2]
|
||
|
|
fx_mode=True
|
||
|
|
|
||
|
|
def help():
|
||
|
|
print("Usage build and upload Project:")
|
||
|
|
print(" run.py list Get a list of all Projects")
|
||
|
|
print(" run.py new <Project-Name> Create a new game in the Project folder")
|
||
|
|
print(" run.py <Project-Name> For uploading a game")
|
||
|
|
print("")
|
||
|
|
print("Usage FX-Data build and upload:")
|
||
|
|
print(" run.py fxbuild <Project-Name> Build your fxdata")
|
||
|
|
print(" run.py fxupload <Project-Name> Upload your fxdata")
|
||
|
|
print(" run.py fxall <Project-Name> Build and Upload your fxdata")
|
||
|
|
print(" and the game in one step")
|
||
|
|
sys.exit()
|
||
|
|
|
||
|
|
def get_projects():
|
||
|
|
global project_list
|
||
|
|
with open("Cargo.toml", "rb") as f:
|
||
|
|
data = tomllib.load(f)
|
||
|
|
path=data["workspace"]["members"]
|
||
|
|
for p in path:
|
||
|
|
name = p.split("/")[-1]
|
||
|
|
if name =="arduboy-rust":
|
||
|
|
continue
|
||
|
|
project=[name,p]
|
||
|
|
project_list.append(project)
|
||
|
|
|
||
|
|
def validate_args():
|
||
|
|
global parameter_validation_check
|
||
|
|
global project
|
||
|
|
if fx_mode:
|
||
|
|
for p in project_list:
|
||
|
|
if param2 in p:
|
||
|
|
parameter_validation_check=True
|
||
|
|
project = p
|
||
|
|
|
||
|
|
else:
|
||
|
|
for p in project_list:
|
||
|
|
if param1 in p:
|
||
|
|
parameter_validation_check=True
|
||
|
|
project = p
|
||
|
|
if not parameter_validation_check:
|
||
|
|
help()
|
||
|
|
|
||
|
|
def execute_options():
|
||
|
|
if param1=="list":
|
||
|
|
group=""
|
||
|
|
for p in project_list:
|
||
|
|
if not group==p[1].split("/")[-2]:
|
||
|
|
group=p[1].split("/")[-2]
|
||
|
|
print(f"\n{group}:")
|
||
|
|
print(f" {p[0]}")
|
||
|
|
print("")
|
||
|
|
exit()
|
||
|
|
if param1=="new":
|
||
|
|
create_new_project()
|
||
|
|
exit()
|
||
|
|
|
||
|
|
def _dumps_value(value):
|
||
|
|
if isinstance(value, bool):
|
||
|
|
return "true" if value else "false"
|
||
|
|
elif isinstance(value, (int, float)):
|
||
|
|
return str(value)
|
||
|
|
elif isinstance(value, str):
|
||
|
|
return f'"{value}"'
|
||
|
|
elif isinstance(value, list):
|
||
|
|
return f"[{', '.join(_dumps_value(v) for v in value)}]"
|
||
|
|
else:
|
||
|
|
raise TypeError(f"{type(value).__name__} {value!r} is not supported")
|
||
|
|
|
||
|
|
def dumps(toml_dict, table=""):
|
||
|
|
toml = []
|
||
|
|
for key, value in toml_dict.items():
|
||
|
|
if isinstance(value, dict):
|
||
|
|
table_key = f"{table}.{key}" if table else key
|
||
|
|
toml.append(f"\n[{table_key}]\n{dumps(value, table_key)}")
|
||
|
|
else:
|
||
|
|
toml.append(f"{key} = {_dumps_value(value)}")
|
||
|
|
return "\n".join(toml)
|
||
|
|
|
||
|
|
def create_new_project():
|
||
|
|
project_name = param2.replace("-","_")
|
||
|
|
for p in project_list:
|
||
|
|
if p[0]==project_name:
|
||
|
|
sys.exit()
|
||
|
|
error_code=os.system(f'cargo new --vcs=none --lib ./Project/{project_name}')
|
||
|
|
if error_code>0:
|
||
|
|
sys.exit()
|
||
|
|
# Edit main Cargo.toml
|
||
|
|
with open("Cargo.toml", "rb") as f:
|
||
|
|
data = tomllib.load(f)
|
||
|
|
data["workspace"]["members"].append(f"Project/{project_name}")
|
||
|
|
with open("Cargo.toml","w") as f:
|
||
|
|
f.write(dumps(data))
|
||
|
|
# Edit new project Cargo.toml
|
||
|
|
with open(f"Project/{project_name}/Cargo.toml", "rb") as f:
|
||
|
|
data = tomllib.load(f)
|
||
|
|
newdata={}
|
||
|
|
newdata["package"]=data["package"]
|
||
|
|
newdata["lib"]={'crate-type':["staticlib"]}
|
||
|
|
newdata["dependencies"]={"arduboy-rust":{"path":"../../arduboy-rust"}}
|
||
|
|
with open(f"Project/{project_name}/Cargo.toml","w") as f:
|
||
|
|
f.write(dumps(newdata))
|
||
|
|
# Edit lib.rs
|
||
|
|
with open(f"Project/{project_name}/src/lib.rs","w") as f:
|
||
|
|
f.write('''#![no_std]
|
||
|
|
#![allow(non_upper_case_globals)]
|
||
|
|
|
||
|
|
//Include the Arduboy Library
|
||
|
|
#[allow(unused_imports)]
|
||
|
|
use arduboy_rust::prelude::*;
|
||
|
|
|
||
|
|
#[allow(dead_code)]
|
||
|
|
const arduboy: Arduboy2 = Arduboy2::new();
|
||
|
|
|
||
|
|
// Progmem data
|
||
|
|
|
||
|
|
// dynamic ram variables
|
||
|
|
|
||
|
|
// The setup() function runs once when you turn your Arduboy on
|
||
|
|
#[no_mangle]
|
||
|
|
pub unsafe extern "C" fn setup() {
|
||
|
|
// put your setup code here, to run once:
|
||
|
|
}
|
||
|
|
|
||
|
|
// The loop() function repeats forever after setup() is done
|
||
|
|
#[no_mangle]
|
||
|
|
#[export_name = "loop"]
|
||
|
|
pub unsafe extern "C" fn loop_() {
|
||
|
|
// put your main code here, to run repeatedly:
|
||
|
|
}''')
|
||
|
|
|
||
|
|
def generate_import_h():
|
||
|
|
with open(f"{project[1]}/config.toml", "rb") as f:
|
||
|
|
data = tomllib.load(f)
|
||
|
|
strings=["#pragma once\n"]
|
||
|
|
for lib in data["Libraries"]:
|
||
|
|
strings.append(f"#define {lib}_Library ;\n")
|
||
|
|
f = open("arduboy-rust/Wrapper-Project/src/imports.h", "w")
|
||
|
|
f.writelines(strings)
|
||
|
|
f.close()
|
||
|
|
|
||
|
|
def upload_to_arduboy():
|
||
|
|
if not fx_mode:
|
||
|
|
game_name=param1
|
||
|
|
else:
|
||
|
|
game_name=param2
|
||
|
|
if platform.system()=="Linux":
|
||
|
|
cmd=f"cargo build -p {game_name} --release; cp ./target/arduboy/release/lib{game_name}.a ./arduboy-rust/Wrapper-Project/lib/libgame.a; cd arduboy-rust/Wrapper-Project/; pio run -v -t upload; cp ./.pio/build/arduboy/firmware.hex ./build/{game_name}.hex; pio run -t clean; rm ./lib/libgame.a; cd ../../"
|
||
|
|
process = subprocess.Popen(cmd, stdout=subprocess.PIPE)
|
||
|
|
|
||
|
|
elif platform.system()=="Windows":
|
||
|
|
cmd=f'cargo build -p {game_name} --release; cp ./target/arduboy/release/lib{game_name}.a ./arduboy-rust/Wrapper-Project/lib/libgame.a; cd arduboy-rust/Wrapper-Project/; pio run -v -t upload; cp ./.pio/build/arduboy/firmware.hex ./build/{game_name}.hex; pio run -t clean; rm lib/libgame.a; cd ../../'
|
||
|
|
process = subprocess.Popen(["powershell", "-Command", cmd], stdout=subprocess.PIPE)
|
||
|
|
else:
|
||
|
|
sys.exit()
|
||
|
|
for c in iter(lambda: process.stdout.read(1),b""):
|
||
|
|
sys.stdout.buffer.write(c)
|
||
|
|
if process.returncode != 0:
|
||
|
|
sys.exit()
|
||
|
|
|
||
|
|
def fx_build():
|
||
|
|
error_code=os.system(f'python ./Tools/Arduboy-Python-Utilities/fxdata-build.py ./{project[1]}/fxdata/fxdata.txt')
|
||
|
|
if error_code>0:
|
||
|
|
sys.exit()
|
||
|
|
|
||
|
|
def fx_upload():
|
||
|
|
error_code=os.system(f'python ./Tools/Arduboy-Python-Utilities/fxdata-upload.py ./{project[1]}/fxdata/fxdata.bin')
|
||
|
|
if error_code>0:
|
||
|
|
sys.exit()
|
||
|
|
|
||
|
|
def fx_commands():
|
||
|
|
print("")
|
||
|
|
if param1=="fxbuild":
|
||
|
|
fx_build()
|
||
|
|
elif param1=="fxupload":
|
||
|
|
fx_upload()
|
||
|
|
elif param1=="fxall":
|
||
|
|
fx_build()
|
||
|
|
fx_upload()
|
||
|
|
upload_to_arduboy()
|
||
|
|
|
||
|
|
|
||
|
|
# Execute Script
|
||
|
|
get_args()
|
||
|
|
get_projects()
|
||
|
|
execute_options()
|
||
|
|
validate_args()
|
||
|
|
|
||
|
|
generate_import_h()
|
||
|
|
|
||
|
|
if fx_mode:
|
||
|
|
fx_commands()
|
||
|
|
else:
|
||
|
|
upload_to_arduboy()
|