104 lines
3.0 KiB
TypeScript
104 lines
3.0 KiB
TypeScript
/*
|
|
* Copyright (C) 2026 Fluxer Contributors
|
|
*
|
|
* This file is part of Fluxer.
|
|
*
|
|
* Fluxer is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* Fluxer is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU Affero General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with Fluxer. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
import {randomUUID} from 'node:crypto';
|
|
import type {IRelayRepository, RelayInfo} from '@app/repositories/RelayRepository';
|
|
import type {GeoLocation, IGeoSelectionService, RelayWithDistance} from '@app/services/GeoSelectionService';
|
|
|
|
export interface RegisterRelayRequest {
|
|
name: string;
|
|
url: string;
|
|
latitude: number;
|
|
longitude: number;
|
|
region: string;
|
|
capacity: number;
|
|
public_key: string;
|
|
}
|
|
|
|
export interface IRelayRegistryService {
|
|
registerRelay(request: RegisterRelayRequest): RelayInfo;
|
|
getRelay(id: string): RelayInfo | null;
|
|
getRelayStatus(id: string): RelayInfo | null;
|
|
listRelays(clientLocation?: GeoLocation, limit?: number): Array<RelayInfo | RelayWithDistance>;
|
|
updateRelayHeartbeat(id: string): void;
|
|
removeRelay(id: string): void;
|
|
}
|
|
|
|
export class RelayRegistryService implements IRelayRegistryService {
|
|
private readonly repository: IRelayRepository;
|
|
private readonly geoService: IGeoSelectionService;
|
|
|
|
constructor(repository: IRelayRepository, geoService: IGeoSelectionService) {
|
|
this.repository = repository;
|
|
this.geoService = geoService;
|
|
}
|
|
|
|
registerRelay(request: RegisterRelayRequest): RelayInfo {
|
|
const now = new Date().toISOString();
|
|
const relay: RelayInfo = {
|
|
id: randomUUID(),
|
|
name: request.name,
|
|
url: request.url,
|
|
latitude: request.latitude,
|
|
longitude: request.longitude,
|
|
region: request.region,
|
|
capacity: request.capacity,
|
|
current_connections: 0,
|
|
public_key: request.public_key,
|
|
registered_at: now,
|
|
last_seen_at: now,
|
|
healthy: true,
|
|
failed_checks: 0,
|
|
};
|
|
|
|
this.repository.saveRelay(relay);
|
|
return relay;
|
|
}
|
|
|
|
getRelay(id: string): RelayInfo | null {
|
|
return this.repository.getRelay(id);
|
|
}
|
|
|
|
getRelayStatus(id: string): RelayInfo | null {
|
|
return this.repository.getRelay(id);
|
|
}
|
|
|
|
listRelays(clientLocation?: GeoLocation, limit?: number): Array<RelayInfo | RelayWithDistance> {
|
|
const healthyRelays = this.repository.getHealthyRelays();
|
|
|
|
if (!clientLocation) {
|
|
return limit ? healthyRelays.slice(0, limit) : healthyRelays;
|
|
}
|
|
|
|
if (limit) {
|
|
return this.geoService.selectNearestRelays(healthyRelays, clientLocation, limit);
|
|
}
|
|
|
|
return this.geoService.sortByProximity(healthyRelays, clientLocation);
|
|
}
|
|
|
|
updateRelayHeartbeat(id: string): void {
|
|
this.repository.updateRelayLastSeen(id);
|
|
}
|
|
|
|
removeRelay(id: string): void {
|
|
this.repository.removeRelay(id);
|
|
}
|
|
}
|