add set-level-letters, un/lock everything, un/buy everything

1 files changed, 158 insertions, 9 deletions

diff --git a/srb.py b/srb.py
index 5d33efe..1687d12 100755
--- a/srb.py
+++ b/srb.py
@@ -4,15 +4,12 @@
 # Purpose: frontend for srb_lib
 # History:
 # Usage:
+#    tl;dr: ./srb.py --profile 1 --unlock-everything --buy-everything "Profile 1.sav"
 # Reference:
 #    blog posts 2024-03
 #    delsum from github
 #    bgconf.py
-
-# WORKHERE:
-# or --reset-level 5 (which zeros out all info about completion of that level)
-# --set-level-letters 0,none 0,all
-# --set-level 0,done 5,new # sets balloons 10, status general, letters all
+# WORKHERE: still some confusion around the --unlock-everything --lock-everything and the levelset available levels.
 
 import srb_lib, argparse, sys
 from srb_lib import ferror, debuglev
@@ -46,8 +43,10 @@ parser.add_argument("--get-purchased-planes",action="store_true",help="Print cur
 parser.add_argument("--add-purchased-planes",action="append",help="For profile, add these purchased planes. Can be used multiple times.")
 parser.add_argument("--remove-purchased-planes",action="append",help="For profile, remove (un-buy) these purchased planes. Can be used multiple times.")
 parser.add_argument("--get-level",help="Print status for this level for profile.")
+parser.add_argument("--set-level",action="append",help="Set completion status, balloons, letters for this level for profile. Example value to pass: 0,all 15,none")
 parser.add_argument("--set-level-status",action="append",help="Set completion status for this level for profile. Example value to pass: \"0,general\"")
 parser.add_argument("--set-level-balloons",action="append",help="Set balloon status for this level for profile. Example value to pass: \"0,all\" or \"0,none\"")
+parser.add_argument("--set-level-letters",action="append",help="Set collected letters for level for profile. Examples: 0,all 15,none")
 parser.add_argument("--get-levelset",help="Print status for this levelset for profile.")
 parser.add_argument("--set-levelset-available-levels",action="append",help="Set number of available levels in this levelset for profile.")
 parser.add_argument("--get-name",action="store_true",help="Print name for profile.")
@@ -60,7 +59,11 @@ parser.add_argument("--set-health",type=int,choices=range(1,5),help="Set stat he
 parser.add_argument("--get-stunt",action="store_true",help="Print stat stunt for profile.")
 parser.add_argument("--set-stunt",type=int,choices=range(1,5),help="Set stat stunt for profile.")
 parser.add_argument("--get-gun",action="store_true",help="Print stat gun for profile.")
-parser.add_argument("--set-gun",type=int,choices=range(1,6),help="Set stat gun for profile.")
+parser.add_argument("--set-gun",type=int,choices=range(1,6),help="Set stat gun for profile. 5 is octo-gun which normally unlocks at 100% completion of the game.")
+parser.add_argument("--lock-everything",action="store_true",help="Set all levels to completed with rank general, all letters, and all balloons.")
+parser.add_argument("--unlock-everything",action="store_true",help="Set all levels to not completed. Take all letters and balloons.")
+parser.add_argument("--buy-everything",action="store_true",help="Give all purchasable items: planes, weapons, plane stats.")
+parser.add_argument("--unbuy-everything",action="store_true",help="Remove all purchasable items: planes, weapons, plane stats.")
 parser.add_argument("--checksum",action=argparse.BooleanOptionalAction,default=True,help="Correct checksum. Default is to do this. It happens at the end of everything else.")
 parser.add_argument("file",default="Profile 1.sav")
 args = parser.parse_args()
@@ -78,8 +81,8 @@ profile_id = args.profile
 
 #print(f"profile_id={profile_id}")
 
-# WORKHERE: new actions that need --profile must be added here.
-if not profile_id and (args.get_money or args.set_money or args.get_weapon or args.set_weapon or args.get_level or args.get_name or args.set_name or args.get_profile_in_use or args.get_purchased_weapons or args.get_tutorial_completed or args.add_purchased_weapons or args.remove_purchased_weapons or args.get_health or args.get_stunt or args.get_gun or args.set_health or args.set_stunt or args.set_gun or args.get_levelset or args.get_purchased_planes or args.add_purchased_planes or args.remove_purchased_planes or args.set_level_status or args.set_level_balloons or args.set_levelset_available_levels):
+# New actions that need --profile must be added here.
+if not profile_id and (args.get_money or args.set_money or args.get_weapon or args.set_weapon or args.get_level or args.get_name or args.set_name or args.get_profile_in_use or args.get_purchased_weapons or args.get_tutorial_completed or args.add_purchased_weapons or args.remove_purchased_weapons or args.get_health or args.get_stunt or args.get_gun or args.set_health or args.set_stunt or args.set_gun or args.get_levelset or args.get_purchased_planes or args.add_purchased_planes or args.remove_purchased_planes or args.set_level_status or args.set_level_balloons or args.set_levelset_available_levels or args.set_level_letters or args.unlock_everything or args.lock_everything or args.buy_everything or args.unbuy_everything or args.set_level):
    ferror("Warning: Cannot perform most actions without --profile. Not all tasks may run.")
 else:
    if args.get_money:
@@ -225,13 +228,159 @@ else:
          try:
             levelset_num, count = l.split(",")
          except:
-            ferror(f"Warning! Skipping un-parseable level completion spec {l}. Please use \"0,general\" format.")
+            ferror(f"Warning! Skipping un-parseable levelset available spec {l}. Please use \"0,all\" format.")
          if levelset_num != -1:
             data, message = srb_lib.set_levelset_available_levels(args.file,profile_id,levelset_num,count)
             if (type(data) == int and data == -1) or message != "":
                ferror(f"Failed to set profile {profile_id} level {levelset_num} available levels to {count} because {message}")
             else:
                srb_lib.write_file(args.file,0,data)
+   if args.set_level_letters:
+      for l in args.set_level_letters:
+         level_num = -1
+         try:
+            level_num, count = l.split(",")
+         except:
+            ferror(f"Warning! Skipping unparseable level letter spec {l}. Please use \"0,all\" format.")
+         if level_num != -1:
+            data, message = srb_lib.set_level_letters(args.file,profile_id,level_num, count)
+            if (type(data) == int and data == -1) or message != "":
+               ferror(f"Failed to set profile {profile_id} level {levelset_num} available levels to {count} because {message}")
+            else:
+               #print(f"good?")
+               srb_lib.write_file(args.file,0,data)
+   if args.unlock_everything:
+      data = srb_lib._get_data_from_data_object(args.file)
+      # counting backwards helps the levelset available levels count.
+      for i in range(len(srb_lib.LEVELS)-1,0,-1):
+         data, new_status, message = srb_lib.set_level_status(data,profile_id,i,"general")
+         if message != "":
+            ferror(f"Failed on loop {i}, set_level_status, {message}")
+            break
+         data, message = srb_lib.set_level_balloons(data,profile_id,i,"all")
+         if message != "":
+            ferror(f"Failed on loop {i}, set_level_balloons, {message}")
+            break
+         data, message = srb_lib.set_level_letters(data,profile_id,i,"all")
+         if message != "":
+            ferror(f"Failed on loop {i}, set_level_letters, {message}")
+            break
+      for i in range(len(srb_lib.LEVELSETS)-1,0,-1):
+         data, message = srb_lib.set_levelset_available_levels(data,profile_id,i,"all")
+         if message != "":
+            ferror(f"Failed on levelset loop {i}, {message}")
+            break
+      if (type(data) == int and data == -1) or message != "":
+         ferror(f"Failed to unlock everything as indicated above.")
+      else:
+         srb_lib.write_file(args.file,0,data)
+   if args.lock_everything:
+      data = srb_lib._get_data_from_data_object(args.file)
+      # counting backwards helps the levelset available levels count.
+      for i in range(len(srb_lib.LEVELS)-1,0,-1):
+         data, new_status, message = srb_lib.set_level_status(data,profile_id,i,"none")
+         if message != "":
+            ferror(f"Failed on loop {i}, set_level_status, {message}")
+            break
+         data, message = srb_lib.set_level_balloons(data,profile_id,i,"none")
+         if message != "":
+            ferror(f"Failed on loop {i}, set_level_balloons, {message}")
+            break
+         data, message = srb_lib.set_level_letters(data,profile_id,i,"none")
+         if message != "":
+            ferror(f"Failed on loop {i}, set_level_letters, {message}")
+            break
+      for i in range(len(srb_lib.LEVELSETS)-1,0,-1):
+         data, message = srb_lib.set_levelset_available_levels(data,profile_id,i,"none")
+         if message != "":
+            ferror(f"Failed on levelset loop {i}, {message}")
+            break
+      if (type(data) == int and data == -1) or message != "":
+         ferror(f"Failed to lock everything as indicated above.")
+      else:
+         srb_lib.write_file(args.file,0,data)
+   if args.buy_everything:
+      data, message = srb_lib.set_purchased_weapons(args.file,profile_id,"add",["all"])
+      if (type(data) == int and data == -1) or message != "":
+         ferror(f"Failed to remove purchased weapons ['all'] because {message}")
+      else:
+         data, message = srb_lib.set_purchased_planes(data,profile_id,"add",["all"])
+         if (type(data) == int and data == -1) or message != "":
+            ferror(f"Failed to add purchased planes ['all'] because {message}")
+         else:
+            data, message = srb_lib.set_plane_stat(data,profile_id,"health",4)
+            if data == -1 or message != "":
+               print(f"Failed to set health to 4 because {message}")
+            else:
+               data, message = srb_lib.set_plane_stat(data,profile_id,"stunt",4)
+               if data == -1 or message != "":
+                  print(f"Failed to set stunt to 4 because {message}")
+               else:
+                  data, message = srb_lib.set_plane_stat(data,profile_id,"gun",5)
+                  if data == -1 or message != "":
+                     print(f"Failed to set gun to 5 because {message}")
+      #   srb_lib.write_file(args.file,0,data)
+      if (type(data) == int and data == -1) or message != "":
+         ferror(f"Failed to buy everything as indicated above.")
+      else:
+         srb_lib.write_file(args.file,0,data)
+   if args.unbuy_everything:
+      data, message = srb_lib.set_purchased_weapons(args.file,profile_id,"remove",["all"])
+      if (type(data) == int and data == -1) or message != "":
+         ferror(f"Failed to remove purchased weapons ['all'] because {message}")
+      else:
+         data, message = srb_lib.set_purchased_planes(data,profile_id,"remove",["all"])
+         if (type(data) == int and data == -1) or message != "":
+            ferror(f"Failed to remove purchased planes ['all'] because {message}")
+         else:
+            data, message = srb_lib.set_plane_stat(data,profile_id,"health",1)
+            if data == -1 or message != "":
+               print(f"Failed to set health to 1 because {message}")
+            else:
+               data, message = srb_lib.set_plane_stat(data,profile_id,"stunt",1)
+               if data == -1 or message != "":
+                  print(f"Failed to set stunt to 1 because {message}")
+               else:
+                  data, message = srb_lib.set_plane_stat(data,profile_id,"gun",1)
+                  if data == -1 or message != "":
+                     print(f"Failed to set gun to 1 because {message}")
+      #   srb_lib.write_file(args.file,0,data)
+      if (type(data) == int and data == -1) or message != "":
+         ferror(f"Failed to unbuy everything as indicated above.")
+      else:
+         srb_lib.write_file(args.file,0,data)
+   if args.set_level:
+      for i in args.set_level:
+         level_num = -1
+         status = ""
+         cont = True
+         try:
+            level_num, status = i.split(",")
+         except:
+            ferror(f"Warning! Unable to set level spec {i}. Use \"0,all\" or \"15,none\" syntax.")
+            cont = False
+         if status not in ["all","none"]:
+            ferror(f"Warning! Unable to set level {level_num} to {status}. Use all or none.")
+            cont = False
+         if level_num != -1:
+            data, new_status, message = srb_lib.set_level_status(args.file,profile_id,level_num,"general" if status == "all" else "none")
+            if message != "":
+               ferror(f"Failed on {i}, set_level_status, {message}")
+               cont = False
+         if cont:
+            data, message = srb_lib.set_level_balloons(data,profile_id,level_num,status)
+            if message != "":
+               ferror(f"Failed on {i}, set_level_balloons, {message}")
+               cont = False
+         if cont:
+            data, message = srb_lib.set_level_letters(data,profile_id,level_num,status)
+            if message != "":
+               ferror(f"Failed on {i}, set_level_letters, {message}")
+               cont = False
+         if (type(data) == int and data == -1) or message != "" or not cont:
+            ferror(f"Failed to set profile {profile_id} level {level_num} status to {completion} because {message}")
+         else:
+            srb_lib.write_file(args.file,0,data)
 
 if args.checksum:
    f = args.file