rpbcopy.py

#!/usr/bin/env python

import contextlib
import io
import os
import sys


inside_tmux = "TMUX" in os.environ


def escape_tmux(msg):
    if inside_tmux:
        rv = io.BytesIO()
        while msg:
            chunk, msg = msg[:100], msg[100:]

            rv.write("{}{}{}".format(
                "\033Ptmux;\033",
                chunk.replace("\033", "\033\033"),
                "\033\\",
            ))

        return rv.getvalue()
    else:
        return msg


def iterm_copy(msg):
    return "{}{}{}".format(
        "\033]1337;CopyToClipboard=\007",
        msg,
        "\033]1337;EndCopy\007",
    )


output = sys.stdin.read()
output = iterm_copy(output)
output = escape_tmux(output)
sys.stdout.write(output)
sys.stdout.flush()