add continue prompt in inject_ucode.lua

glankk · Oct 31, 2018 · c906c29 · c906c29
1 parent 42b3b2c
commit c906c29
Showing 1 changed file with 20 additions and 5 deletions.
diff --git a/patch/lua/inject_ucode.lua b/patch/lua/inject_ucode.lua
@@ -26,9 +26,23 @@ local data_loc =
 }
 
 local arg = {...}
+local opt_y = false
+if arg[1] == "-y" then
+  opt_y = true
+  table.remove(arg, 1)
+end
+
+function quit(code)
+  if not opt_y then
+    print("press enter to continue")
+    io.read()
+  end
+  os.exit(code)
+end
+
 if #arg < 2 then
   print("usage: `inject_ucode <dst-file> <src-file>`")
-  return
+  quit(1)
 end
 local dst = gru.n64rom_load(arg[1])
 local src = gru.n64rom_load(arg[2])
@@ -52,7 +66,7 @@ if text_insert then
   print(string.format("found text insertion point at 0x%08X in `%s`", text_insert, arg[1]))
 else
   print("unable find text insertion point")
-  return 1
+  quit(1)
 end
 
 -- find data insertion point
@@ -63,7 +77,7 @@ if data_insert then
   print(string.format("found data insertion point at 0x%08X in `%s`", data_insert, arg[1]))
 else
   print("unable find data insertion point")
-  return 1
+  quit(1)
 end
 
 -- include destination locations as search locations in source
@@ -92,7 +106,7 @@ for i = 1,#text_loc do
 end
 if not text then
   print("unable find text chunk")
-  return 1
+  quit(1)
 end
 
 -- find data
@@ -112,7 +126,7 @@ for i = 1,#data_loc do
 end
 if not data then
   print("unable find data chunk")
-  return 1
+  quit(1)
 end
 
 -- insert microcode
@@ -125,3 +139,4 @@ print("saving rom to `" .. arg[1] .. "`")
 dst:save(arg[1])
 
 print("done")
+quit(0)