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
|
node_modules
|
||||||
dist
|
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 { 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
|
* fetches datafrom user server, especially validates user tokens
|
||||||
*/
|
*/
|
||||||
export class UserServerAPI extends DataSource {
|
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
|
* validates user token
|
||||||
*/
|
*/
|
||||||
async validateToken (token:string) {
|
async validateToken (token:string): Promise<boolean> {
|
||||||
return true
|
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