scrypted/server/test/rpc-python-test.ts

import child_process from 'child_process';
import net from 'net';
import path from 'path';
import type { Readable, Writable } from "stream";
import { createDuplexRpcPeer } from '../src/rpc-serializer';

async function main() {
    const server = net.createServer(client => {
        console.log('got client');
        client.on('data', b => console.log('data', b.toString()));
    });
    server.listen(6666);

    const cp = child_process.spawn('python3', [path.join(__dirname, '../python/rpc-iterator-test.py')], {
        stdio: ['pipe', 'inherit', 'inherit', 'pipe', 'pipe'],
    });

    cp.on('exit', code => console.log('exited', code))

    const rpcPeer = createDuplexRpcPeer('node', 'python', cp.stdio[3] as Readable, cp.stdio[4] as Writable);

    async function* test() {
        try {
            for (let i = 0; ; i++) {
                yield i;
            }
        }
        finally {
            console.log('closed');
        }
    }

    rpcPeer.params['test'] = test;

    // const foo = await rpcPeer.getParam('foo');
    // assert.equal(foo, 3);

    // const bar = await rpcPeer.getParam('bar');
    // console.log(bar);

    // const ticker = await rpcPeer.getParam('ticker');
    // for await (const v of ticker) {
    //     console.log(v);
    // }
    // process.exit();
}

main();