Add RPC Client implementation
- Add Message class to serialize and deserialize rpc messages - Add Method enum for all rpc methods represented as 32 bit unsigned integers - Add responses file to map rpc responses to types - Add implementation of userserver api with PromiseSocket Signed-off-by: trivernis <trivernis@protonmail.com>pull/3/head
parent
3164404218
commit
d475851bb3
GPG Key ID:
DFFFCC2C7A02DB45
@ -1,2 +1,4 @@
|
||||
node_modules
|
||||
dist
|
||||
.env
|
||||
.idea
|
||||
|
||||
@ -0,0 +1,43 @@
|
||||
import { encode, decode } from 'messagepack'
|
||||
import { Method } from './method'
|
||||
import { crc32 } from 'crc'
|
||||
|
||||
export class RPCMessage<T> {
|
||||
private readonly method: Method
|
||||
readonly data: T
|
||||
|
||||
constructor (method: Method, data: any) {
|
||||
this.method = method
|
||||
this.data = data
|
||||
}
|
||||
|
||||
static fromBuffer<T> (raw: Buffer): RPCMessage<T> {
|
||||
const length = raw.readUInt32BE()
|
||||
if (raw.length !== length) {
|
||||
throw new Error('Invalid Buffer length')
|
||||
}
|
||||
const crcNum = raw.readUInt32BE(length - 4)
|
||||
|
||||
if (crc32(raw.slice(0, length - 4)) !== crcNum) {
|
||||
throw new Error('Validation check failed')
|
||||
}
|
||||
const method = raw.readUInt32BE(4)
|
||||
const msgData = decode(raw.slice(8, length))
|
||||
|
||||
return new RPCMessage(method, msgData)
|
||||
}
|
||||
|
||||
public toBuffer (): Buffer {
|
||||
const msgData = encode(this.data)
|
||||
const length = msgData.length + 12
|
||||
const buffer = Buffer.alloc(length - 4)
|
||||
buffer.writeUInt32BE(length)
|
||||
buffer.writeUInt32BE(this.method, 4)
|
||||
buffer.fill(msgData, 8)
|
||||
const crcNum = crc32(buffer)
|
||||
const resultBuffer = Buffer.alloc(length, buffer)
|
||||
resultBuffer.writeUInt32BE(crcNum, length - 4)
|
||||
|
||||
return resultBuffer
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,11 @@
|
||||
/* eslint no-unused-vars: 0 */
|
||||
export enum Method {
|
||||
Null = 0x00,
|
||||
Error = 0x0F_0F_0F_0F,
|
||||
Info = 0x49_4e_46_4f,
|
||||
ValidateToken = 0x56_41_4c_49,
|
||||
GetRoles = 0x52_4f_4c_45,
|
||||
GetRolePermissions = 0x50_45_52_4d,
|
||||
CreateRole = 0x43_52_4f_4c,
|
||||
CreatePermissionn = 0x43_50_45_52
|
||||
}
|
||||
@ -0,0 +1,5 @@
|
||||
// Response in the form of [valid, ttl]
|
||||
export type ValidateTokenResponse = [boolean, number]
|
||||
|
||||
// Info in the form of [name, binary, description, request_structure][]
|
||||
export type GetInfoResponse = [string, string, string][]
|
||||
@ -1,13 +1,64 @@
|
||||
import { DataSource } from 'apollo-datasource'
|
||||
import { Socket } from 'net'
|
||||
import { PromiseSocket } from 'promise-socket'
|
||||
import { RPCMessage } from './message'
|
||||
import { Method } from './method'
|
||||
import { GetInfoResponse, ValidateTokenResponse } from './responses'
|
||||
|
||||
/**
|
||||
* fetches datafrom user server, especially validates user tokens
|
||||
*/
|
||||
export class UserServerAPI extends DataSource {
|
||||
port: number
|
||||
host: string
|
||||
|
||||
constructor (address: string) {
|
||||
super()
|
||||
const parts = address.split(':')
|
||||
this.host = parts[0]
|
||||
if (parts.length === 2) {
|
||||
this.port = Number(parts[1])
|
||||
} else {
|
||||
this.port = 5000
|
||||
}
|
||||
}
|
||||
|
||||
async getInfo (): Promise<GetInfoResponse> {
|
||||
const response = await this.send<GetInfoResponse>(new RPCMessage(Method.Info, null))
|
||||
|
||||
return response.data
|
||||
}
|
||||
|
||||
/**
|
||||
* validates user token
|
||||
*/
|
||||
async validateToken (token:string) {
|
||||
return true
|
||||
async validateToken (token:string): Promise<boolean> {
|
||||
const response = await this.send<ValidateTokenResponse>(new RPCMessage(Method.ValidateToken, { token }))
|
||||
if (response) {
|
||||
return response.data[0]
|
||||
} else {
|
||||
return false
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Connects to the socket and returns the client
|
||||
*/
|
||||
async getSocket (): Promise<PromiseSocket<Socket>> {
|
||||
const socket = new Socket()
|
||||
const promiseSocket = new PromiseSocket(socket)
|
||||
await promiseSocket.connect(this.port, this.host)
|
||||
|
||||
return promiseSocket
|
||||
}
|
||||
|
||||
async send<T> (message: RPCMessage<any>): Promise<RPCMessage<T>> {
|
||||
const socket = await this.getSocket()
|
||||
await socket.writeAll(message.toBuffer())
|
||||
const response = await socket.readAll() as Buffer
|
||||
|
||||
if (response?.length > 0) {
|
||||
return RPCMessage.fromBuffer<T>(response)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Loading…
Reference in New Issue