1 files changed, 163 insertions, 0 deletions
diff --git a/src/sysclip.cc b/src/sysclip.cc
new file mode 100644
index 0000000..725e404
--- /dev/null
+++ b/src/sysclip.cc
@@ -0,0 +1,163 @@
+/**
+ * Copyright (c) 2014, Timothy Stack
+ *
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ * * Redistributions of source code must retain the above copyright notice, this
+ * list of conditions and the following disclaimer.
+ * * Redistributions in binary form must reproduce the above copyright notice,
+ * this list of conditions and the following disclaimer in the documentation
+ * and/or other materials provided with the distribution.
+ * * Neither the name of Timothy Stack nor the names of its contributors
+ * may be used to endorse or promote products derived from this software
+ * without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ''AS IS'' AND ANY
+ * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+ * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ * DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE FOR ANY
+ * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+ * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
+ * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+ * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ *
+ * @file sysclip.cc
+ */
+
+#include "sysclip.hh"
+
+#include <stdio.h>
+#include <unistd.h>
+
+#include "base/injector.hh"
+#include "base/lnav_log.hh"
+#include "config.h"
+#include "fmt/format.h"
+#include "libbase64.h"
+#include "sysclip.cfg.hh"
+
+#define ANSI_OSC "\x1b]"
+
+namespace sysclip {
+
+static nonstd::optional<clipboard>
+get_commands()
+{
+    const auto& cfg = injector::get<const config&>();
+
+    for (const auto& pair : cfg.c_clipboard_impls) {
+        const auto full_cmd = fmt::format(FMT_STRING("{} > /dev/null 2>&1"),
+                                          pair.second.c_test_command);
+
+        log_debug("testing clipboard impl %s using: %s",
+                  pair.first.c_str(),
+                  full_cmd.c_str());
+        if (system(full_cmd.c_str()) == 0) {
+            log_info("detected clipboard: %s", pair.first.c_str());
+            return pair.second;
+        }
+    }
+
+    return nonstd::nullopt;
+}
+
+static int
+osc52_close(FILE* file)
+{
+    static const char ANSI_OSC_COPY_TO_CLIP[] = ANSI_OSC "52;c;";
+
+    log_debug("writing %d bytes of clipboard data using OSC 52", ftell(file));
+    write(STDOUT_FILENO, ANSI_OSC_COPY_TO_CLIP, strlen(ANSI_OSC_COPY_TO_CLIP));
+
+    base64_state b64state{};
+    base64_stream_encode_init(&b64state, 0);
+
+    fseek(file, 0, SEEK_SET);
+
+    auto done = false;
+    while (!done) {
+        char in_buffer[1024];
+        char out_buffer[2048];
+        size_t outlen = 0;
+
+        auto rc = fread(in_buffer, 1, sizeof(in_buffer), file);
+        if (rc <= 0) {
+            base64_stream_encode_final(&b64state, out_buffer, &outlen);
+            write(STDOUT_FILENO, out_buffer, outlen);
+            break;
+        }
+
+        base64_stream_encode(&b64state, in_buffer, rc, out_buffer, &outlen);
+        write(STDOUT_FILENO, out_buffer, outlen);
+    }
+
+    write(STDOUT_FILENO, "\a", 1);
+
+    fclose(file);
+
+    return 0;
+}
+
+/* XXX For one, this code is kinda crappy.  For two, we should probably link
+ * directly with X so we don't need to have xclip installed and it'll work if
+ * we're ssh'd into a box.
+ */
+Result<auto_mem<FILE>, std::string>
+open(type_t type, op_t op)
+{
+    const char* mode = op == op_t::WRITE ? "w" : "r";
+    static const auto clip_opt = sysclip::get_commands();
+
+    std::string cmd;
+
+    if (clip_opt) {
+        cmd = clip_opt.value().select(type).select(op);
+        if (cmd.empty()) {
+            log_info("configured clipboard does not support type/op");
+        }
+    } else {
+        log_info("unable to detect clipboard");
+    }
+
+    if (cmd.empty()) {
+        log_info("  ... falling back to OSC 52");
+        auto_mem<FILE> retval(osc52_close);
+
+        retval = tmpfile();
+        if (retval.in() == nullptr) {
+            return Err(
+                fmt::format(FMT_STRING("unable to open temporary file: {}"),
+                            strerror(errno)));
+        }
+
+        return Ok(std::move(retval));
+    }
+
+    switch (op) {
+        case op_t::WRITE:
+            cmd = fmt::format(FMT_STRING("{} > /dev/null 2>&1"), cmd);
+            break;
+        case op_t::READ:
+            cmd = fmt::format(FMT_STRING("{} < /dev/null 2>/dev/null"), cmd);
+            break;
+    }
+
+    auto_mem<FILE> retval(pclose);
+
+    log_debug("trying detected clipboard command: %s", cmd.c_str());
+    retval = popen(cmd.c_str(), mode);
+    if (retval.in() == nullptr) {
+        return Err(fmt::format(FMT_STRING("failed to open clipboard: {} -- {}"),
+                               cmd,
+                               strerror(errno)));
+    }
+
+    return Ok(std::move(retval));
+}
+
+}  // namespace sysclip