From 00bda02b371c6214873486505b64785f15bb2411 Mon Sep 17 00:00:00 2001
From: "google-labs-jules[bot]"
 <161369871+google-labs-jules[bot]@users.noreply.github.com>
Date: Sun, 10 Aug 2025 13:40:15 +0000
Subject: [PATCH] feat: Add 'Run on Startup' checkbox to system tray

This commit introduces a new feature that allows you to configure the application to run on startup.

- A "Run on Startup" checkbox has been added to the system tray menu.
- The application now uses a `config.json` file to store settings, including the "run on startup" state.
- On Windows, the application adds/removes itself from the startup programs by creating/deleting a key in the Windows Registry.
---
 src/__pycache__/config.cpython-312.pyc  | Bin 0 -> 3891 bytes
 src/__pycache__/startup.cpython-312.pyc | Bin 0 -> 2620 bytes
 src/config.py                           |  62 ++++++++++++++++++------
 src/main.pyw                            |  31 ++++++++++--
 src/startup.py                          |  48 ++++++++++++++++++
 5 files changed, 120 insertions(+), 21 deletions(-)
 create mode 100644 src/__pycache__/config.cpython-312.pyc
 create mode 100644 src/__pycache__/startup.cpython-312.pyc
 create mode 100644 src/startup.py

diff --git a/src/__pycache__/config.cpython-312.pyc b/src/__pycache__/config.cpython-312.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..5dd0e834f698ac380a05d1a389ef6f2dda035abc
GIT binary patch
literal 3891
zcmbVPO>7&-6`m!R|34xri?U2x8b_8aTDBB5jh(oTomRGN8Ls@J1ce2(?3y!@NRzwl
z?9ws`Dshnll@bGD8W)zE6j4rTY8Noji;pebOVJAz(J-~vAwhG=rEu*426XD1T`nov
zu2FP|ot=3z@6DSx-}`3yk3hhSp!{uVaeCH=&_8*{Dz19t@y~4tT}K2FA%iB&w=f|X
zecOc1c-trJ#5N%k`-Fptzd{pE;(*#koKU-|hqx~GAygKa&xWY;X5H|F7kWP8rrvai
zwP^EAJj8nwR_5hn^n2S)!E6i2(DMBkE&mI(0wj2=#j<Kx3RvExcbN7!*47vaasSyz
z9=^6c-MMj&*(Twq{EKYr+tT<sI<2?1ShFU=#;)3E_?f*vf4!)sY3_owTm2X9V<JK$
zFB-4!GfKnMOFW0bW$?;Hy_W5I?6)@ivgvn6Q*!k;#!j|1B}PBCsUINQoBe`N5aK(&
z<pCNi`jUz~nVK3%tBPFotJ4b8l6h@FyR0QQM$iczJ3Np71+ch|w8mJv4w;4k8fnMI
zp4ry81*2Gws8@+VY{dSFZTA1M=YHBA`-$eWWLjgCh!*@_MYBRlNC&Jr&9?=+bW2la
zwXCsOTgk>8C0hv*2R|s{<Uh-{fV6tL!0tnqZu8yr$J#w;v|8gO)RdOcINGp}M$|U%
zLu<2i(^0F{e3t~vUeflopW~<-4Ix#Su?grr65x0BXP52g(Two4yx=tDaaZwAUJT`w
zA`um9(-fPS$umhyDKeg9O13UCU^HON9-TC%PNp&x6NResZVl7RDOKBxQyNY(Drr>Z
z&6$)ojR~EU@)-?Zk}`R!)-7mQ%BTv?rlw#en9P?}G6Zi#DNaptBZlR=-bm4#)L5op
z%}Gfbx9LHamlKMdP&J8Z`JC>y^mTzT4%M-^U$-f$F6Ja{T6dbZbl>plv60c^iILG$
z$MnGXiPPua8h-mNvnwh&D(fO_g6IxaW2v0(WHhT>qPkC&F42S;AzkEYrHdJeT->fZ
z>r!FTSXSi;1#w-E3`)7&pvsbi^<>NybRXBLn_>KnJ@BZ1gz5_V&X4@Tg#+^k77ouJ
z{-nR^>#gmI-!3hOkIp&Qe1XQuzN)WpE!=i3yEI-6$NzA)7Tf;m{u}#0e(lr4Hx94F
z`pfRs)>th*R32G8xV)wJi3>&ceCI}imW9#z(QDbNf7hD3wLDUF$7(HYbAfLH;cNYi
zYBkWk^ozCD_Uo^I`1<16j}AStqu{RPK=+d{s8_=A^(P?^R07@W>Q->~uiJZvyX<#S
z+i<`AZkJ=Y&vm!oG5md3vXL+_sX;IAKnLihF{U->g$P!%@kG`d6q9bG2&mRe36+3a
zZrN}02tyGN5&6%uAfSTy4BW*!+#SUs1Hj{Oj?aU;raTLFU6}b+goWaQ?o7#q%Gy|5
zWPXq^e&Mq)RJus=*<9Slc;2%uP^o-#7&D&CjI)MO@q1C-3l$u+8+n6QFD|}P^>i%_
zt-1XRo_SBXcX8*^5AV5qA@E*&-_y11=~`Dg;eFWgiv7=yL#|EC&D}DXJ8Cj_BeV@>
zgAa`ox+Jg}tsX95w++|72W^nsEoK4!df^ce@s`7y3&+*8>?{eJuUw5I+bG0kX`~xW
zZjB+-@E4evxQ!2lNC`YA0_S6`!RwpYQ~)+@uI;hx1M6Mnn*2Lx9L2rG-+Cbt@i_{q
zOs!+LXu>48ChuX+Y5xn!<$xG-nu(#jN*U(RF^CT<GQ}4PILD}}rsOGu+6Yxsj1qj2
z7V1o9&_wagQ4On<X1SphP^M{cGR0I4YpE>71tpIqMltZ5Y#9Sa$)u77K@Y5-JCmW3
zO3merwO|cEarQz=HaVP|n2}^Ks9<mvODdKcJc}ina4orR;Mn-^z{nQ@5NSSV>|i}`
z9`ubSB?s~P*3r7JNrf?*)Q;f3!k`>Kh;^Z`9y&#}-7wAySelYj@<7~W5S;GFv6QUo
zZt(4bA~B--IKL*Q6%8c1BPHkZn(oLbGn5&L61SVF#Br#*4C-WAi&TxI(*09ZOIYW~
z_Q1ql@Tk|Ig6#65V8;hPy)sgBg%_jCuC6tApb|V&b)T&TBWwO}B{Es{Pp!3g-EZ$-
zZtuVScD4Nnb0=V+{Nts*rEQhS-l~7!S~OPKzJDcp;C}Rt<>(tL(Ia!mYc0{m?)TrR
z_3SG<%f;o;OKYLn;z%{rU2E?uw|o<7t3>;%q4>i#)YAGeiu{rCo@L*Tin!x3_=R+L
zB9Vo-Bof6A2rFwI1Dw1{`4Jt&CNf-O3WtVP+;%Q61Ez5OisKk*%;W@6>_4g)FtN@!
zIW5Uk)PRB&p5`HO8naWGu$Mq}>UTGI`TG&ZiD`uKk8y_i06b8>I>L6stJpbS|8~Ur
z9D(Z^uIYR-3GklGXEKEW*3Bp3F~|U8X%%NB0uCsfd>ANhJ$5;zaS-#*0woV$3g+~I
z;+8W!S@1DgGcyUUX>B0h%J}~#gJ4+?v~-cDgvq*`L=0$|<jRkq2UnJ|G?6f|!{MOY
z0VQlVpA7#x5M1aPN^`82xBW7mX#yw?qAw8Z=TkXp>+VBYh2%5zFnblcoLbcps2|z|
zL3n^X50LKxYJvKzVC;&o7V5kbcqB#y{HV_%?0yt%6*?d8L85=IxMJ^E4Q%CDTopYP
zf6vn7=jqSVx3}FnztVfW(sQEX9j%Hdjj`UPOP?1%D^|YuqrZ%<>>jK1p00S$RK>Gj
zgQEAqopblZH&<Q$_hx=Q`^(w)-d*he(zOHNT=PfgjxL;>KUr3<o?f-EU5gDazIH!0
zxEve2(^icgnL7nTfbP9Zot4PTRsYatwdX^9VRnA@gLjv@zijEL^$gV!zgkE9DhK=i
uYUn^M)b@1)_%Q(fzdMm9SP8ydbq^V?@2ZMDe1l$mPu%(N6(nvm;`|>X(SI@k

literal 0
HcmV?d00001

diff --git a/src/__pycache__/startup.cpython-312.pyc b/src/__pycache__/startup.cpython-312.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..9b6c32893529700cf3b7f72dbdf402f02e86fe2a
GIT binary patch
literal 2620
zcmds3PfQe77=N>GXLlB6mj7y1U=fSK1gN6L7&V#2wIt$7cLB?mI9-?#2G^O*o7t|r
zg-VPEIeC=Cn3(hefuvq*n$+cF(jM591UquDX)nDIIGDht-<w?q1gq(>eUtg#_x}Cf
zpZWd1H$MwP34-+9lj+f89)$j&g{_2(&iYs2JVY{*nFNXzawf*eu9!<fV=!uW8uRV0
zK}g1POrbW$%I<q8=8;*jUfBbd%P`H}zi4yV60Yh=;~LSmk?y!QlJ%05);bh7hj&8K
zHPh9thXL~;@*&e{9dkLZ<4nNHCm6FhKSE<}r^#`~(KzgA2hKb8HCG&W8cn!%pm7d}
z<BYpZ`&?<oTd2I##c^b8`nL*?;G7=gIkr${knE<<Tyakf%D5ZR31l#77lQ^615d_%
z8xNo~(}>LSEtI2(9GJjn)mB+%$MGZd*tPWq=vdi3fxpEWK&N?{NuvTzvnPA9Bmha2
zWg}pYDuH-1nNWx0rmAZJQx9BIHCay^0mF<FGnH($J8!f^^&4h7PL#nb>M+p_C=C{S
z4tAypQ8cq(AqHR!Mp7D~8?u-q;p|5tLUa;{%d)DCQ1~M3*+6rN<w>g=QAVtauFIhT
zsk1K<359#5zGx_7d9NlF?Xr@w1lkl0^-BF>cVEaVi7IA)JdsjDw=GWW>5;<Xm5}9$
zgg%p^F^iA%g{7d_EA@!IU4Rj_XSdiSQ8jbcZQ+}M=rvM95_Nc(TQy9Qu}h3%(9chg
zW}~VN4yDZ?!);rqkjz+q*o0*2Qn8Bchu&UzjeB4w(JJR*+gAM5%l@Xkzv(`HSsA?V
zT@k7tT%Wo=-T7qyOv8Nta*LR65toGE^TY~Yxy;w*`P%8keAm<XkNbYA{h{_n!!O-0
zE-Vhj7Q4P&;s@Vg<nLsbg<xI?ZvI(XxrP{+f)f_Yn`atksuuk%%Uo-oYu#MM(6C{g
zf`!~Y6U6A*DX|0RS~+lY9c&PJbLTMl3&^(E7d(m=cz>`SFVwT(zok8@3tM}i{0FqR
ztsEKCXS<xidz6G7B{?buC1I+T%npqf)l~uKU?<uA4s~H$T`=P=b+OqFTQCvjmj1P}
zS70}Ye(T*5BQz-3!U<<<3#uU$gzCl~wXGZ<p}ZqBE)g1v35`Wo=|v@>n2N0qLU(9Y
zTv8KCST`@}DNVK^Ng3@<0|}w~a~Mzd(f(ETgo;F`b7+1DT4<y(z5zpN?ql2kCC%&d
zeBJc;{1;37r@JV9cn!H&;dhQ-DX_cUW_Qa>d?vc+Ke5c6%yTDgcDHY=pN0AFo)M3u
zXNSZ(oU7r$&DF8uF>kIBQ~#K4kNbi(ICq+(_Pjq>ju*;V@c)+_f8TAfm*edAJ+z18
zo`2*xyNBc6>{*-RoukU|O(US**mI4jP^*2A=DnA)^#!62QzT0CBFAZNrVO|wc2nKr
z=zS0gi5D&Y^OQm|#mm4ZJUIvyc|XMk)sR%pA$I@<9D&!E07J=L!q$9%+}}&H(voog
z`C&VP*XQ~AC;a^A65m0oec^Rk)f={uZ?2RAy&1OVx3X%R-Y&qV!Mp7#Gh;0J+n2ag
zD`nN2t1L8Zki)QoW}oHKje@#u1-p}RbJX(2hYacv`X6E6JA|r|1i;Lqb+D))VD8*4
zT}~yG3#19^=>L#$0t{Ra4D$;4ULoOi>A^|vWm(08^i=x6ovAz1nP1DAC;2sZ3DdTA
QKw#R}>-<a&`4~q14Ul{-XaE2J

literal 0
HcmV?d00001

diff --git a/src/config.py b/src/config.py
index 72051be..e9da6d1 100644
--- a/src/config.py
+++ b/src/config.py
@@ -1,14 +1,45 @@
 import os
+import json
 import keyboard
 import time
-CONFIG_FILE = "shortcut.txt"
+
+CONFIG_FILE = "config.json"
+SHORTCUT_FILE = "shortcut.txt" # For migrating old config
+
+def load_config():
+    """
+    Loads the configuration from config.json.
+    If the file doesn't exist, it creates it with default values.
+    It also migrates the old shortcut.txt if it exists.
+    """
+    if not os.path.exists(CONFIG_FILE):
+        config = {"run_on_startup": False, "shortcut": "ctrl+space"}
+        if os.path.exists(SHORTCUT_FILE):
+            with open(SHORTCUT_FILE, "r") as f:
+                shortcut = f.read().strip()
+                if shortcut:
+                    config["shortcut"] = shortcut
+            os.remove(SHORTCUT_FILE) # remove old config file
+        save_config(config)
+        return config
+    else:
+        with open(CONFIG_FILE, "r") as f:
+            return json.load(f)
+
+def save_config(config):
+    """
+    Saves the configuration to config.json.
+    """
+    with open(CONFIG_FILE, "w") as f:
+        json.dump(config, f, indent=4)
 
 def get_shortcut():
     """
     Retrieves the shortcut from the config file.
     If the file doesn't exist, it prompts the user to create one by pressing the desired keys.
     """
-    if not os.path.exists(CONFIG_FILE):
+    config = load_config()
+    if "shortcut" not in config:
         print("It seems this is the first time you are running the application.")
         while True:
             print("Please press the key combination you want to use as a shortcut and then press ESC.")
@@ -16,20 +47,19 @@ def get_shortcut():
             print(f"You pressed: {shortcut}")
             confirm = input("Is this correct? (y/n): ").lower()
             if confirm == 'y':
-                with open(CONFIG_FILE, "w") as f:
-                    f.write(shortcut)
+                config["shortcut"] = shortcut
+                save_config(config)
                 return shortcut
             else:
                 print("Let's try again.")
-    else:
-        with open(CONFIG_FILE, "r") as f:
-            return f.read().strip()
+    return config["shortcut"]
 
 if __name__ == "__main__":
-    shortcut = get_shortcut()
-    print(f"Your shortcut is set to: {shortcut}")
-    print("You can now use this shortcut in the application.")
-    print('do you want to change the shortcut? (y/n): ', end='')
+    config = load_config()
+    print(f"Your shortcut is set to: {config.get('shortcut', 'Not set')}")
+    print(f"Run on startup is set to: {config.get('run_on_startup', 'Not set')}")
+
+    print('Do you want to change the shortcut? (y/n): ', end='')
     if input().lower() == 'y':
         while True:
             time.sleep(0.2)
@@ -39,14 +69,14 @@ if __name__ == "__main__":
             time.sleep(0.5)
             confirm = input("Is this correct? (y/n): ").lower()
             if confirm.lower() == 'y':
-                with open(CONFIG_FILE, "w") as f:
-                    f.write(new_shortcut)
+                config['shortcut'] = new_shortcut
+                save_config(config)
                 print("Shortcut updated successfully.")
                 break
             else:
                 print("Let's try again.")
-
     else:
         print("No changes made to the shortcut.")
-        print("Exiting without changes.")
-        input("Press Enter to exit.")
\ No newline at end of file
+
+    print("Exiting without changes.")
+    input("Press Enter to exit.")
diff --git a/src/main.pyw b/src/main.pyw
index ed43ea8..5d02414 100644
--- a/src/main.pyw
+++ b/src/main.pyw
@@ -8,7 +8,8 @@ import threading
 import pystray
 from PIL import Image
 import re
-from config import get_shortcut
+from config import load_config, save_config, get_shortcut
+from startup import add_to_startup, remove_from_startup, is_in_startup
 
 def convert_en_characters(input_str):
     """
@@ -127,15 +128,35 @@ def quit_app(icon, item):
     import os
     os._exit(0)
 
-def setup_tray():
-    image = Image.open("assets/icon.png")  # Make sure icon.png exists in your project folder
-    menu = pystray.Menu(pystray.MenuItem('Quit', quit_app))
+def setup_tray(config):
+    def on_toggle_startup(icon, item):
+        config["run_on_startup"] = not item.checked
+        save_config(config)
+        if config["run_on_startup"]:
+            add_to_startup()
+        else:
+            remove_from_startup()
+
+    image = Image.open("assets/icon.png")
+    menu = pystray.Menu(
+        pystray.MenuItem(
+            'Run on Startup',
+            on_toggle_startup,
+            checked=lambda item: config["run_on_startup"]
+        ),
+        pystray.MenuItem('Quit', quit_app)
+    )
     icon = pystray.Icon("oopswronglang", image, "OopsWrongLang", menu)
     icon.run()
 
 def main():
+    config = load_config()
+    # Sync startup state just in case it was changed externally
+    config["run_on_startup"] = is_in_startup()
+    save_config(config)
+
     # Start the tray icon in a separate thread
-    tray_thread = threading.Thread(target=setup_tray, daemon=True)
+    tray_thread = threading.Thread(target=lambda: setup_tray(config), daemon=True)
     tray_thread.start()
 
     # Register the keyboard shortcut
diff --git a/src/startup.py b/src/startup.py
new file mode 100644
index 0000000..9e66093
--- /dev/null
+++ b/src/startup.py
@@ -0,0 +1,48 @@
+import os
+import winreg
+
+APP_NAME = "OopsWrongLang"
+RUN_BAT_PATH = os.path.abspath("run.bat")
+
+def add_to_startup():
+    """
+    Adds the application to Windows startup.
+    """
+    try:
+        key = winreg.HKEY_CURRENT_USER
+        sub_key = r"Software\Microsoft\Windows\CurrentVersion\Run"
+        with winreg.OpenKey(key, sub_key, 0, winreg.KEY_SET_VALUE) as registry_key:
+            winreg.SetValueEx(registry_key, APP_NAME, 0, winreg.REG_SZ, f'"{RUN_BAT_PATH}"')
+    except OSError as e:
+        print(f"Error adding to startup: {e}")
+
+def remove_from_startup():
+    """
+    Removes the application from Windows startup.
+    """
+    try:
+        key = winreg.HKEY_CURRENT_USER
+        sub_key = r"Software\Microsoft\Windows\CurrentVersion\Run"
+        with winreg.OpenKey(key, sub_key, 0, winreg.KEY_SET_VALUE) as registry_key:
+            winreg.DeleteValue(registry_key, APP_NAME)
+    except FileNotFoundError:
+        # This is fine, it means the key doesn't exist
+        pass
+    except OSError as e:
+        print(f"Error removing from startup: {e}")
+
+def is_in_startup():
+    """
+    Checks if the application is in Windows startup.
+    """
+    try:
+        key = winreg.HKEY_CURRENT_USER
+        sub_key = r"Software\Microsoft\Windows\CurrentVersion\Run"
+        with winreg.OpenKey(key, sub_key, 0, winreg.KEY_READ) as registry_key:
+            winreg.QueryValueEx(registry_key, APP_NAME)
+        return True
+    except FileNotFoundError:
+        return False
+    except OSError as e:
+        print(f"Error checking startup status: {e}")
+        return False