@cero-base/rpc 0.0.1 → 0.0.2

package/README.md

@@ -4,13 +4,11 @@ RPC client/server for [@cero-base/core](../core).
 
 ## Install
 
-```
+```bash
 npm install @cero-base/rpc
 ```
 
-## Usage
-
-### Client (renderer / UI process)
+## Client
 
 ```js
 import { Client, getRpc } from '@cero-base/rpc'
@@ -19,36 +17,62 @@ const rpc = getRpc(ipcStream, spec)
 const client = new Client(rpc, schema, spec)
 await client.ready()
 
-// Profile
+// Built-in
 const profile = await client.profile.get()
-client.profile.set({ name: 'Alice' })
-
-// Rooms
 const room = await client.rooms.open(null, { name: 'My Room' })
-const messages = room.collection('messages')
-await messages.put({ id: '1', text: 'Hello' })
 
-// Devices
-const devices = await client.devices.get()
-const invite = await client.devices.invite()
+// Custom methods auto-forward via Proxy
+await client.ping({}) // forwards to rpc.ping({})
 ```
 
-### Server (worker / backend process)
+Extend with custom logic:
 
 ```js
-import { Server, getRpc } from '@cero-base/rpc'
+class ChatClient extends Client {
+  constructor(rpc) {
+    super(rpc, schema, spec)
+  }
+
+  async ping() {
+    const { time } = await this.rpc.ping({})
+    return Date.now() - time
+  }
+}
+```
+
+## Server
+
+```js
+import { Server } from '@cero-base/rpc/server'
+import { getRpc } from '@cero-base/rpc'
 
 const rpc = getRpc(ipcStream, spec)
 const server = new Server(db, rpc)
 await server.ready()
 ```
 
-### Build (generate RPC spec)
+Extend with custom handlers — define `onXxx` methods, auto-registered:
+
+```js
+class ChatServer extends Server {
+  onPing() {
+    return { time: Date.now() }
+  }
+
+  async onStats() {
+    const rooms = await this.db.rooms.list()
+    return { rooms: rooms.length }
+  }
+}
+```
+
+## Build
 
 ```js
-import { build } from '@cero-base/rpc'
+import { build } from '@cero-base/rpc/build'
 
 await build('./spec', schema)
+await build('./spec', schema, { rpc: customRpc }) // with custom RPC types
 ```
 
 ## Exports
@@ -60,7 +84,7 @@ await build('./spec', schema)
 | `Room`       | Client-side room with collections, members, invites     |
 | `Collection` | Client-side collection with put/get/del/sub             |
 | `getRpc`     | Create an RPC instance from an IPC stream + spec        |
-| `rpc`        | Register RPC types into a hyperschema build             |
+| `rpc`        | Register base RPC types into a hyperschema build        |
 | `build`      | Build spec with RPC types included                      |
 
 ## License

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cero-base/rpc",
-  "version": "0.0.1",
+  "version": "0.0.2",
   "description": "RPC client/server for cero-base",
   "files": [
     "src",
@@ -24,7 +24,7 @@
     "test": "brittle test/*.js"
   },
   "dependencies": {
-    "@cero-base/core": "^0.0.1",
+    "@cero-base/core": "^0.0.2",
     "compact-encoding": "^2.19.1",
     "framed-stream": "^1.0.1",
     "ms": "^2.1.3",

package/src/build.js

@@ -1,6 +1,10 @@
 import { build as coreBuild } from '@cero-base/core/builder'
 import { rpc } from './schema.js'
 
-export async function build(dir, schema) {
-  return coreBuild(dir, schema, { rpc })
+export async function build(dir, schema, opts = {}) {
+  const builder = (ctx) => {
+    rpc(ctx)
+    if (opts.rpc) opts.rpc(ctx)
+  }
+  return coreBuild(dir, schema, { rpc: builder })
 }

package/src/client.js

@@ -57,6 +57,15 @@ export class Client extends ReadyResource {
     }
 
     this.ready().catch(safetyCatch)
+
+    return new Proxy(this, {
+      get(target, prop) {
+        if (prop in target) return target[prop]
+        const fn = target.rpc[prop]
+        if (typeof fn === 'function') return (...args) => fn.call(target.rpc, ...args)
+        return undefined
+      }
+    })
   }
 
   async _open() {

package/src/server.js

@@ -7,6 +7,19 @@ export class Server {
     this.rpc = rpc
     this.rooms = new Map()
     this._registerHandlers()
+    this._registerCustomHandlers()
+  }
+
+  _registerCustomHandlers() {
+    let proto = Object.getPrototypeOf(this)
+    while (proto && proto !== Server.prototype) {
+      for (const key of Object.getOwnPropertyNames(proto)) {
+        if (key.startsWith('on') && key.length > 2 && typeof this[key] === 'function') {
+          this.rpc[key](this[key].bind(this))
+        }
+      }
+      proto = Object.getPrototypeOf(proto)
+    }
   }
 
   async ready() {