diff options
Diffstat (limited to 'src/sysclip.cc')
| -rw-r--r-- | src/sysclip.cc | 163 |
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 |