sesamefilter

src/Backend.ts

@@ -25,6 +25,7 @@ export class RPCServer<
     private errorHandler: T.ErrorHandler
     private connectionHandler: T.ConnectionHandler
     private sesame? : T.SesameFunction
+    private accessFilter: T.AccessFilter<InterfaceT, SubResType>
 
     /**
      * @throws On RPC with no name
@@ -35,10 +36,12 @@ export class RPCServer<
     constructor(
         private port:number,
         private exporters: Exporters<InterfaceT, SubResType> = [],
-        conf: T.ServerConf =  {}
+        conf: T.ServerConf<InterfaceT, SubResType> =  {}
     ){
         if(!conf.visibility) this.visibility = "127.0.0.1"        
         
+        this.accessFilter = conf.accessFilter || (async () => true)
+
         if(conf.sesame){
             this.sesame = U.makeSesameFunction(conf.sesame)
         }
@@ -87,10 +90,14 @@ export class RPCServer<
     }
 
     protected initRPCs(socket:I.Socket){
-        socket.hook('info', () => rpcInfos)
-        const rpcInfos:T.ExtendedRpcInfo[] = [
-            ...this.exporters.flatMap(exporter => U.rpcHooker(socket, exporter, this.errorHandler, this.sesame))
-        ]
+        socket.hook('info', async (sesame? : string) => {
+            const rpcs = await Promise.all(this.exporters.map(async exp => {
+                const allowed = await this.accessFilter(sesame, exp)
+                if(!allowed) return []
+                return U.rpcHooker(socket, exp, this.errorHandler, this.sesame)
+            }))
+            return rpcs.flat()
+        })
     }
 
     /**

src/Frontend.ts

@@ -136,7 +136,7 @@ export class RPCSocket implements I.Socket{
             this.socket.hook(kv[0], kv[1])
         })
 
-        const info:T.ExtendedRpcInfo[] = await this.info()
+        const info:T.ExtendedRpcInfo[] = await this.info(sesame)
         info.forEach(i => {
             let f: any
             
@@ -159,9 +159,9 @@ export class RPCSocket implements I.Socket{
     /**
      * Get a list of available RPCs from the server
      */
-    public async info(){
+    public async info(sesame?:string){
         if(!this.socket) throw new Error("The socket is not connected! Use socket.connect() first")
-        return await this.socket.call('info')
+        return await this.socket.call('info', sesame)
     }
 
     /**

src/Types.ts

@@ -2,7 +2,7 @@ import * as I from "./Interfaces";
 
 export type AnyFunction = (...args:any) => any
 export type HookFunction<F extends AnyFunction = AnyFunction, SubresT = {}> = (...args: Parameters<F>) => Promise<SubscriptionResponse<SubresT> | ErrorResponse>
-
+export type AccessFilter<InterfaceT extends RPCInterface, SubresT> = (sesame:string|undefined, exporter: I.RPCExporter<RPCInterface<InterfaceT>, keyof InterfaceT, SubresT>) => Promise<boolean> 
 export type Visibility = "127.0.0.1" | "0.0.0.0"
 export type ConnectionHandler = (socket:I.Socket) => void
 export type ErrorHandler = (socket:I.Socket, error:any, rpcName: string, args: any[]) => void
@@ -17,7 +17,8 @@ export type FrontEndHandlerType = {
     'close' : () => void
 }
 
-export type ServerConf = {
+export type ServerConf<InterfaceT extends RPCInterface, SubresT> = {
+    accessFilter?: AccessFilter<InterfaceT, SubresT>
     connectionHandler?: ConnectionHandler
     errorHandler?: ErrorHandler
     closeHandler?: CloseHandler