Added pybar-memsnap to repo

Author: thnikk

docs/content/ipc/_index.md

@@ -57,10 +57,10 @@ Tracks Python memory allocations. The first call starts tracing and records a ba
 
 The `top` field controls how many allocation sites are returned (default: 30).
 
-The helper script `pybar-memsnap` automates this workflow — it records a baseline, waits 5 minutes, then prints the diff:
+The helper script `pybar-memsnap` (found in `scripts/` in the repo) automates this workflow — it records a baseline, waits 5 minutes, then prints the diff:
 
 ```bash
-    pybar-memsnap
+    python3 scripts/pybar-memsnap
 ```
 
 ### Widget instance counter

scripts/pybar-memsnap

@@ -0,0 +1,92 @@
+#!/usr/bin/env python3
+"""
+Description: Measure live Python object growth over time via pybar's
+             debug IPC commands. Requires pybar to be running with --debug.
+Author: thnikk
+
+Usage:
+    pybar-memsnap           baseline + wait 5 min + diff
+    pybar-memsnap --quick   baseline + wait 1 min + diff (less accurate)
+"""
+import argparse
+import json
+import socket
+import sys
+import time
+import os
+
+SOCK = os.path.expanduser('~/.cache/pybar/pybar.sock')
+
+
+def ipc(cmd):
+    """ Send a command to pybar's IPC socket and return the response. """
+    with socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) as s:
+        s.connect(SOCK)
+        s.sendall((json.dumps(cmd) + '\n').encode())
+        data = b''
+        while True:
+            chunk = s.recv(65536)
+            if not chunk:
+                break
+            data += chunk
+            if b'\n' in data:
+                break
+    return json.loads(data.decode().strip())
+
+
+def parse_counts(lines):
+    """ Parse object count lines into a dict of {type: count}. """
+    result = {}
+    for line in lines:
+        parts = line.strip().split('x  ', 1)
+        if len(parts) == 2:
+            result[parts[1]] = int(parts[0])
+    return result
+
+
+def run(wait):
+    """ Take a baseline count, wait, then print the diff. """
+    print('Taking baseline object count...', flush=True)
+    r0 = ipc({'action': 'objcount', 'top': 200})
+    if r0.get('status') != 'ok':
+        print(f"Error: {r0.get('message', 'unknown error')}", file=sys.stderr)
+        sys.exit(1)
+    counts0 = parse_counts(r0['counts'])
+    total0 = sum(counts0.values())
+    print(f'Baseline: {total0} total objects', flush=True)
+
+    print(f'Waiting {wait}s...', flush=True)
+    time.sleep(wait)
+
+    print('Taking second object count...', flush=True)
+    r1 = ipc({'action': 'objcount', 'top': 200})
+    counts1 = parse_counts(r1['counts'])
+    total1 = sum(counts1.values())
+    print(f'After:    {total1} total objects')
+
+    diff = {}
+    for t in set(counts0) | set(counts1):
+        d = counts1.get(t, 0) - counts0.get(t, 0)
+        if d != 0:
+            diff[t] = d
+
+    print(f'\n=== Object count diff over {wait}s ===')
+    print('(positive = grew, negative = shrank)')
+    for name, delta in sorted(diff.items(), key=lambda x: -x[1])[:40]:
+        bar_len = min(abs(delta) // 10, 40)
+        bar = ('+' if delta > 0 else '-') * bar_len
+        print(f'  {delta:+6d}  {name:40s}  {bar}')
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        description='Profile pybar memory growth via debug IPC.')
+    parser.add_argument(
+        '--quick', action='store_true',
+        help='Use a 60s window instead of 300s (less accurate)')
+    args = parser.parse_args()
+    run(wait=60 if args.quick else 300)
+
+
+if __name__ == '__main__':
+    main()