omdev 0.0.0.dev80__py3-none-any.whl → 0.0.0.dev82__py3-none-any.whl

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/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/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/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.dist-info → omdev-0.0.0.dev82.dist-info}/METADATA

@@ -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.dist-info → omdev-0.0.0.dev82.dist-info}/RECORD

@@ -1,4 +1,4 @@
-omdev/.manifests.json,sha256=TmB_ctSLXsOq5UGeoiGLy5FoI490OCjT2Q2kRWyWG9g,5993
+omdev/.manifests.json,sha256=eAMErpBPPY0YM3fBzmtDzhtOO1SEvG3iYBf2jZi3LNU,6257
 omdev/__about__.py,sha256=do_MjpnIpB_DX4GL4VV9gFQzs6bkoyDIYu29yB3rYxw,1131
 omdev/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 omdev/bracepy.py,sha256=I8EdqtDvxzAi3I8TuMEW-RBfwXfqKbwp06CfOdj3L1o,2743
@@ -66,6 +66,9 @@ omdev/cli/install.py,sha256=C-W171YlIHt4Cfok-nWSMbHwWhqF_PFqq2HixFttYx8,4460
 omdev/cli/main.py,sha256=nD3bW3MkI0KuDSLpX_PTnl-lBnm4hIxVqczvvMHtupo,7058
 omdev/cli/managers.py,sha256=BV98_n30Jj63OJrFgRoVZRfICxMLXEZKoEn4rMj9LV4,1160
 omdev/cli/types.py,sha256=bqKw9SbtBtAip2vF9v4khh0CqKG6LBr6n9VzWBz7AJE,474
+omdev/clipboard/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+omdev/clipboard/clipboard.py,sha256=CmnozkRljJ9UQDHLfCS8wKU0Xg-HfbsAkagAfMEias0,567
+omdev/clipboard/darwin.py,sha256=AE4oYrDXTV0uBgZ-8pWxtqXes3k0IL2pxrJN8WjK1C4,6370
 omdev/interp/__init__.py,sha256=Y3l4WY4JRi2uLG6kgbGp93fuGfkxkKwZDvhsa0Rwgtk,15
 omdev/interp/__main__.py,sha256=GMCqeGYltgt5dlJzHxY9gqisa8cRkrPfmZYuZnjg4WI,162
 omdev/interp/cli.py,sha256=sh7PZQoLletUViw1Y9OXNr9ekyNZ6YyxYuOQ_n9hyqU,2072
@@ -109,6 +112,7 @@ omdev/scripts/exectime.py,sha256=dBdn3KV2jR6tCrzGvb9dTl2uGe2vNpLxmHMtExmnaiM,410
 omdev/scripts/importtrace.py,sha256=Jbo3Yk2RAbE8_tJ97iTcVNpoxCJxrRb2tl1W_CV3NG0,14067
 omdev/scripts/interp.py,sha256=2poCQ_45tO0-X16KeoxiAyuTfltSjnPETejlYEm9VY4,71527
 omdev/scripts/pyproject.py,sha256=wtmF5ji8fHw2qzc1aEve09Ozv0ncuZmTQ8IbKCJVAQ0,158223
+omdev/scripts/slowcat.py,sha256=Rglbg0TZ6GmcJgePxN2Bq9JQChZyvQK_qExOYWajaYE,2601
 omdev/toml/__init__.py,sha256=Y3l4WY4JRi2uLG6kgbGp93fuGfkxkKwZDvhsa0Rwgtk,15
 omdev/toml/parser.py,sha256=84bn09uhYHwQGyfww6Rw6y1RxPAE_HDltODOSakcqDM,29186
 omdev/toml/writer.py,sha256=lk3on3YXVbWuLJa-xsOzOhs1bBAT1vXqw4mBbluZl_w,3040
@@ -122,9 +126,9 @@ omdev/tools/piptools.py,sha256=-jR5q3w4sHqntxCLExFCBNIARB788FUsAbJ62PK2sBU,2774
 omdev/tools/proftools.py,sha256=8ZU9x_Dq8eT2ZFwU9sJpDIvxcIn9qBc8y2ELKPb5e5M,1382
 omdev/tools/rsttool.py,sha256=suwsfseUf8GH8rYmYygTUdif-Jk_bX1g9fYRLXaKkmM,1340
 omdev/tools/sqlrepl.py,sha256=tmFZh80-xsGM62dyQ7_UGLebChrj7IHbIPYBWDJMgVk,5741
-omdev-0.0.0.dev80.dist-info/LICENSE,sha256=B_hVtavaA8zCYDW99DYdcpDLKz1n3BBRjZrcbv8uG8c,1451
-omdev-0.0.0.dev80.dist-info/METADATA,sha256=68sUTkl9Uc5f4dvq8nA5q3OQ7eDcDDh0MPClMBFBAps,1492
-omdev-0.0.0.dev80.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
-omdev-0.0.0.dev80.dist-info/entry_points.txt,sha256=dHLXFmq5D9B8qUyhRtFqTGWGxlbx3t5ejedjrnXNYLU,33
-omdev-0.0.0.dev80.dist-info/top_level.txt,sha256=1nr7j30fEWgLYHW3lGR9pkdHkb7knv1U1ES1XRNVQ6k,6
-omdev-0.0.0.dev80.dist-info/RECORD,,
+omdev-0.0.0.dev82.dist-info/LICENSE,sha256=B_hVtavaA8zCYDW99DYdcpDLKz1n3BBRjZrcbv8uG8c,1451
+omdev-0.0.0.dev82.dist-info/METADATA,sha256=D6gJ4sHSoH-ucg-IsG2pk8Ft_XYv1bhTxObxFAdrLIU,1492
+omdev-0.0.0.dev82.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
+omdev-0.0.0.dev82.dist-info/entry_points.txt,sha256=dHLXFmq5D9B8qUyhRtFqTGWGxlbx3t5ejedjrnXNYLU,33
+omdev-0.0.0.dev82.dist-info/top_level.txt,sha256=1nr7j30fEWgLYHW3lGR9pkdHkb7knv1U1ES1XRNVQ6k,6
+omdev-0.0.0.dev82.dist-info/RECORD,,