Add copy icon for copying text

Author: celonymire

CHANGELOG.md

@@ -3,8 +3,10 @@
 ### Added
 
 - Support for opened files without active workspaces
+- Icon to copy the text of the text boxes
 
 ### Changed
+
 - Made text expansion editor's title human-friendly
 
 ### Fixed

src/extension/providers/JudgeViewProvider.ts

@@ -33,6 +33,7 @@ import {
 import { getLogger } from "../utils/logging";
 import {
   ActionMessageSchema,
+  CopyMessageSchema,
   NewInteractorSecretMessageSchema,
   NextMessageSchema,
   ProviderMessageSchema,
@@ -541,6 +542,9 @@ export default class extends BaseViewProvider<typeof ProviderMessageSchema, Webv
       case "VIEW":
         this._viewStdio(msg);
         break;
+      case "COPY":
+        void this._copyStdio(msg);
+        break;
       case "STDIN":
         this._stdin(msg);
         break;
@@ -1494,6 +1498,34 @@ export default class extends BaseViewProvider<typeof ProviderMessageSchema, Webv
     }
   }
 
+  private async _copyStdio({ uuid, stdio }: v.InferOutput<typeof CopyMessageSchema>) {
+    const testcase = this._findTestcase(uuid);
+    if (!testcase) {
+      return;
+    }
+
+    let value: string;
+    switch (stdio) {
+      case "STDIN":
+        value = testcase.stdin.data;
+        break;
+      case "STDERR":
+        value = testcase.stderr.data;
+        break;
+      case "STDOUT":
+        value = testcase.stdout.data;
+        break;
+      case "ACCEPTED_STDOUT":
+        value = testcase.acceptedStdout.data;
+        break;
+      case "INTERACTOR_SECRET":
+        value = testcase.interactorSecret.data;
+        break;
+    }
+
+    await vscode.env.clipboard.writeText(value);
+  }
+
   private _openInteractor() {
     if (!this._currentFile) {
       return;

src/extension/providers/StressViewProvider.ts

@@ -26,6 +26,7 @@ import { getLogger } from "../utils/logging";
 import type JudgeViewProvider from "./JudgeViewProvider";
 import {
   AddMessageSchema,
+  CopyMessageSchema,
   OpenMessageSchema,
   ProviderMessageSchema,
   SaveMessageSchema,
@@ -117,6 +118,9 @@ export default class extends BaseViewProvider<typeof ProviderMessageSchema, Webv
       case "VIEW":
         this._view(msg);
         break;
+      case "COPY":
+        void this._copy(msg);
+        break;
       case "ADD":
         this._add(msg);
         break;
@@ -609,6 +613,34 @@ export default class extends BaseViewProvider<typeof ProviderMessageSchema, Webv
     }
   }
 
+  private async _copy({ id, stdio }: v.InferOutput<typeof CopyMessageSchema>) {
+    const ctx = this._currentContext;
+    if (!ctx) return;
+    const state = ctx.state.find((s) => s.state === id);
+    if (!state) {
+      return;
+    }
+
+    let value: string;
+    switch (stdio) {
+      case "STDIN":
+        value = state.stdin.data;
+        break;
+      case "STDOUT":
+        value = state.stdout.data;
+        break;
+      case "STDERR":
+        value = state.stderr.data;
+        break;
+      case "ACCEPTED_STDOUT":
+      case "INTERACTOR_SECRET":
+        value = "";
+        break;
+    }
+
+    await vscode.env.clipboard.writeText(value);
+  }
+
   private _add({ id }: v.InferOutput<typeof AddMessageSchema>) {
     if (!this._currentFile) {
       return;

src/shared/judge-messages.ts

@@ -30,6 +30,7 @@ export const ProviderMessageTypeValues = [
   "STDIN",
   "TL",
   "ML",
+  "COPY",
 ] as const;
 
 export type ProviderMessageTypeValue = (typeof ProviderMessageTypeValues)[number];
@@ -64,6 +65,12 @@ export const ViewMessageSchema = v.object({
   stdio: StdioSchema,
 });
 
+export const CopyMessageSchema = v.object({
+  type: v.literal("COPY"),
+  uuid: v.string(),
+  stdio: StdioSchema,
+});
+
 export const StdinMessageSchema = v.object({
   type: v.literal("STDIN"),
   uuid: v.string(),
@@ -104,6 +111,7 @@ export const ProviderMessageSchema = v.union([
   ActionMessageSchema,
   SaveMessageSchema,
   ViewMessageSchema,
+  CopyMessageSchema,
   StdinMessageSchema,
   SetTimeLimitSchema,
   SetMemoryLimitSchema,

src/shared/stress-messages.ts

@@ -72,6 +72,7 @@ export const ProviderMessageTypeValues = [
   "RUN",
   "STOP",
   "VIEW",
+  "COPY",
   "ADD",
   "OPEN",
   "CLEAR",
@@ -102,6 +103,12 @@ export const ViewMessageSchema = v.object({
   stdio: v.picklist(StdioValues),
 });
 
+export const CopyMessageSchema = v.object({
+  type: v.literal("COPY"),
+  id: v.picklist(StateIdValue),
+  stdio: v.picklist(StdioValues),
+});
+
 export const AddMessageSchema = v.object({
   type: v.literal("ADD"),
   id: v.picklist(StateIdValue),
@@ -135,6 +142,7 @@ export const ProviderMessageSchema = v.union([
   RunMessageSchema,
   StopMessageSchema,
   ViewMessageSchema,
+  CopyMessageSchema,
   AddMessageSchema,
   OpenMessageSchema,
   ResetMessageSchema,

src/webview/AutoresizeTextarea.svelte

@@ -15,6 +15,7 @@
     onpreedit?: () => void;
     onsave?: () => void;
     oncancel?: () => void;
+    oncopy?: () => void;
     variant?: Variant;
     actions?: Snippet;
     ctrlEnterNewline?: boolean;
@@ -31,6 +32,7 @@
     onpreedit,
     onsave,
     oncancel,
+    oncopy,
     variant = "default",
     actions,
     ctrlEnterNewline = false,
@@ -39,9 +41,9 @@
   let textarea: HTMLTextAreaElement | undefined = $state();
   let containerElement: HTMLDivElement | undefined = $state();
   let actionButtonsElement: HTMLDivElement | undefined = $state();
-  let isHovered = $state();
+  let isHovered = $state(false);
   let cursorOverlapsActions = $state(false);
-  let showExpandButton = $state(false);
+  let showHoverButtons = $state(false);
   let showEditButtons = $state(false);
 
   let canvasContext: CanvasRenderingContext2D | null = null;
@@ -158,6 +160,36 @@
     onexpand?.();
   }
 
+  async function handleCopy() {
+    if (!editing && oncopy) {
+      oncopy();
+      return;
+    }
+
+    try {
+      await navigator.clipboard.writeText(value);
+    } catch {
+      if (!textarea) {
+        const fallbackTextarea = document.createElement("textarea");
+        fallbackTextarea.value = value;
+        fallbackTextarea.style.position = "fixed";
+        fallbackTextarea.style.opacity = "0";
+        document.body.appendChild(fallbackTextarea);
+        fallbackTextarea.focus();
+        fallbackTextarea.select();
+        document.execCommand("copy");
+        document.body.removeChild(fallbackTextarea);
+        return;
+      }
+
+      const { selectionStart, selectionEnd } = textarea;
+      textarea.select();
+      document.execCommand("copy");
+      textarea.selectionStart = selectionStart;
+      textarea.selectionEnd = selectionEnd;
+    }
+  }
+
   function handleBlur(event: FocusEvent) {
     const relatedTarget = event.relatedTarget as HTMLElement;
     if (relatedTarget?.closest(".action-buttons")) {
@@ -191,12 +223,12 @@
     if (containerElement) {
       const handleMouseEnter = () => {
         isHovered = true;
-        showExpandButton = !editing && !!onexpand;
+        showHoverButtons = !editing;
       };
       const handleMouseLeave = () => {
         isHovered = false;
         setTimeout(() => {
-          if (!isHovered) showExpandButton = false;
+          if (!isHovered) showHoverButtons = false;
         }, 200);
       };
 
@@ -213,7 +245,7 @@
   $effect(() => {
     if (editing) {
       showEditButtons = true;
-      showExpandButton = false;
+      showHoverButtons = false;
     } else {
       setTimeout(() => {
         if (!editing) showEditButtons = false;
@@ -265,8 +297,17 @@
       ></textarea>
     {/if}
     {#if !editing}
-      <div class="action-buttons" class:has-buttons={showExpandButton || actions}>
-        {#if showExpandButton && onexpand}
+      <div class="action-buttons" class:has-buttons={showHoverButtons || actions}>
+        {#if showHoverButtons}
+          <button
+            type="button"
+            data-tooltip="Copy"
+            aria-label="Copy"
+            class="action-button codicon codicon-copy"
+            onclick={handleCopy}
+          ></button>
+        {/if}
+        {#if showHoverButtons && onexpand}
           <button
             type="button"
             data-tooltip="Expand"
@@ -280,10 +321,20 @@
     {/if}
     {#if showEditButtons}
       <div
-        class="action-buttons has-buttons"
+        class="action-buttons"
+        class:has-buttons={isHovered || !!oncancel || !!onsave}
         bind:this={actionButtonsElement}
         class:overlapped={cursorOverlapsActions}
       >
+        {#if isHovered}
+          <button
+            type="button"
+            data-tooltip="Copy"
+            aria-label="Copy"
+            class="action-button codicon codicon-copy"
+            onclick={handleCopy}
+          ></button>
+        {/if}
         {#if oncancel}
           <button
             type="button"