feat(modify): add unlock and lock ability for map files

Author: gynt

sourcehold/__main__.py

@@ -10,6 +10,7 @@
 
 from sourcehold.tool.argparsers.common import main_parser
 from sourcehold.tool.argparsers.services import services_parser, convert_parser
+from sourcehold.tool.modify.map import modify_map
 
 file_input_output = argparse.ArgumentParser(add_help=False)
 file_input_output.add_argument("--in", help="files or folders to (un)pack", nargs='+', required=True)
@@ -77,7 +78,10 @@
 from .tool.convert.aiv import convert_aiv
 
 def main():
-  convert_aiv(args)
+  if convert_aiv(args):
+      return
+  if modify_map(args):
+      return
 
   if args.service == "aiv":
     if args.method == "file":

sourcehold/tool/argparsers/services.py

@@ -11,3 +11,10 @@
 convert_aiv_parser.add_argument('--to-format', required=False, default='')
 
 memory_parser = services_parser.add_parser('memory')
+
+modify_parser = services_parser.add_parser('modify')
+modify_subparser = modify_parser.add_subparsers(dest='type', required=True, title='type')
+
+modify_map_parser = modify_subparser.add_parser('map', parents=[file_input_file_output])
+modify_map_parser.add_argument("--unlock", required=False, action='store_const', const=True)
+modify_map_parser.add_argument("--lock", required=False, action='store_const', const=True)

sourcehold/tool/modify/__init__.py

@@ -0,0 +1,2 @@
+
+

sourcehold/tool/modify/map/__init__.py

@@ -0,0 +1,29 @@
+import pathlib, sys
+
+from sourcehold.maps.Map import Map
+
+def modify_map(args):
+  if args.service != "modify" or args.type != "map":
+    return False
+  inp = args.input
+  if inp != '-' and not pathlib.Path(inp).exists():
+    raise Exception(f"file does not exist: {inp}")
+  m = Map().from_file(inp)
+  print(args)
+  if args.unlock:
+    if m.u3.map_locked:
+      print(f"map file was locked, is now unlocked", file=sys.stderr)
+      m.u3.map_locked = 0
+    else:
+      print(f"map file was already locked", file=sys.stderr)
+  if args.lock:
+    if m.u3.map_locked:
+      print(f"map file was already locked", file=sys.stderr)
+    else:
+      print(f"map file was unlocked, is now locked", file=sys.stderr)
+      m.u3.map_locked = 1
+  outp = args.output
+  if not outp:
+    outp = inp
+  m.to_file(outp)
+  return True