omdev 0.0.0.dev80__tar.gz → 0.0.0.dev82__tar.gz

{omdev-0.0.0.dev80/omdev.egg-info → omdev-0.0.0.dev82}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omdev
-Version: 0.0.0.dev80
+Version: 0.0.0.dev82
 Summary: omdev
 Author: wrmsr
 License: BSD-3-Clause
@@ -12,7 +12,7 @@ Classifier: Operating System :: OS Independent
 Classifier: Operating System :: POSIX
 Requires-Python: ~=3.12
 License-File: LICENSE
-Requires-Dist: omlish==0.0.0.dev80
+Requires-Dist: omlish==0.0.0.dev82
 Provides-Extra: all
 Requires-Dist: black~=24.10; extra == "all"
 Requires-Dist: pycparser~=2.22; extra == "all"

{omdev-0.0.0.dev80 → omdev-0.0.0.dev82}/omdev/.manifests.json

@@ -167,6 +167,18 @@
       }
     }
   },
+  {
+    "module": ".scripts.slowcat",
+    "attr": "_CLI_MODULE",
+    "file": "omdev/scripts/slowcat.py",
+    "line": 113,
+    "value": {
+      "$.cli.types.CliModule": {
+        "cmd_name": "slowcat",
+        "mod_name": "omdev.scripts.slowcat"
+      }
+    }
+  },
   {
     "module": ".tools.dockertools",
     "attr": "_CLI_MODULE",

omdev-0.0.0.dev82/omdev/clipboard/clipboard.py

@@ -0,0 +1,23 @@
+"""
+https://askubuntu.com/questions/11925/a-command-line-clipboard-copy-and-paste-utility
+"""
+import abc
+import subprocess
+
+
+class Clipboard(abc.ABC):
+    @abc.abstractmethod
+    def get(self) -> str:
+        raise NotImplementedError
+
+    @abc.abstractmethod
+    def put(self, s: str) -> None:
+        raise NotImplementedError
+
+
+class DarwinClipboard(Clipboard):
+    def get(self) -> str:
+        return subprocess.check_output(['pbpaste']).decode('utf-8')
+
+    def put(self, s: str) -> None:
+        subprocess.run(['pbcopy'], input=s.encode('utf-8'), check=True)

omdev-0.0.0.dev82/omdev/clipboard/darwin.py

@@ -0,0 +1,229 @@
+# ruff: noqa: N802 N816
+import ctypes as ct
+import ctypes.util
+import dataclasses as dc
+import sys
+import typing as ta
+
+from omlish import check
+
+
+##
+
+
+if getattr(sys, 'platform') != 'darwin':
+    raise OSError(sys.platform)
+
+
+##
+# CoreFoundation
+
+cf = ct.cdll.LoadLibrary(check.not_none(ct.util.find_library('CoreFoundation')))
+
+#
+
+CFArrayRef = ct.c_void_p
+CFDataRef = ct.c_void_p
+CFIndex = ct.c_long
+CFStringEncoding = ct.c_uint32
+CFStringRef = ct.c_void_p
+CFTypeID = ct.c_ulong
+
+#
+
+cf.CFArrayGetCount.argtypes = [CFArrayRef]
+cf.CFArrayGetCount.restype = CFIndex
+
+cf.CFArrayGetValueAtIndex.argtypes = [CFArrayRef, CFIndex]
+cf.CFArrayGetValueAtIndex.restype = CFStringRef
+
+cf.CFDataGetBytePtr.argtypes = [CFDataRef]
+cf.CFDataGetBytePtr.restype = ct.POINTER(ct.c_uint8)
+
+cf.CFDataGetLength.argtypes = [CFDataRef]
+cf.CFDataGetLength.restype = CFIndex
+
+cf.CFGetTypeID.argtypes = [ct.c_void_p]
+cf.CFGetTypeID.restype = CFTypeID
+
+cf.CFRelease.argtypes = [ct.c_void_p]
+cf.CFRelease.restype = None
+
+cf.CFStringCreateWithCString.argtypes = [ct.c_void_p, ct.c_char_p, ct.c_int32]
+cf.CFStringCreateWithCString.restype = CFStringRef
+
+cf.CFStringGetCString.argtypes = [CFStringRef, ct.c_char_p, ct.c_long, ct.c_uint32]
+cf.CFStringGetCString.restype = ct.c_bool
+
+cf.CFStringGetLength.argtypes = [CFStringRef]
+cf.CFStringGetLength.restype = CFIndex
+
+cf.CFStringGetMaximumSizeForEncoding.argtypes = [CFIndex, CFStringEncoding]
+cf.CFStringGetMaximumSizeForEncoding.restype = CFIndex
+
+cf.CFStringGetTypeID.argtypes = []
+cf.CFStringGetTypeID.restype = CFTypeID
+
+
+##
+# ApplicationServices
+
+aps = ct.cdll.LoadLibrary(check.not_none(ct.util.find_library('ApplicationServices')))
+
+#
+
+OSStatus = ct.c_int32
+
+PasteboardItemID = ct.c_ulong
+PasteboardRef = ct.c_void_p
+
+#
+
+aps.PasteboardCopyItemFlavorData.argtypes = [PasteboardRef, PasteboardItemID, CFStringRef, ct.POINTER(CFDataRef)]
+aps.PasteboardCopyItemFlavorData.restype = OSStatus
+
+aps.PasteboardCopyItemFlavors.argtypes = [PasteboardRef, PasteboardItemID, ct.POINTER(CFArrayRef)]
+aps.PasteboardCopyItemFlavors.restype = OSStatus
+
+aps.PasteboardCreate.argtypes = [CFStringRef, ct.POINTER(PasteboardRef)]
+aps.PasteboardCreate.restype = OSStatus
+
+aps.PasteboardGetItemCount.argtypes = [PasteboardRef, ct.POINTER(ct.c_ulong)]
+aps.PasteboardGetItemCount.restype = OSStatus
+
+aps.PasteboardGetItemIdentifier.argtypes = [PasteboardRef, ct.c_ulong, ct.POINTER(PasteboardItemID)]
+aps.PasteboardGetItemIdentifier.restype = OSStatus
+
+
+##
+
+
+def CFSTR(string):
+    return cf.CFStringCreateWithCString(None, string.encode('utf-8'), 0)
+
+
+kCFStringEncodingUTF8 = 0x08000100
+kPasteboardClipboard = CFSTR('com.apple.pasteboard.clipboard')
+
+
+def cfstring_to_string(cf_string: CFStringRef) -> str:
+    if not cf_string:
+        return ''
+
+    length = cf.CFStringGetLength(cf_string)
+    max_size = cf.CFStringGetMaximumSizeForEncoding(length, kCFStringEncodingUTF8) + 1
+    buffer = ct.create_string_buffer(max_size)
+
+    if success := cf.CFStringGetCString(cf_string, buffer, max_size, kCFStringEncodingUTF8):  # noqa
+        return ''
+
+    return buffer.value.decode('utf-8')
+
+
+##
+
+
+class OsxClipboardError(Exception):
+    pass
+
+
+@dc.dataclass(frozen=True)
+class StatusOsxClipboardError(OsxClipboardError):
+    fn: str
+    status: int
+
+
+@dc.dataclass(frozen=True)
+class OsxClipboardItem:
+    type: str | None
+    data: bytes | None
+
+
+def get_osx_clipboard_data(
+        *,
+        types: ta.Container[str | None] | None = None,
+        strict: bool = False,
+        types_only: bool = False,
+) -> list[OsxClipboardItem]:
+    lst: list[OsxClipboardItem] = []
+
+    pasteboard = PasteboardRef()
+    if status := aps.PasteboardCreate(kPasteboardClipboard, ct.byref(pasteboard)):
+        raise StatusOsxClipboardError('PasteboardCreate', status)
+
+    try:
+        item_count = ct.c_ulong(0)
+        if status := aps.PasteboardGetItemCount(pasteboard, ct.byref(item_count)):
+            raise StatusOsxClipboardError('PasteboardGetItemCount', status)
+
+        for i in range(1, item_count.value + 1):
+            item_id = PasteboardItemID()
+            if status := aps.PasteboardGetItemIdentifier(pasteboard, i, ct.byref(item_id)):
+                raise StatusOsxClipboardError('PasteboardGetItemIdentifier', status)
+
+            data_types = CFArrayRef()
+            if status := aps.PasteboardCopyItemFlavors(pasteboard, item_id, ct.byref(data_types)):
+                raise StatusOsxClipboardError('PasteboardCopyItemFlavors', status)
+            if not data_types:
+                continue
+
+            try:
+                type_count = cf.CFArrayGetCount(data_types)
+                for j in range(type_count):
+                    data_type = cf.CFArrayGetValueAtIndex(data_types, j)
+
+                    if cf.CFGetTypeID(data_type) == cf.CFStringGetTypeID():
+                        data_type_str = cfstring_to_string(data_type)
+                    else:
+                        data_type_str = None
+
+                    if types is not None and data_type_str not in types:
+                        continue
+
+                    if types_only:
+                        lst.append(OsxClipboardItem(
+                            type=data_type_str,
+                            data=None,
+                        ))
+                        continue
+
+                    data = CFDataRef()
+                    if status := aps.PasteboardCopyItemFlavorData(pasteboard, item_id, data_type, ct.byref(data)):
+                        if not strict:
+                            continue
+                        raise StatusOsxClipboardError('PasteboardCopyItemFlavorData', status)
+                    if not data:
+                        continue
+
+                    try:
+                        data_size = cf.CFDataGetLength(data)
+                        data_ptr = cf.CFDataGetBytePtr(data)
+                        data_bytes = ct.string_at(data_ptr, data_size)
+
+                        lst.append(OsxClipboardItem(
+                            type=data_type_str,
+                            data=data_bytes,
+                        ))
+
+                    finally:
+                        cf.CFRelease(data)
+
+            finally:
+                cf.CFRelease(data_types)
+
+    finally:
+        cf.CFRelease(pasteboard)
+
+    return lst
+
+
+##
+
+
+def _main() -> None:
+    for i in get_osx_clipboard_data():
+        print(i)
+
+
+if __name__ == '__main__':
+    _main()

omdev-0.0.0.dev82/omdev/scripts/slowcat.py

@@ -0,0 +1,121 @@
+#!/usr/bin/env python3
+# @omlish-script
+import argparse
+import contextlib
+import os
+import random
+import sys
+import time
+import typing as ta
+
+
+T = ta.TypeVar('T')
+
+
+DEFAULT_SLEEP_N = 128
+DEFAULT_SLEEP_S = .5
+
+
+def _main() -> None:
+    parser = argparse.ArgumentParser()
+
+    parser.add_argument('files', nargs='*')
+
+    parser.add_argument('-b', '--buffer-size', type=int, default=0x4000)
+
+    parser.add_argument('--initial-sleep')
+
+    parser.add_argument('--sleep-n')
+    parser.add_argument('--sleep-s')
+
+    args = parser.parse_args()
+
+    #
+
+    def parse_range(s: str, d: T) -> tuple[T, T]:
+        if not s:
+            return d, d
+        elif '-' in s:
+            l, r = map(type(d), s.split('-'))
+            return l, r
+        else:
+            l = r = type(d)(s)  # type: ignore
+            return l, r
+
+    sleep_s_min, sleep_s_max = parse_range(args.sleep_s, DEFAULT_SLEEP_S)
+    sleep_n_min, sleep_n_max = parse_range(args.sleep_n, DEFAULT_SLEEP_N)
+
+    #
+
+    in_files: list
+    if args.files:
+        in_files = args.files
+    else:
+        in_files = [sys.stdin.buffer]
+
+    #
+
+    of = sys.stdout.buffer
+
+    #
+
+    def next_sleep() -> int:
+        if sleep_n_min != sleep_n_max:
+            o = random.randint(sleep_n_min, sleep_n_max)
+        else:
+            o = sleep_n_min
+        return n + o
+
+    def do_sleep() -> None:
+        if sleep_s_min != sleep_s_max:
+            s = random.uniform(sleep_s_min, sleep_s_max)
+        else:
+            s = sleep_s_min
+        time.sleep(s)
+
+    #
+
+    if args.initial_sleep:
+        do_sleep()
+
+    n = 0
+    ns = next_sleep()
+
+    for in_file in in_files:
+        with contextlib.ExitStack() as es:
+            if isinstance(in_file, str):
+                fo = es.enter_context(open(in_file, 'rb'))
+            else:
+                fo = in_file
+
+            fd = fo.fileno()
+            while buf := os.read(fd, args.buffer_size):
+                p = 0
+                while p < len(buf):
+                    c = ns - n
+                    r = len(buf) - p
+                    if r >= c:
+                        r = c
+                        sleep = True
+                    else:
+                        sleep = False
+
+                    of.write(buf[p:p + r])
+                    of.flush()
+
+                    n += len(buf)
+                    p += r
+                    if sleep:
+                        do_sleep()
+                        ns = next_sleep()
+
+
+# @omlish-manifest
+_CLI_MODULE = {'$omdev.cli.types.CliModule': {
+    'cmd_name': 'slowcat',
+    'mod_name': __name__,
+}}
+
+
+if __name__ == '__main__':
+    _main()

{omdev-0.0.0.dev80 → omdev-0.0.0.dev82/omdev.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omdev
-Version: 0.0.0.dev80
+Version: 0.0.0.dev82
 Summary: omdev
 Author: wrmsr
 License: BSD-3-Clause
@@ -12,7 +12,7 @@ Classifier: Operating System :: OS Independent
 Classifier: Operating System :: POSIX
 Requires-Python: ~=3.12
 License-File: LICENSE
-Requires-Dist: omlish==0.0.0.dev80
+Requires-Dist: omlish==0.0.0.dev82
 Provides-Extra: all
 Requires-Dist: black~=24.10; extra == "all"
 Requires-Dist: pycparser~=2.22; extra == "all"

{omdev-0.0.0.dev80 → omdev-0.0.0.dev82}/omdev.egg-info/SOURCES.txt

@@ -76,6 +76,9 @@ omdev/cli/install.py
 omdev/cli/main.py
 omdev/cli/managers.py
 omdev/cli/types.py
+omdev/clipboard/__init__.py
+omdev/clipboard/clipboard.py
+omdev/clipboard/darwin.py
 omdev/interp/__init__.py
 omdev/interp/__main__.py
 omdev/interp/cli.py
@@ -119,6 +122,7 @@ omdev/scripts/exectime.py
 omdev/scripts/importtrace.py
 omdev/scripts/interp.py
 omdev/scripts/pyproject.py
+omdev/scripts/slowcat.py
 omdev/toml/__init__.py
 omdev/toml/parser.py
 omdev/toml/writer.py

{omdev-0.0.0.dev80 → omdev-0.0.0.dev82}/omdev.egg-info/requires.txt

@@ -1,4 +1,4 @@
-omlish==0.0.0.dev80
+omlish==0.0.0.dev82
 
 [all]
 black~=24.10

{omdev-0.0.0.dev80 → omdev-0.0.0.dev82}/pyproject.toml

@@ -12,7 +12,7 @@ authors = [
 urls = {source = 'https://github.com/wrmsr/omlish'}
 license = {text = 'BSD-3-Clause'}
 requires-python = '~=3.12'
-version = '0.0.0.dev80'
+version = '0.0.0.dev82'
 classifiers = [
     'License :: OSI Approved :: BSD License',
     'Development Status :: 2 - Pre-Alpha',
@@ -22,7 +22,7 @@ classifiers = [
 ]
 description = 'omdev'
 dependencies = [
-    'omlish == 0.0.0.dev80',
+    'omlish == 0.0.0.dev82',
 ]
 
 [project.optional-dependencies]