import json
import os
import socket
import struct
from pathlib import Path

ENV = os.environ.copy()

def ipc(method, data=None):
    with socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) as sock:
        sock.settimeout(10)
        sock.connect(ENV['WAYFIRE_SOCKET'])
        payload = json.dumps({'method': method, 'data': data or {}}).encode()
        sock.sendall(struct.pack('<I', len(payload)) + payload)
        def read(n):
            result = b''
            while len(result) < n:
                chunk = sock.recv(n - len(result))
                if not chunk:
                    raise RuntimeError('IPC disconnected')
                result += chunk
            return result
        return json.loads(read(struct.unpack('<I', read(4))[0]))

if __name__ == '__main__':
    import sys
    print(json.dumps(ipc(sys.argv[1], json.loads(sys.argv[2]) if len(sys.argv) > 2 else None), indent=2))