From abb1a0cfc3aa06817a5e400342e8c12b68e80506 Mon Sep 17 00:00:00 2001
From: embire2 <ceo@openweb.co.za>
Date: Fri, 6 Feb 2026 15:31:27 +0200
Subject: [PATCH] feat: add file lock button

---
 app/components/ui/PanelHeaderButton.tsx  |  4 +-
 app/components/workbench/EditorPanel.tsx | 63 +++++++++++++++++++-----
 2 files changed, 54 insertions(+), 13 deletions(-)

diff --git a/app/components/ui/PanelHeaderButton.tsx b/app/components/ui/PanelHeaderButton.tsx
index 9faea1cc45..89b6da1078 100644
--- a/app/components/ui/PanelHeaderButton.tsx
+++ b/app/components/ui/PanelHeaderButton.tsx
@@ -5,12 +5,13 @@ interface PanelHeaderButtonProps {
   className?: string;
   disabledClassName?: string;
   disabled?: boolean;
+  title?: string;
   children: string | JSX.Element | Array<JSX.Element | string>;
   onClick?: (event: React.MouseEvent<HTMLButtonElement, MouseEvent>) => void;
 }
 
 export const PanelHeaderButton = memo(
-  ({ className, disabledClassName, disabled = false, children, onClick }: PanelHeaderButtonProps) => {
+  ({ className, disabledClassName, disabled = false, title, children, onClick }: PanelHeaderButtonProps) => {
     return (
       <button
         className={classNames(
@@ -21,6 +22,7 @@ export const PanelHeaderButton = memo(
           className,
         )}
         disabled={disabled}
+        title={title}
         onClick={(event) => {
           if (disabled) {
             return;
diff --git a/app/components/workbench/EditorPanel.tsx b/app/components/workbench/EditorPanel.tsx
index 48c1c0ccad..f185d16969 100644
--- a/app/components/workbench/EditorPanel.tsx
+++ b/app/components/workbench/EditorPanel.tsx
@@ -1,5 +1,6 @@
 import { useStore } from '@nanostores/react';
 import { memo, useMemo } from 'react';
+import { toast } from 'react-toastify';
 import { Panel, PanelGroup, PanelResizeHandle } from 'react-resizable-panels';
 import * as Tabs from '@radix-ui/react-tabs';
 import {
@@ -80,6 +81,34 @@ export const EditorPanel = memo(
       return unsavedFiles instanceof Set && unsavedFiles.has(editorDocument.filePath);
     }, [editorDocument, unsavedFiles]);
 
+    const lockState = useMemo(() => {
+      if (!editorDocument) {
+        return { locked: false, lockedBy: undefined as string | undefined };
+      }
+
+      return workbenchStore.isFileLocked(editorDocument.filePath);
+    }, [editorDocument, files]);
+
+    const isLockedByFolder = Boolean(
+      editorDocument && lockState.locked && lockState.lockedBy && lockState.lockedBy !== editorDocument.filePath,
+    );
+
+    const toggleFileLock = () => {
+      if (!editorDocument) {
+        return;
+      }
+
+      if (lockState.locked) {
+        const success = workbenchStore.unlockFile(editorDocument.filePath);
+        toast[success ? 'success' : 'error'](success ? 'File unlocked' : 'Failed to unlock file');
+
+        return;
+      }
+
+      const success = workbenchStore.lockFile(editorDocument.filePath);
+      toast[success ? 'success' : 'error'](success ? 'File locked' : 'Failed to lock file');
+    };
+
     return (
       <PanelGroup direction="vertical">
         <Panel defaultSize={showTerminal ? DEFAULT_EDITOR_SIZE : 100} minSize={20}>
@@ -148,18 +177,28 @@ export const EditorPanel = memo(
                 {activeFileSegments?.length && (
                   <div className="flex items-center flex-1 text-sm">
                     <FileBreadcrumb pathSegments={activeFileSegments} files={files} onFileSelect={onFileSelect} />
-                    {activeFileUnsaved && (
-                      <div className="flex gap-1 ml-auto -mr-1.5">
-                        <PanelHeaderButton onClick={onFileSave}>
-                          <div className="i-ph:floppy-disk-duotone" />
-                          Save
-                        </PanelHeaderButton>
-                        <PanelHeaderButton onClick={onFileReset}>
-                          <div className="i-ph:clock-counter-clockwise-duotone" />
-                          Reset
-                        </PanelHeaderButton>
-                      </div>
-                    )}
+                    <div className="flex gap-1 ml-auto -mr-1.5">
+                      <PanelHeaderButton
+                        onClick={toggleFileLock}
+                        disabled={!editorDocument || isLockedByFolder}
+                        title={isLockedByFolder ? 'Locked by folder' : lockState.locked ? 'Unlock file' : 'Lock file'}
+                      >
+                        <div className={lockState.locked ? 'i-ph:lock-simple-duotone' : 'i-ph:lock-key-open-duotone'} />
+                        {lockState.locked ? 'Unlock' : 'Lock'}
+                      </PanelHeaderButton>
+                      {activeFileUnsaved && (
+                        <>
+                          <PanelHeaderButton onClick={onFileSave}>
+                            <div className="i-ph:floppy-disk-duotone" />
+                            Save
+                          </PanelHeaderButton>
+                          <PanelHeaderButton onClick={onFileReset}>
+                            <div className="i-ph:clock-counter-clockwise-duotone" />
+                            Reset
+                          </PanelHeaderButton>
+                        </>
+                      )}
+                    </div>
                   </div>
                 )}
               </PanelHeader>