From 79bcdeadc20e550ca554eecf9c5c012d608ed2c7 Mon Sep 17 00:00:00 2001
From: ShadowOne333 <shadow_one666@hotmail.com>
Date: Fri, 19 Apr 2024 14:26:04 -0600
Subject: [PATCH] Add CCScript address as 'compile' argument for CLI version

---
 coilsnake/ui/cli.py    | 33 ++++++++++++++++++---------------
 coilsnake/ui/common.py |  2 +-
 2 files changed, 19 insertions(+), 16 deletions(-)

diff --git a/coilsnake/ui/cli.py b/coilsnake/ui/cli.py
index 1c109480..169ac133 100755
--- a/coilsnake/ui/cli.py
+++ b/coilsnake/ui/cli.py
@@ -21,6 +21,7 @@ def main():
     compile_parser.add_argument("project_directory")
     compile_parser.add_argument("base_rom")
     compile_parser.add_argument("output_rom")
+    compile_parser.add_argument("ccscript_offset", nargs='?')
     compile_parser.set_defaults(func=_compile)
 
     decompile_parser = subparsers.add_parser("decompile", help="decompile from rom to project")
@@ -33,23 +34,23 @@ def main():
     upgrade_parser.add_argument("base_rom")
     upgrade_parser.add_argument("project_directory")
     upgrade_parser.set_defaults(func=_upgrade)
-    
+
     decomp_script_parser = subparsers.add_parser("scriptdump", help="Decompile a ROM's script to an already existing project.")
-    
+
     decomp_script_parser.add_argument("rom_filename")
     decomp_script_parser.add_argument("project_directory")
     decomp_script_parser.set_defaults(func=_scriptdump)
-    
+
     patch_rom_parser = subparsers.add_parser("patchrom", help="Apply an EBP or IPS patch to a ROM (for headered give true or false)")
-    
+
     patch_rom_parser.add_argument("clean_rom")
     patch_rom_parser.add_argument("output_rom")
     patch_rom_parser.add_argument("patch")
     patch_rom_parser.add_argument("headered")
     patch_rom_parser.set_defaults(func=_patchrom)
-    
+
     createpatch_parser = subparsers.add_parser("createpatch", help="Create a patch from a clean ROM and a hacked ROM.")
-    
+
     createpatch_parser.add_argument("clean_rom")
     createpatch_parser.add_argument("hacked_rom")
     createpatch_parser.add_argument("output_path")
@@ -57,20 +58,20 @@ def main():
     createpatch_parser.add_argument("description", nargs='?')
     createpatch_parser.add_argument("title", nargs='?')
     createpatch_parser.set_defaults(func=_createpatch)
-    
+
     expand_parser = subparsers.add_parser("expand", help="Expand a ROM's size to 32 MBits (4MB) or 48 MBits (6MB). exhi is false for 4MB, and true for 6MB.")
-    
+
     expand_parser.add_argument("rom")
     expand_parser.add_argument("exhi")
     expand_parser.set_defaults(func=_expand)
-    
+
     addheader_parser = subparsers.add_parser("addheader", help="Add a header to a ROM.")
-    
+
     addheader_parser.add_argument("rom")
     addheader_parser.set_defaults(func=_addheader)
-    
+
     stripheader_parser = subparsers.add_parser("stripheader", help="Remove a header from a ROM.")
-    
+
     stripheader_parser.add_argument("rom")
     stripheader_parser.set_defaults(func=_stripheader)
 
@@ -85,10 +86,12 @@ def main():
 
 
 def _compile(args):
+    if args.ccscript_offset == None:
+        args.ccscript_offset = 'F10000'
     compile_project(project_path=args.project_directory,
                     base_rom_filename=args.base_rom,
-                    output_rom_filename=args.output_rom)
-
+                    output_rom_filename=args.output_rom,
+                    ccscript_offset=args.ccscript_offset)
 
 def _decompile(args):
     decompile_rom(rom_filename=args.rom,
@@ -100,7 +103,7 @@ def _upgrade(args):
                     project_path=args.project_directory)
 
 def _scriptdump(args):
-    decompile_script(rom_filename=args.rom_filename, 
+    decompile_script(rom_filename=args.rom_filename,
                      project_path=args.project_directory)
 
 def _patchrom(args):
diff --git a/coilsnake/ui/common.py b/coilsnake/ui/common.py
index efc0aa3f..57ce8f39 100644
--- a/coilsnake/ui/common.py
+++ b/coilsnake/ui/common.py
@@ -93,7 +93,7 @@ def upgrade_project(project_path, base_rom_filename, progress_bar=None):
     log.info("Upgraded {} in {:.2f}s".format(project_path, time.time() - upgrade_start_time))
 
 
-def compile_project(project_path, base_rom_filename, output_rom_filename, ccscript_offset=None, progress_bar=None):
+def compile_project(project_path, base_rom_filename, output_rom_filename, ccscript_offset, progress_bar=None):
     if not os.path.isdir(project_path):
         raise RuntimeError("Project directory \"" + project_path + "\" is not a directory.")
     if not os.path.isfile(base_rom_filename):