Script is almost finished

Slackadays · Oct 16, 2024 · 916616f · 916616f
1 parent 0c1dbb2
commit 916616f
Show file tree

Hide file tree

Showing 4 changed files with 50 additions and 6 deletions.
diff --git a/src/cb/src/actions/script.cpp b/src/cb/src/actions/script.cpp
@@ -25,27 +25,66 @@ void script() {
         if (copying.items.empty()) {
             stopIndicator();
             if (fs::is_regular_file(path.metadata.script)) {
-                fprintf(stderr, formatColors("[info]┃ The current script is: [help]%s[blank]\n").data(), fileContents(path.metadata.script).value().data());
+                fprintf(stderr, formatColors("[info]┃ Here is this clipboard's current script: [help]%s[blank]\n").data(), fileContents(path.metadata.script).value().data());
             } else {
 
-                fprintf(stderr, "%s", formatColors("[error][inverse] ✘ [noinverse] No script is currently set. [help]⬤ Try providing a script file to set one.[blank]\n").data());
+                fprintf(stderr, formatColors("[error][inverse] ✘ [noinverse] There is currently no script set for this clipboard. [help]⬤ To set a script, add it to the end, like [bold]%s %s myscript.sh[nobold], or specify it as an argument, like [bold]%s %s \"echo Hello World!\".[blank]\n").data(), clipboard_invocation.data(), actions[action].data(), clipboard_invocation.data(), actions[action].data());
             }
             return;
         }
+        fs::remove(path.metadata.script);
+        fs::copy(copying.items.at(0), path.metadata.script);
+        fs::permissions(path.metadata.script, fs::perms::owner_exec, fs::perm_options::add);
+        if (output_silent || confirmation_silent) return;
+        stopIndicator();
+        fprintf(stderr, formatColors("[success][inverse] ✔ [noinverse] Saved script \"%s\"[blank]\n").data(), fileContents(path.metadata.script).value().data());
+    } else if (io_type == IOType::Text) {
         if (copying.items.at(0).string() == "") {
             fs::remove(path.metadata.script);
             if (output_silent || confirmation_silent) return;
             stopIndicator();
             fprintf(stderr, "%s", formatColors("[success][inverse] ✔ [noinverse] Removed script[blank]\n").data());
         } else {
             fs::remove(path.metadata.script);
-            fs::copy(copying.items.at(0), path.metadata.script);
+            writeToFile(path.metadata.script, copying.items.at(0).string());
+            fs::permissions(path.metadata.script, fs::perms::owner_exec, fs::perm_options::add);
             if (output_silent || confirmation_silent) return;
             stopIndicator();
             fprintf(stderr, formatColors("[success][inverse] ✔ [noinverse] Saved script \"%s\"[blank]\n").data(), fileContents(path.metadata.script).value().data());
         }
-    } else if (io_type == IOType::Text) {
     }
 }
 
-} // namespace PerformAction
+} // namespace PerformAction
+
+void runClipboardScript() {
+    if (!fs::is_regular_file(path.metadata.script)) return;
+
+    static bool secondRun = false;
+
+    fs::path currentPath = fs::current_path();
+
+    fs::current_path(path.data);
+
+    auto execute = [&](const std::string_view& timing) {
+        // Set the CLIPBOARD_ACTION environment variable to the action that was performed
+        int res = setenv("CLIPBOARD_ACTION", actions[action].data(), 1);
+        if (res != 0) throw std::runtime_error("Failed to set the CLIPBOARD_ACTION environment variable");
+
+        // Set the CLIPBOARD_SCRIPT_TIMING environment variable to "before" or "after" depending on the timing
+        res = setenv("CLIPBOARD_SCRIPT_TIMING", timing.data(), 1);
+        if (res != 0) throw std::runtime_error("Failed to set the CLIPBOARD_SCRIPT_TIMING environment variable"); 
+
+        res = system(path.metadata.script.string().c_str());
+        if (res != 0) throw std::runtime_error("Failed to run the clipboard script");
+    };
+
+    if (!secondRun)
+        execute("before");
+    else
+        execute("after");
+
+    fs::current_path(currentPath);
+
+    secondRun = true;
+}
diff --git a/src/cb/src/clipboard.hpp b/src/cb/src/clipboard.hpp
@@ -420,6 +420,7 @@ void showSuccesses();
 void setTheme(const std::string_view& theme);
 size_t totalDirectorySize(const fs::path& directory);
 size_t directoryOverhead(const fs::path& directory);
+void runClipboardScript();
 
 extern Message help_message;
 extern Message check_clipboard_status_message;

diff --git a/src/cb/src/main.cpp b/src/cb/src/main.cpp
@@ -70,8 +70,12 @@ int main(int argc, char* argv[]) {
 
         checkItemSize(totalItemSize());
 
+        runClipboardScript();
+
         performAction();
 
+        runClipboardScript();
+
         if (isAWriteAction()) path.applyIgnoreRules();
 
         copying.mime = getMIMEType();

diff --git a/src/cb/src/utils/formatting.cpp b/src/cb/src/utils/formatting.cpp
@@ -1,5 +1,5 @@
 /*  The Clipboard Project - Cut, copy, and paste anything, anytime, anywhere, all from the terminal.
-    Copyright (C) 2023 Jackson Huff and other contributors on GitHub.com
+    Copyright (C) 2024 Jackson Huff and other contributors on GitHub.com
     SPDX-License-Identifier: GPL-3.0-or-later
     This program is free software: you can redistribute it and/or modify
     it under the terms of the GNU General Public License as published by