import re

with open('c:/ScriptoriumAI/scriptoriumai-ui/src/stores/editorStore.ts', 'r', encoding='utf-8') as f:
    text = f.read()

if 'zenMode' not in text:
    text = text.replace('showSidebar: boolean', 'showSidebar: boolean\n  zenMode: boolean')
    text = text.replace('toggleDarkMode: () => void', 'toggleDarkMode: () => void\n  toggleZenMode: () => void\n  setZenMode: (zen: boolean) => void')
    
    text = text.replace('darkMode: true,', 'darkMode: true,\n      zenMode: false,')
    
    text = text.replace('toggleDarkMode: () => set((state) => ({ darkMode: !state.darkMode })),', 'toggleDarkMode: () => set((state) => ({ darkMode: !state.darkMode })),\n      toggleZenMode: () => set((state) => ({ zenMode: !state.zenMode })),\n      setZenMode: (zenMode) => set({ zenMode }),')

    with open('c:/ScriptoriumAI/scriptoriumai-ui/src/stores/editorStore.ts', 'w', encoding='utf-8') as f:
        f.write(text)
    
    print('Added zenMode to editorStore.ts')
else:
    print('zenMode already in editorStore.ts')