From a4c2c039d2cc53380f45e6300dd2505eb05858b5 Mon Sep 17 00:00:00 2001
From: 409 <409dev@protonmail.com>
Date: Sat, 6 Sep 2025 01:21:13 +0200
Subject: [PATCH] basic text editor

---
 backend/src/lib/outbound/file_system.rs       |   1 +
 frontend/components/actions/UploadDialog.vue  |   2 +-
 .../components/{ => viewers}/ImageViewer.vue  |  15 +-
 frontend/components/viewers/TextEditor.vue    | 152 ++++++++++++++++++
 frontend/layouts/default.vue                  |   5 +
 frontend/layouts/share.vue                    |   8 +-
 frontend/lib/api/warrens.ts                   |   9 --
 frontend/pages/share.vue                      |  27 +++-
 frontend/pages/warrens/files.vue              |  29 +++-
 frontend/stores/index.ts                      |   3 -
 frontend/stores/viewers/image.ts              |  13 ++
 frontend/stores/viewers/index.ts              |   4 +
 frontend/stores/viewers/text.ts               |  71 ++++++++
 13 files changed, 315 insertions(+), 24 deletions(-)
 rename frontend/components/{ => viewers}/ImageViewer.vue (60%)
 create mode 100644 frontend/components/viewers/TextEditor.vue
 create mode 100644 frontend/stores/viewers/image.ts
 create mode 100644 frontend/stores/viewers/index.ts
 create mode 100644 frontend/stores/viewers/text.ts

diff --git a/backend/src/lib/outbound/file_system.rs b/backend/src/lib/outbound/file_system.rs
index 53784dc..3458d5b 100644
--- a/backend/src/lib/outbound/file_system.rs
+++ b/backend/src/lib/outbound/file_system.rs
@@ -214,6 +214,7 @@ impl FileSystem {
             let mut file = fs::OpenOptions::new()
                 .write(true)
                 .create(true)
+                .truncate(true)
                 .open(&file_path)
                 .await?;
 
diff --git a/frontend/components/actions/UploadDialog.vue b/frontend/components/actions/UploadDialog.vue
index f637a03..21fdc92 100644
--- a/frontend/components/actions/UploadDialog.vue
+++ b/frontend/components/actions/UploadDialog.vue
@@ -35,7 +35,7 @@ const fileInputElement = ref<HTMLInputElement>(
 const uploading = ref(false);
 const dropZoneRef = ref<HTMLElement>();
 
-const dropZone = useDropZone(dropZoneRef, {
+useDropZone(dropZoneRef, {
     onDrop,
     multiple: true,
 });
diff --git a/frontend/components/ImageViewer.vue b/frontend/components/viewers/ImageViewer.vue
similarity index 60%
rename from frontend/components/ImageViewer.vue
rename to frontend/components/viewers/ImageViewer.vue
index 39a5b06..f80b202 100644
--- a/frontend/components/ImageViewer.vue
+++ b/frontend/components/viewers/ImageViewer.vue
@@ -1,18 +1,17 @@
 <script setup lang="ts">
-const warrenStore = useWarrenStore();
+import { useImageViewer } from '~/stores/viewers';
+
+const imageViewer = useImageViewer();
 
 function onOpenUpdate(state: boolean) {
     if (!state) {
-        warrenStore.imageViewer.src = null;
+        imageViewer.close();
     }
 }
 </script>
 
 <template>
-    <Dialog
-        :open="warrenStore.imageViewer.src != null"
-        @update:open="onOpenUpdate"
-    >
+    <Dialog :open="imageViewer.src != null" @update:open="onOpenUpdate">
         <DialogTrigger>
             <slot />
         </DialogTrigger>
@@ -20,9 +19,9 @@ function onOpenUpdate(state: boolean) {
             class="w-full overflow-hidden p-0 sm:!max-h-[90vh] sm:!max-w-[90vw]"
         >
             <img
-                v-if="warrenStore.imageViewer.src"
+                v-if="imageViewer.src"
                 class="h-full w-full overflow-hidden !object-contain"
-                :src="warrenStore.imageViewer.src"
+                :src="imageViewer.src"
             />
         </DialogContent>
     </Dialog>
diff --git a/frontend/components/viewers/TextEditor.vue b/frontend/components/viewers/TextEditor.vue
new file mode 100644
index 0000000..56e944c
--- /dev/null
+++ b/frontend/components/viewers/TextEditor.vue
@@ -0,0 +1,152 @@
+<script setup lang="ts">
+import { toast } from 'vue-sonner';
+import { useTextEditor } from '~/stores/viewers';
+
+const editor = useTextEditor();
+const textarea = ref<HTMLTextAreaElement>(
+    null as unknown as HTMLTextAreaElement
+);
+
+const currentLine = ref<number>(0);
+const totalLines = computed(
+    () => editor.data?.editedContent.split('\n').length ?? 0
+);
+
+function onOpenUpdate(state: boolean) {
+    if (!state) {
+        editor.close();
+    }
+}
+
+async function saveFile() {
+    if (editor.data == null) {
+        return;
+    }
+
+    const name = editor.data.entry.name;
+
+    const success = await editor.save();
+
+    if (success) {
+        toast.success('Save', {
+            id: 'TEXT_EDITOR_SAVE_TOAST',
+            description: `Successfully saved ${name}`,
+        });
+    } else {
+        toast.error('Save', {
+            id: 'TEXT_EDITOR_SAVE_TOAST',
+            description: `Failed to save ${name}`,
+        });
+    }
+}
+
+function onEditorKeyDown(e: KeyboardEvent) {
+    if (editor.saving) {
+        e.preventDefault();
+        return;
+    }
+
+    requestAnimationFrame(updateCurrentLine);
+
+    if (!e.ctrlKey || e.key !== 's') {
+        return;
+    }
+
+    e.preventDefault();
+    saveFile();
+}
+
+function updateCurrentLine() {
+    if (editor.data == null) {
+        return;
+    }
+
+    currentLine.value = editor.data.editedContent
+        .substring(0, textarea.value.selectionStart)
+        .split('\n').length;
+}
+
+function onEditorClick() {
+    requestAnimationFrame(updateCurrentLine);
+}
+
+onMounted(() => {
+    requestAnimationFrame(updateCurrentLine);
+});
+</script>
+
+<template>
+    <Dialog :open="editor.data != null" @update:open="onOpenUpdate">
+        <DialogContent
+            v-if="editor.data"
+            class="flex h-full max-h-[min(98vh,700px)] w-full !max-w-[min(98vw,1000px)] flex-col"
+        >
+            <DialogHeader>
+                <DialogTitle>{{ editor.data.entry.name }}</DialogTitle>
+            </DialogHeader>
+
+            <div v-if="editor.data.content == null">
+                <span class="text-muted-foreground animate-pulse"
+                    >Loading content...</span
+                >
+            </div>
+            <div v-else class="flex h-full w-full flex-col gap-2">
+                <div
+                    class="flex h-fit w-full flex-row items-center justify-end gap-1"
+                >
+                    <Button
+                        title="Reset"
+                        variant="destructive"
+                        size="icon"
+                        :disabled="
+                            editor.data.content === editor.data.editedContent ||
+                            editor.saving
+                        "
+                        @click="() => editor.discardEdits()"
+                    >
+                        <Icon name="lucide:bomb" />
+                    </Button>
+
+                    <Separator orientation="vertical" class="mx-1" />
+
+                    <Button
+                        title="Save"
+                        size="icon"
+                        :disabled="
+                            editor.saving ||
+                            editor.data.content === editor.data.editedContent
+                        "
+                        @click="saveFile"
+                    >
+                        <Icon name="lucide:save" />
+                    </Button>
+                </div>
+
+                <textarea
+                    id="editor"
+                    ref="textarea"
+                    v-model="editor.data.editedContent"
+                    class="h-full w-full resize-none overflow-auto rounded-md border p-4 whitespace-pre focus:outline-0"
+                    @keydown="onEditorKeyDown"
+                    @click="onEditorClick"
+                ></textarea>
+
+                <div class="flex w-full flex-row justify-end">
+                    <p class="text-muted-foreground">
+                        Line {{ currentLine }} / {{ totalLines }}
+                    </p>
+                </div>
+            </div>
+        </DialogContent>
+    </Dialog>
+</template>
+
+<style>
+#editor {
+    scrollbar-width: 0 !important;
+}
+
+#editor::-webkit-scrollbar {
+    display: none !important;
+}
+</style>
diff --git a/frontend/layouts/default.vue b/frontend/layouts/default.vue
index acab31c..18e3051 100644
--- a/frontend/layouts/default.vue
+++ b/frontend/layouts/default.vue
@@ -1,4 +1,6 @@
 <script setup lang="ts">
+import TextEditor from '~/components/viewers/TextEditor.vue';
+import ImageViewer from '@/components/viewers/ImageViewer.vue';
 import { Separator } from '@/components/ui/separator';
 import CreateDirectoryDialog from '~/components/actions/CreateDirectoryDialog.vue';
 import UploadDialog from '~/components/actions/UploadDialog.vue';
@@ -18,7 +20,10 @@ await useAsyncData('warrens', async () => {
     <SidebarProvider>
         <SelectionRect />
         <ActionsShareDialog />
+
+        <TextEditor />
         <ImageViewer />
+
         <AppSidebar />
         <SidebarInset class="flex flex-col-reverse md:flex-col">
             <header
diff --git a/frontend/layouts/share.vue b/frontend/layouts/share.vue
index 3d62735..308c0ce 100644
--- a/frontend/layouts/share.vue
+++ b/frontend/layouts/share.vue
@@ -1,9 +1,15 @@
-<script lang="ts" setup></script>
+<script lang="ts" setup>
+import TextEditor from '~/components/viewers/TextEditor.vue';
+import ImageViewer from '@/components/viewers/ImageViewer.vue';
+</script>
 
 <template>
     <main class="flex h-full w-full items-center justify-center">
         <SelectionRect />
+
+        <TextEditor />
         <ImageViewer />
+
         <slot />
     </main>
 </template>
diff --git a/frontend/lib/api/warrens.ts b/frontend/lib/api/warrens.ts
index e8256f9..40c7aad 100644
--- a/frontend/lib/api/warrens.ts
+++ b/frontend/lib/api/warrens.ts
@@ -245,20 +245,11 @@ export async function uploadToWarren(
     try {
         await promise;
     } catch {
-        toast.error('Upload', {
-            id: 'UPLOAD_FILE_TOAST',
-            description: `Failed to upload`,
-        });
         return { success: false };
     }
 
     await refreshNuxtData('current-directory');
 
-    toast.success('Upload', {
-        id: 'UPLOAD_FILE_TOAST',
-        description: `Successfully uploaded ${files.length} file${files.length !== 1 ? 's' : ''}`,
-    });
-
     return { success: true };
 }
 
diff --git a/frontend/pages/share.vue b/frontend/pages/share.vue
index 3466ee6..0e45176 100644
--- a/frontend/pages/share.vue
+++ b/frontend/pages/share.vue
@@ -2,6 +2,7 @@
 import { fetchShareFile, getShare, listShareFiles } from '~/lib/api/shares';
 import type { DirectoryEntry } from '~/shared/types';
 import type { Share } from '~/shared/types/shares';
+import { useImageViewer, useTextEditor } from '~/stores/viewers';
 
 definePageMeta({
     layout: 'share',
@@ -10,6 +11,9 @@ definePageMeta({
 const warrenStore = useWarrenStore();
 const route = useRoute();
 
+const imageViewer = useImageViewer();
+const textEditor = useTextEditor();
+
 const entries = computed(() =>
     warrenStore.current != null && warrenStore.current.dir != null
         ? warrenStore.current.dir.entries
@@ -127,8 +131,29 @@ async function onEntryClicked(entry: DirectoryEntry, event: MouseEvent) {
 
         if (result.success) {
             const url = URL.createObjectURL(result.data);
-            warrenStore.imageViewer.src = url;
+            imageViewer.open(url);
         }
+
+        return;
+    }
+
+    if (entry.mimeType.startsWith('text/')) {
+        const result = await fetchShareFile(
+            share!.data.id,
+            entryPath,
+            password.value.length > 0 ? password.value : null
+        );
+
+        if (result.success) {
+            textEditor.open(
+                warrenStore.current.warrenId,
+                warrenStore.current.path,
+                entry,
+                result.data
+            );
+        }
+
+        return;
     }
 }
 
diff --git a/frontend/pages/warrens/files.vue b/frontend/pages/warrens/files.vue
index d266801..ba6e177 100644
--- a/frontend/pages/warrens/files.vue
+++ b/frontend/pages/warrens/files.vue
@@ -5,12 +5,17 @@ import DirectoryListContextMenu from '~/components/DirectoryListContextMenu.vue'
 import RenameEntryDialog from '~/components/actions/RenameEntryDialog.vue';
 import { fetchFile, getWarrenDirectory, warrenRm } from '~/lib/api/warrens';
 import type { DirectoryEntry } from '~/shared/types';
+import { useImageViewer, useTextEditor } from '~/stores/viewers';
 
 definePageMeta({
     middleware: ['authenticated'],
 });
 
 const warrenStore = useWarrenStore();
+
+const imageViewer = useImageViewer();
+const textEditor = useTextEditor();
+
 const loadingIndicator = useLoadingIndicator();
 const uploadStore = useUploadStore();
 const warrenPath = computed(() => useWarrenPath());
@@ -105,10 +110,32 @@ async function onEntryClicked(entry: DirectoryEntry, event: MouseEvent) {
             warrenStore.current.path,
             entry.name
         );
+
         if (result.success) {
             const url = URL.createObjectURL(result.data);
-            warrenStore.imageViewer.src = url;
+            imageViewer.open(url);
         }
+
+        return;
+    }
+
+    if (entry.mimeType.startsWith('text/')) {
+        const result = await fetchFile(
+            warrenStore.current.warrenId,
+            warrenStore.current.path,
+            entry.name
+        );
+
+        if (result.success) {
+            textEditor.open(
+                warrenStore.current.warrenId,
+                warrenStore.current.path,
+                entry,
+                result.data
+            );
+        }
+
+        return;
     }
 }
 
diff --git a/frontend/stores/index.ts b/frontend/stores/index.ts
index 84a6973..65983bb 100644
--- a/frontend/stores/index.ts
+++ b/frontend/stores/index.ts
@@ -6,9 +6,6 @@ import { getParentPath } from '~/utils/files';
 export const useWarrenStore = defineStore('warrens', {
     state: () => ({
         warrens: {} as Record<string, WarrenData>,
-        imageViewer: {
-            src: null as string | null,
-        },
         current: null as {
             warrenId: string;
             path: string;
diff --git a/frontend/stores/viewers/image.ts b/frontend/stores/viewers/image.ts
new file mode 100644
index 0000000..9f09058
--- /dev/null
+++ b/frontend/stores/viewers/image.ts
@@ -0,0 +1,13 @@
+export const useImageViewer = defineStore('image-viewer', {
+    state: () => ({
+        src: null as string | null,
+    }),
+    actions: {
+        open(src: string) {
+            this.src = src;
+        },
+        close() {
+            this.src = null;
+        },
+    },
+});
diff --git a/frontend/stores/viewers/index.ts b/frontend/stores/viewers/index.ts
new file mode 100644
index 0000000..55768ae
--- /dev/null
+++ b/frontend/stores/viewers/index.ts
@@ -0,0 +1,4 @@
+import { useImageViewer } from './image';
+import { useTextEditor } from './text';
+
+export { useImageViewer, useTextEditor };
diff --git a/frontend/stores/viewers/text.ts b/frontend/stores/viewers/text.ts
new file mode 100644
index 0000000..57696b3
--- /dev/null
+++ b/frontend/stores/viewers/text.ts
@@ -0,0 +1,71 @@
+import { uploadToWarren } from '~/lib/api/warrens';
+import type { DirectoryEntry } from '~/shared/types';
+
+export const useTextEditor = defineStore('text-editor', {
+    state: () => ({
+        data: null as {
+            warrenId: string;
+            parentPath: string;
+            entry: DirectoryEntry;
+            content: string;
+            editedContent: string;
+        } | null,
+        saving: false as boolean,
+    }),
+    actions: {
+        async open(
+            warrenId: string,
+            parentPath: string,
+            entry: DirectoryEntry,
+            content: Blob
+        ) {
+            const contentString = await content.text();
+
+            this.data = {
+                warrenId,
+                parentPath,
+                entry,
+                content: contentString,
+                editedContent: contentString,
+            };
+        },
+        async save(): Promise<boolean> {
+            if (this.saving || this.data == null) {
+                return false;
+            }
+
+            this.saving = true;
+
+            const dt = new DataTransfer();
+            dt.items.add(
+                new File([this.data.editedContent], this.data.entry.name)
+            );
+
+            const result = await uploadToWarren(
+                this.data.warrenId,
+                this.data.parentPath,
+                dt.files,
+                undefined
+            );
+
+            this.$patch({
+                data: {
+                    content: this.data.editedContent,
+                },
+                saving: false,
+            });
+
+            return result.success;
+        },
+        discardEdits() {
+            if (this.data == null) {
+                return;
+            }
+
+            this.data.editedContent = this.data.content;
+        },
+        close() {
+            this.data = null;
+        },
+    },
+});