parent
2b73b5c97d
commit
68f973afb4
@ -1,12 +1,12 @@
|
|||||||
#!/usr/bin/env node
|
#!/usr/bin/env node
|
||||||
|
|
||||||
import path from "path";
|
import path from "node:path";
|
||||||
import {version} from "../package.json";
|
import {version} from "../package.json";
|
||||||
import fs from "fs";
|
import fs from "node:fs";
|
||||||
const optimistUsageLength = 98;
|
const optimistUsageLength = 98;
|
||||||
import yargs from "yargs";
|
import yargs from "yargs";
|
||||||
import { PeerServer } from "../src";
|
import { PeerServer } from "../src";
|
||||||
import { AddressInfo } from "net";
|
import { AddressInfo } from "node:net";
|
||||||
const opts = yargs
|
const opts = yargs
|
||||||
.usage("Usage: $0")
|
.usage("Usage: $0")
|
||||||
.wrap(Math.min(optimistUsageLength, yargs.terminalWidth()))
|
.wrap(Math.min(optimistUsageLength, yargs.terminalWidth()))
|
||||||
@ -74,7 +74,7 @@ const opts = yargs
|
|||||||
default: false,
|
default: false,
|
||||||
},
|
},
|
||||||
})
|
})
|
||||||
.boolean("allow_discovery").argv;
|
.boolean("allow_discovery").parseSync();
|
||||||
|
|
||||||
process.on("uncaughtException", function (e) {
|
process.on("uncaughtException", function (e) {
|
||||||
console.error("Error: " + e);
|
console.error("Error: " + e);
|
||||||
|
@ -1,9 +1,9 @@
|
|||||||
import cors from "cors";
|
import cors from "cors";
|
||||||
import express from "express";
|
import express from "express";
|
||||||
import publicContent from "../../app.json";
|
import publicContent from "../../app.json";
|
||||||
import { IConfig } from "../config";
|
|
||||||
import { IRealm } from "../models/realm";
|
|
||||||
import PublicApi from "./v1/public";
|
import PublicApi from "./v1/public";
|
||||||
|
import type {IConfig} from "../config";
|
||||||
|
import type {IRealm} from "../models/realm";
|
||||||
|
|
||||||
export const Api = ({ config, realm }: {
|
export const Api = ({ config, realm }: {
|
||||||
config: IConfig;
|
config: IConfig;
|
||||||
|
@ -1,6 +1,6 @@
|
|||||||
import express from "express";
|
import express from "express";
|
||||||
import { IConfig } from "../../../config";
|
import type {IConfig} from "../../../config";
|
||||||
import { IRealm } from "../../../models/realm";
|
import type {IRealm} from "../../../models/realm";
|
||||||
|
|
||||||
export default ({ config, realm }: {
|
export default ({ config, realm }: {
|
||||||
config: IConfig; realm: IRealm;
|
config: IConfig; realm: IRealm;
|
||||||
|
26
src/index.ts
26
src/index.ts
@ -1,16 +1,18 @@
|
|||||||
import express from "express";
|
import express from "express";
|
||||||
import http from "http";
|
import http from "node:http";
|
||||||
import https from "https";
|
import https from "node:https";
|
||||||
import { Server } from "net";
|
import {Server as HttpServer} from "node:http";
|
||||||
|
import {Server as HttpsServer} from "node:https";
|
||||||
|
import type {Express} from 'express-serve-static-core';
|
||||||
|
|
||||||
import defaultConfig, { IConfig } from "./config";
|
import type {IConfig} from "./config";
|
||||||
import { createInstance } from "./instance";
|
import defaultConfig from "./config";
|
||||||
|
import type {PeerServerEvents} from "./instance";
|
||||||
|
import {createInstance} from "./instance";
|
||||||
|
|
||||||
type Optional<T> = {
|
export type {MessageType} from "./enums"
|
||||||
[P in keyof T]?: (T[P] | undefined);
|
|
||||||
};
|
|
||||||
|
|
||||||
function ExpressPeerServer(server: Server, options?: IConfig) {
|
function ExpressPeerServer(server: HttpsServer | HttpServer, options?: Partial<IConfig>) {
|
||||||
const app = express();
|
const app = express();
|
||||||
|
|
||||||
const newOptions: IConfig = {
|
const newOptions: IConfig = {
|
||||||
@ -31,10 +33,10 @@ function ExpressPeerServer(server: Server, options?: IConfig) {
|
|||||||
createInstance({ app, server, options: newOptions });
|
createInstance({ app, server, options: newOptions });
|
||||||
});
|
});
|
||||||
|
|
||||||
return app;
|
return app as Express & PeerServerEvents
|
||||||
}
|
}
|
||||||
|
|
||||||
function PeerServer(options: Optional<IConfig> = {}, callback?: (server: Server) => void) {
|
function PeerServer(options: Partial<IConfig> = {}, callback?: (server: HttpsServer | HttpServer) => void) {
|
||||||
const app = express();
|
const app = express();
|
||||||
|
|
||||||
let newOptions: IConfig = {
|
let newOptions: IConfig = {
|
||||||
@ -45,7 +47,7 @@ function PeerServer(options: Optional<IConfig> = {}, callback?: (server: Server)
|
|||||||
const port = newOptions.port;
|
const port = newOptions.port;
|
||||||
const host = newOptions.host;
|
const host = newOptions.host;
|
||||||
|
|
||||||
let server: Server;
|
let server: HttpsServer | HttpServer;
|
||||||
|
|
||||||
const { ssl, ...restOptions } = newOptions;
|
const { ssl, ...restOptions } = newOptions;
|
||||||
if (ssl && Object.keys(ssl).length) {
|
if (ssl && Object.keys(ssl).length) {
|
||||||
|
@ -1,20 +1,30 @@
|
|||||||
import express from "express";
|
import express from "express";
|
||||||
import { Server } from "net";
|
import {Server as HttpServer} from "node:http";
|
||||||
import path from "path";
|
import {Server as HttpsServer} from "node:https";
|
||||||
import { IClient } from "./models/client";
|
import path from "node:path";
|
||||||
import { IMessage } from "./models/message";
|
import type {IRealm} from "./models/realm";
|
||||||
import { Realm } from "./models/realm";
|
import {Realm} from "./models/realm";
|
||||||
import { IRealm } from "./models/realm";
|
import {CheckBrokenConnections} from "./services/checkBrokenConnections";
|
||||||
import { CheckBrokenConnections } from "./services/checkBrokenConnections";
|
import type {IMessagesExpire} from "./services/messagesExpire";
|
||||||
import { IMessagesExpire, MessagesExpire } from "./services/messagesExpire";
|
import {MessagesExpire} from "./services/messagesExpire";
|
||||||
import { IWebSocketServer, WebSocketServer } from "./services/webSocketServer";
|
import type {IWebSocketServer} from "./services/webSocketServer";
|
||||||
import { MessageHandler } from "./messageHandler";
|
import {WebSocketServer} from "./services/webSocketServer";
|
||||||
import { Api } from "./api";
|
import {MessageHandler} from "./messageHandler";
|
||||||
import { IConfig } from "./config";
|
import {Api} from "./api";
|
||||||
|
import type {IClient} from "./models/client";
|
||||||
|
import type {IMessage} from "./models/message";
|
||||||
|
import type {IConfig} from "./config";
|
||||||
|
|
||||||
|
export interface PeerServerEvents {
|
||||||
|
on(event: 'connection', listener: (client: IClient) => void): this;
|
||||||
|
on(event: "message", listener: (client: IClient, message: IMessage) => void): this;
|
||||||
|
on(event: "disconnect", listener: (client: IClient) => void): this;
|
||||||
|
on(event: "error", listener: (client: Error) => void): this;
|
||||||
|
}
|
||||||
|
|
||||||
export const createInstance = ({ app, server, options }: {
|
export const createInstance = ({ app, server, options }: {
|
||||||
app: express.Application;
|
app: express.Application;
|
||||||
server: Server;
|
server: HttpServer | HttpsServer;
|
||||||
options: IConfig;
|
options: IConfig;
|
||||||
}): void => {
|
}): void => {
|
||||||
const config = options;
|
const config = options;
|
||||||
|
@ -1,4 +1,4 @@
|
|||||||
import { IClient } from "../models/client";
|
import type {IClient} from "../models/client";
|
||||||
import { IMessage } from "../models/message";
|
import type {IMessage} from "../models/message";
|
||||||
|
|
||||||
export type Handler = (client: IClient | undefined, message: IMessage) => boolean;
|
export type Handler = (client: IClient | undefined, message: IMessage) => boolean;
|
||||||
|
@ -1,4 +1,4 @@
|
|||||||
import { IClient } from "../../../models/client";
|
import type {IClient} from "../../../models/client";
|
||||||
|
|
||||||
export const HeartbeatHandler = (client: IClient | undefined): boolean => {
|
export const HeartbeatHandler = (client: IClient | undefined): boolean => {
|
||||||
if (client) {
|
if (client) {
|
||||||
|
@ -1,7 +1,7 @@
|
|||||||
import { MessageType } from "../../../enums";
|
import {MessageType} from "../../../enums";
|
||||||
import { IClient } from "../../../models/client";
|
import type {IClient} from "../../../models/client";
|
||||||
import { IMessage } from "../../../models/message";
|
import type {IMessage} from "../../../models/message";
|
||||||
import { IRealm } from "../../../models/realm";
|
import type {IRealm} from "../../../models/realm";
|
||||||
|
|
||||||
export const TransmissionHandler = ({ realm }: { realm: IRealm; }): (client: IClient | undefined, message: IMessage) => boolean => {
|
export const TransmissionHandler = ({ realm }: { realm: IRealm; }): (client: IClient | undefined, message: IMessage) => boolean => {
|
||||||
const handle = (client: IClient | undefined, message: IMessage) => {
|
const handle = (client: IClient | undefined, message: IMessage) => {
|
||||||
|
@ -1,7 +1,7 @@
|
|||||||
import { MessageType } from "../enums";
|
import {MessageType} from "../enums";
|
||||||
import { IClient } from "../models/client";
|
import type {IClient} from "../models/client";
|
||||||
import { IMessage } from "../models/message";
|
import type {IMessage} from "../models/message";
|
||||||
import { Handler } from "./handler";
|
import type {Handler} from "./handler";
|
||||||
|
|
||||||
export interface IHandlersRegistry {
|
export interface IHandlersRegistry {
|
||||||
registerHandler(messageType: MessageType, handler: Handler): void;
|
registerHandler(messageType: MessageType, handler: Handler): void;
|
||||||
|
@ -1,10 +1,11 @@
|
|||||||
import { MessageType } from "../enums";
|
import {MessageType} from "../enums";
|
||||||
import { IClient } from "../models/client";
|
import {HeartbeatHandler, TransmissionHandler} from "./handlers";
|
||||||
import { IMessage } from "../models/message";
|
import type {IHandlersRegistry} from "./handlersRegistry";
|
||||||
import { IRealm } from "../models/realm";
|
import {HandlersRegistry} from "./handlersRegistry";
|
||||||
import { Handler } from "./handler";
|
import type {IClient} from "../models/client";
|
||||||
import { HeartbeatHandler, TransmissionHandler } from "./handlers";
|
import type {IMessage} from "../models/message";
|
||||||
import { IHandlersRegistry, HandlersRegistry } from "./handlersRegistry";
|
import type {IRealm} from "../models/realm";
|
||||||
|
import type {Handler} from "./handler";
|
||||||
|
|
||||||
export interface IMessageHandler {
|
export interface IMessageHandler {
|
||||||
handle(client: IClient | undefined, message: IMessage): boolean;
|
handle(client: IClient | undefined, message: IMessage): boolean;
|
||||||
|
@ -1,8 +1,8 @@
|
|||||||
import { MessageType } from "../enums";
|
import type {MessageType} from "../enums";
|
||||||
|
|
||||||
export interface IMessage {
|
export interface IMessage {
|
||||||
readonly type: MessageType;
|
readonly type: MessageType;
|
||||||
readonly src: string;
|
readonly src: string;
|
||||||
readonly dst: string;
|
readonly dst: string;
|
||||||
readonly payload?: any;
|
readonly payload?: string;
|
||||||
}
|
}
|
||||||
|
@ -1,4 +1,4 @@
|
|||||||
import { IMessage } from "./message";
|
import type {IMessage} from "./message";
|
||||||
|
|
||||||
export interface IMessageQueue {
|
export interface IMessageQueue {
|
||||||
getLastReadAt(): number;
|
getLastReadAt(): number;
|
||||||
|
@ -1,7 +1,8 @@
|
|||||||
import { IClient } from "./client";
|
import type {IMessageQueue} from "./messageQueue";
|
||||||
import { IMessage } from "./message";
|
import {MessageQueue} from "./messageQueue";
|
||||||
import { IMessageQueue, MessageQueue } from "./messageQueue";
|
import {randomUUID} from "node:crypto";
|
||||||
import {randomUUID} from "crypto";
|
import type {IClient} from "./client";
|
||||||
|
import type {IMessage} from "./message";
|
||||||
|
|
||||||
export interface IRealm {
|
export interface IRealm {
|
||||||
getClientsIds(): string[];
|
getClientsIds(): string[];
|
||||||
|
@ -1,6 +1,6 @@
|
|||||||
import { IConfig } from "../../config";
|
import type {IConfig} from "../../config";
|
||||||
import { IClient } from "../../models/client";
|
import type {IClient} from "../../models/client";
|
||||||
import { IRealm } from "../../models/realm";
|
import type {IRealm} from "../../models/realm";
|
||||||
|
|
||||||
const DEFAULT_CHECK_INTERVAL = 300;
|
const DEFAULT_CHECK_INTERVAL = 300;
|
||||||
|
|
||||||
|
@ -1,7 +1,7 @@
|
|||||||
import { IConfig } from "../../config";
|
import {MessageType} from "../../enums";
|
||||||
import { MessageType } from "../../enums";
|
import type {IConfig} from "../../config";
|
||||||
import { IMessageHandler } from "../../messageHandler";
|
import type {IMessageHandler} from "../../messageHandler";
|
||||||
import { IRealm } from "../../models/realm";
|
import type {IRealm} from "../../models/realm";
|
||||||
|
|
||||||
export interface IMessagesExpire {
|
export interface IMessagesExpire {
|
||||||
startMessagesExpiration(): void;
|
startMessagesExpiration(): void;
|
||||||
|
@ -1,12 +1,15 @@
|
|||||||
import EventEmitter from "events";
|
import {EventEmitter} from "node:events";
|
||||||
import { IncomingMessage } from "http";
|
import {IncomingMessage} from "node:http";
|
||||||
import url from "url";
|
import url from "node:url";
|
||||||
import WebSocketLib from "ws";
|
|
||||||
import { IConfig } from "../../config";
|
|
||||||
import { Errors, MessageType } from "../../enums";
|
|
||||||
import { Client, IClient } from "../../models/client";
|
|
||||||
import { IRealm } from "../../models/realm";
|
|
||||||
import type WebSocket from "ws";
|
import type WebSocket from "ws";
|
||||||
|
import * as WebSocketLib from "ws";
|
||||||
|
import {Errors, MessageType} from "../../enums";
|
||||||
|
import type {IClient} from "../../models/client";
|
||||||
|
import {Client} from "../../models/client";
|
||||||
|
import type {IConfig} from "../../config";
|
||||||
|
import type {IRealm} from "../../models/realm";
|
||||||
|
import {Server as HttpServer} from "node:http";
|
||||||
|
import {Server as HttpsServer} from "node:https";
|
||||||
|
|
||||||
export interface IWebSocketServer extends EventEmitter {
|
export interface IWebSocketServer extends EventEmitter {
|
||||||
readonly path: string;
|
readonly path: string;
|
||||||
@ -29,7 +32,7 @@ export class WebSocketServer extends EventEmitter implements IWebSocketServer {
|
|||||||
private readonly config: CustomConfig;
|
private readonly config: CustomConfig;
|
||||||
public readonly socketServer: WebSocketLib.Server;
|
public readonly socketServer: WebSocketLib.Server;
|
||||||
|
|
||||||
constructor({ server, realm, config }: { server: any; realm: IRealm; config: CustomConfig; }) {
|
constructor({ server, realm, config }: { server: HttpServer | HttpsServer; realm: IRealm; config: CustomConfig; }) {
|
||||||
super();
|
super();
|
||||||
|
|
||||||
this.setMaxListeners(0);
|
this.setMaxListeners(0);
|
||||||
|
@ -1,5 +1,6 @@
|
|||||||
import { expect } from 'chai';
|
import { expect } from 'chai';
|
||||||
import { Server, WebSocket } from 'mock-socket';
|
import { Server, WebSocket } from 'mock-socket';
|
||||||
|
import type {Server as HttpServer} from 'node:http';
|
||||||
import { Realm } from '../../../src/models/realm';
|
import { Realm } from '../../../src/models/realm';
|
||||||
import { WebSocketServer } from '../../../src/services/webSocketServer';
|
import { WebSocketServer } from '../../../src/services/webSocketServer';
|
||||||
import { Errors, MessageType } from '../../../src/enums';
|
import { Errors, MessageType } from '../../../src/enums';
|
||||||
@ -52,10 +53,10 @@ const checkSequence = async (c: WebSocket, msgs: { type: MessageType; error?: Er
|
|||||||
};
|
};
|
||||||
|
|
||||||
const createTestServer = ({ realm, config, url }: { realm: Realm; config: { path: string; key: string; concurrent_limit: number; }; url: string; }): Destroyable<WebSocketServer> => {
|
const createTestServer = ({ realm, config, url }: { realm: Realm; config: { path: string; key: string; concurrent_limit: number; }; url: string; }): Destroyable<WebSocketServer> => {
|
||||||
const server = new Server(url);
|
const server = new Server(url) as Server & HttpServer;
|
||||||
const webSocketServer: Destroyable<WebSocketServer> = new WebSocketServer({ server, realm, config });
|
const webSocketServer: Destroyable<WebSocketServer> = new WebSocketServer({ server, realm, config });
|
||||||
|
|
||||||
server.on('connection', (socket) => {
|
server.on('connection', (socket: WebSocket & { on?: (eventName: string, callback: () => void) => void; }) => {
|
||||||
const s = webSocketServer.socketServer;
|
const s = webSocketServer.socketServer;
|
||||||
s.emit('connection', socket, { url: socket.url });
|
s.emit('connection', socket, { url: socket.url });
|
||||||
|
|
||||||
@ -101,8 +102,8 @@ describe('WebSocketServer', () => {
|
|||||||
const realm = new Realm();
|
const realm = new Realm();
|
||||||
const config = { path: '/', key: 'testKey', concurrent_limit: 1 };
|
const config = { path: '/', key: 'testKey', concurrent_limit: 1 };
|
||||||
const config2 = { ...config, path: 'path' };
|
const config2 = { ...config, path: 'path' };
|
||||||
const server = new Server('path1');
|
const server = new Server('path1') as Server & HttpServer;
|
||||||
const server2 = new Server('path2');
|
const server2 = new Server('path2') as Server & HttpServer;
|
||||||
|
|
||||||
const webSocketServer = new WebSocketServer({ server, realm, config });
|
const webSocketServer = new WebSocketServer({ server, realm, config });
|
||||||
|
|
||||||
|
@ -14,7 +14,7 @@
|
|||||||
"noUnusedLocals": true,
|
"noUnusedLocals": true,
|
||||||
"noUnusedParameters": true,
|
"noUnusedParameters": true,
|
||||||
"resolveJsonModule": true,
|
"resolveJsonModule": true,
|
||||||
"skipLibCheck": true,
|
"skipLibCheck": false,
|
||||||
"sourceMap": true,
|
"sourceMap": true,
|
||||||
"outDir": "dist"
|
"outDir": "dist"
|
||||||
},
|
},
|
||||||
|
Loading…
x
Reference in New Issue
Block a user