pste 0.2.0__py3-none-any.whl

pste/cli.py

@@ -0,0 +1,179 @@
+import argparse
+import os
+import re
+import sys
+from datetime import datetime, timedelta, timezone
+from urllib.parse import parse_qs, urlparse
+
+import requests
+
+TTL_RE = re.compile(r"^(\d+)([MmWwDdHh])$")
+_UNIT_SECONDS = {
+    "H": 3600,
+    "D": 86400,
+    "W": 7 * 86400,
+    "M": 60,
+}
+
+_TIMEOUT = 10
+_URL_ERROR = (
+    "error: PSTE_URL not set — set it to your pste bookmark URL, "
+    "e.g. https://pste.example.com/?key=your-key"
+)
+
+
+def _parse_pste_url():
+    """Return (server, api_key) from PSTE_URL, or (None, None) if unset."""
+    raw = os.environ.get("PSTE_URL", "").strip()
+    if not raw:
+        return None, None
+    parsed = urlparse(raw)
+    key = parse_qs(parsed.query).get("key", [None])[0]
+    server = f"{parsed.scheme}://{parsed.netloc}".rstrip("/")
+    return server, key
+
+
+def _parse_ttl(value: str) -> str:
+    """Parse TTL string like '7d', '2W', '48H'; return ISO8601 UTC expires_at."""
+    m = TTL_RE.match(value)
+    if not m:
+        print(f"error: invalid expire format {value!r} (expected e.g. 7d, 2W, 48H, 3M)", file=sys.stderr)
+        sys.exit(1)
+    n, unit = int(m.group(1)), m.group(2).upper()
+    dt = datetime.now(timezone.utc) + timedelta(seconds=n * _UNIT_SECONDS[unit])
+    return dt.strftime("%Y-%m-%dT%H:%M:%SZ")
+
+
+def _validate_lang(lang: str) -> str:
+    from pygments.lexers import get_lexer_by_name
+    from pygments.util import ClassNotFound
+    try:
+        get_lexer_by_name(lang)
+    except ClassNotFound:
+        print(f"error: unknown language/lexer {lang!r}", file=sys.stderr)
+        sys.exit(1)
+    return lang
+
+
+def _fetch_paste(url: str) -> int:
+    try:
+        resp = requests.get(url, timeout=_TIMEOUT)
+    except requests.exceptions.ConnectTimeout:
+        print("error: http connection timed out", file=sys.stderr)
+        return 1
+    except requests.exceptions.ReadTimeout:
+        print("error: http request timed out", file=sys.stderr)
+        return 1
+
+    if resp.status_code == 404:
+        print(f"error: paste not found", file=sys.stderr)
+        return 1
+    if resp.status_code != 200:
+        print(f"error: server returned HTTP {resp.status_code}", file=sys.stderr)
+        return 1
+
+    print(resp.text.rstrip())
+    return 0
+
+
+def _create_paste(content: str, args, server: str, api_key: str) -> int:
+    data = {"pste": content}
+    if args.single_view:
+        data["single_view"] = "1"
+    if args.expire:
+        data["expires_at"] = _parse_ttl(args.expire)
+    if args.lang is None:
+        pass  # no -l flag: no lang, no auto-detection
+    elif args.lang == "":
+        data["auto_detect"] = "1"  # -l with no value: request auto-detection
+    else:
+        data["lang"] = _validate_lang(args.lang)  # -l python: explicit lang
+
+    try:
+        resp = requests.post(
+            server + "/",
+            data=data,
+            headers={"Authorization": f"Bearer {api_key}"},
+            timeout=_TIMEOUT,
+        )
+    except requests.exceptions.ConnectTimeout:
+        print("error: http connection timed out", file=sys.stderr)
+        return 1
+    except requests.exceptions.ReadTimeout:
+        print("error: http request timed out", file=sys.stderr)
+        return 1
+
+    if resp.status_code == 401:
+        print("error: authentication failed — check your API key", file=sys.stderr)
+        return 1
+    if resp.status_code == 403:
+        print("error: API key is disabled", file=sys.stderr)
+        return 1
+    if resp.status_code != 200:
+        print(f"error: server returned HTTP {resp.status_code}: {resp.text.strip()}", file=sys.stderr)
+        return 1
+
+    print(resp.text.strip())
+    return 0
+
+
+def main() -> int:
+    parser = argparse.ArgumentParser(
+        description=(
+            "Upload text from stdin to a pste server. "
+            "If [id] is provided, the paste is fetched instead."
+        ),
+        epilog="environment variables:\n  PSTE_URL  server URL with API key, e.g. https://pste.example.com/?key=your-key",
+        formatter_class=argparse.RawTextHelpFormatter,
+    )
+    parser.add_argument("id", nargs="?", help="Fetch paste by ID or full URL")
+    parser.add_argument("-s", "--single-view", action="store_true", help="Single-view paste")
+    parser.add_argument("-e", "--expire", metavar="TTL", help="Expiry: e.g. 7D, 2W, 48H, 30M (M=minutes)")
+    parser.add_argument(
+        "-l", "--lang", "--language",
+        metavar="LEXER", nargs="?", const="", default=None,
+        help="Syntax highlighting language (omit value to auto-detect)",
+    )
+    args = parser.parse_args()
+
+    if args.id is not None:
+        # Fetch mode
+        if args.single_view or args.expire or args.lang is not None:
+            print("error: -s/-e/-l flags are not valid when fetching a paste", file=sys.stderr)
+            return 1
+
+        paste_id = args.id
+        if paste_id.startswith("http"):
+            # Strip any ?lang query string — bare GET always returns plain text
+            parsed = urlparse(paste_id)
+            clean_url = parsed._replace(query="").geturl()
+            return _fetch_paste(clean_url)
+
+        # Bare ID — need server from PSTE_URL
+        server, _ = _parse_pste_url()
+        if not server:
+            print(_URL_ERROR, file=sys.stderr)
+            return 1
+        return _fetch_paste(f"{server}/{paste_id}")
+
+    # Create mode — read from stdin
+    server, api_key = _parse_pste_url()
+    if not server or not api_key:
+        print(_URL_ERROR, file=sys.stderr)
+        return 1
+
+    try:
+        content = sys.stdin.read()
+    except UnicodeDecodeError as e:
+        print(f"error: {e}", file=sys.stderr)
+        return 2
+
+    if not content:
+        print("error: empty input", file=sys.stderr)
+        return 1
+
+    return _create_paste(content, args, server, api_key)
+
+
+if __name__ == "__main__":
+    sys.exit(main())

pste-0.2.0.dist-info/METADATA

@@ -0,0 +1,76 @@
+Metadata-Version: 2.4
+Name: pste
+Version: 0.2.0
+Summary: Command-line client for pste, a self-hosted paste server inspired by sprunge
+Project-URL: Homepage, https://github.com/crognlie/pste
+Project-URL: Repository, https://github.com/crognlie/pste
+License: MIT
+Keywords: cli,paste,pastebin,sprunge
+Requires-Python: >=3.11
+Requires-Dist: pygments>=2.17
+Requires-Dist: requests>=2.31
+Provides-Extra: test
+Requires-Dist: pytest>=8.0; extra == 'test'
+Requires-Dist: responses>=0.25; extra == 'test'
+Description-Content-Type: text/markdown
+
+# pste
+
+Command-line client for [pste-server](../server/), a self-hosted paste server inspired by [sprunge](http://sprunge.us).
+
+## Installation
+
+```bash
+pip install pste
+# or
+pipx install pste
+```
+
+## Configuration
+
+Set `PSTE_URL` to the bookmark URL printed by `pste-admin key add` on the server — the same URL you'd save in a browser to use the web form:
+
+```bash
+export PSTE_URL="https://pste.example.com/?key=your-api-key"
+```
+
+## Usage
+
+```bash
+# Create a paste from stdin
+echo "hello world" | pste
+pste < file.txt
+cat file.txt | pste
+
+# Create with options
+echo "hello" | pste -s              # single-view (deleted after first read)
+echo "hello" | pste -e 7d           # expires in 7 days
+echo "hello" | pste -e 2W           # expires in 2 weeks
+pste -l python < script.py          # explicit syntax highlighting language
+pste -l < script.py                 # auto-detect language (Pygments, >0.5 confidence)
+
+# Fetch a paste (by ID or full URL)
+pste AB1234
+pste https://pste.example.com/AB1234
+pste https://pste.example.com/AB1234?python   # ?lang is stripped; always fetches plain text
+```
+
+## Language flag behaviour
+
+| Invocation | What is sent | URL returned |
+|---|---|---|
+| `pste` (no `-l`) | nothing | plain URL |
+| `pste -l python` | `lang=python` | `https://host/AB1234?python` |
+| `pste -l` (no value) | `auto_detect=1` | `https://host/AB1234?python` if detected, else plain URL |
+
+When a `?<lang>` URL is returned, opening it in a browser shows Pygments-highlighted HTML with table line numbers (Ctrl-A selects only the code) and a Copy button. Fetching via `pste` always strips the `?lang` and returns plain text.
+
+## Expiry format
+
+`-e`/`--expire` accepts an integer followed by a unit:
+- `H` — hours
+- `D` — days
+- `W` — weeks
+- `M` — minutes
+
+Examples: `24H`, `7D`, `2W`, `30M`

pste-0.2.0.dist-info/RECORD

@@ -0,0 +1,6 @@
+pste/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pste/cli.py,sha256=hKj1vFmXqH-KW_S1RGeQ7z7qQ9SCPdo6RNv4Z9gfDwg,5786
+pste-0.2.0.dist-info/METADATA,sha256=8J3HvnGUAhZMMFkzVU5rPkT9ZTvri6yuqPPqcD2BJ9s,2349
+pste-0.2.0.dist-info/WHEEL,sha256=mffPy8wBnZQn2VnJUU5jE99KsxaSfiyMHV9Yt0aLVxs,87
+pste-0.2.0.dist-info/entry_points.txt,sha256=4sw_Ig-h57j_Db07kdV_29L3K3xPV2stpq53HcVdVTU,39
+pste-0.2.0.dist-info/RECORD,,

pste-0.2.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.30.1
+Root-Is-Purelib: true
+Tag: py3-none-any

pste-0.2.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+pste = pste.cli:main