@docker-digital/dockernet-agent 0.0.1

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 DockerNet
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,120 @@
+# DockerNet Agent
+
+NestJS-based agent application for DockerNet system.
+
+## Installation
+
+### Global Installation (Recommended)
+
+```bash
+npm install -g @dockernet/agent
+```
+
+Sau khi cài đặt, bạn có thể chạy `agent` hoặc `dockernet-agent` từ bất kỳ đâu:
+
+```bash
+# Set API key
+agent --apiKey YOUR_API_KEY
+
+# Show config
+agent --config
+
+# List profiles
+agent --profiles
+
+# Run as daemon
+agent --run
+
+# Stop daemon
+agent --stop
+```
+
+### Local Installation
+
+```bash
+npm install
+```
+
+## Usage
+
+### CLI Commands (After Global Installation)
+
+```bash
+# Set API key
+agent --apiKey YOUR_API_KEY
+
+# Show config
+agent --config
+
+# List profiles
+agent --profiles
+
+# Show version
+agent --version
+
+# Run as daemon
+agent --run
+
+# Stop daemon
+agent --stop
+
+# Restart daemon
+agent --restart
+```
+
+### Local Development Commands
+
+```bash
+# Set API key
+npm run agent -- --apiKey YOUR_API_KEY
+
+# Show config
+npm run agent -- --config
+
+# List profiles
+npm run agent -- --profiles
+
+# Run as daemon
+npm run agent -- --run
+```
+
+### Development
+
+```bash
+# Run in development mode
+npm run dev
+
+# Build
+npm run build
+
+# Run production build
+npm run start:prod
+```
+
+## Architecture
+
+The agent is structured following NestJS best practices:
+
+- **Services**: Core business logic
+  - `ApiService`: HTTP client for backend API
+  - `ConfigService`: Configuration management
+  - `DockerService`: Docker operations
+  - `EventService`: Event polling and processing
+  - `LogService`: Container log streaming
+  - `MachineService`: Machine registration and status
+  - `ProfileService`: Profile management (start/stop)
+  - `VolumeSyncService`: Volume synchronization
+
+- **Commands**: CLI interface
+  - `AgentCommand`: Main command handler
+
+## Features
+
+- Machine registration with server
+- Event polling and processing
+- Profile start/stop operations
+- Container log streaming
+- Service status updates
+- Volume synchronization
+- Private registry authentication
+

package/dist/agent/agent.module.d.ts

@@ -0,0 +1,2 @@
+export declare class AgentModule {
+}

package/dist/agent/agent.module.js

@@ -0,0 +1,47 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AgentModule = void 0;
+const common_1 = require("@nestjs/common");
+const config_1 = require("@nestjs/config");
+const agent_command_1 = require("./commands/agent.command");
+const api_service_1 = require("./services/api.service");
+const config_service_1 = require("./services/config.service");
+const docker_service_1 = require("./services/docker.service");
+const docker_event_service_1 = require("./services/docker-event.service");
+const event_service_1 = require("./services/event.service");
+const log_service_1 = require("./services/log.service");
+const machine_service_1 = require("./services/machine.service");
+const profile_service_1 = require("./services/profile.service");
+const volume_sync_service_1 = require("./services/volume-sync.service");
+const websocket_client_service_1 = require("./services/websocket-client.service");
+const event_websocket_client_service_1 = require("./services/event-websocket-client.service");
+let AgentModule = class AgentModule {
+};
+exports.AgentModule = AgentModule;
+exports.AgentModule = AgentModule = __decorate([
+    (0, common_1.Module)({
+        imports: [config_1.ConfigModule],
+        providers: [
+            agent_command_1.AgentCommand,
+            api_service_1.ApiService,
+            config_service_1.ConfigService,
+            docker_service_1.DockerService,
+            docker_event_service_1.DockerEventService,
+            event_service_1.EventService,
+            log_service_1.LogService,
+            machine_service_1.MachineService,
+            profile_service_1.ProfileService,
+            volume_sync_service_1.VolumeSyncService,
+            websocket_client_service_1.WebSocketClientService,
+            event_websocket_client_service_1.EventWebSocketClientService,
+        ],
+        exports: [agent_command_1.AgentCommand],
+    })
+], AgentModule);
+//# sourceMappingURL=agent.module.js.map

package/dist/agent/agent.module.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.module.js","sourceRoot":"","sources":["../../src/agent/agent.module.ts"],"names":[],"mappings":";;;;;;;;;AAAA,2CAAwC;AACxC,2CAA8C;AAC9C,4DAAwD;AACxD,wDAAoD;AACpD,8DAA0D;AAC1D,8DAA0D;AAC1D,0EAAqE;AACrE,4DAAwD;AACxD,wDAAoD;AACpD,gEAA4D;AAC5D,gEAA4D;AAC5D,wEAAmE;AACnE,kFAA6E;AAC7E,8FAAwF;AAoBjF,IAAM,WAAW,GAAjB,MAAM,WAAW;CAAG,CAAA;AAAd,kCAAW;sBAAX,WAAW;IAlBvB,IAAA,eAAM,EAAC;QACN,OAAO,EAAE,CAAC,qBAAY,CAAC;QACvB,SAAS,EAAE;YACT,4BAAY;YACZ,wBAAU;YACV,8BAAa;YACb,8BAAa;YACb,yCAAkB;YAClB,4BAAY;YACZ,wBAAU;YACV,gCAAc;YACd,gCAAc;YACd,uCAAiB;YACjB,iDAAsB;YACtB,4DAA2B;SAC5B;QACD,OAAO,EAAE,CAAC,4BAAY,CAAC;KACxB,CAAC;GACW,WAAW,CAAG"}

package/dist/agent/commands/agent.command.d.ts

@@ -0,0 +1,58 @@
+import { OnModuleInit } from '@nestjs/common';
+import { CommandRunner } from 'nest-commander';
+import { ApiService } from '../services/api.service';
+import { ConfigService } from '../services/config.service';
+import { DockerService } from '../services/docker.service';
+import { EventService } from '../services/event.service';
+import { LogService } from '../services/log.service';
+import { MachineService } from '../services/machine.service';
+import { ProfileService } from '../services/profile.service';
+import { VolumeSyncService } from '../services/volume-sync.service';
+import { WebSocketClientService } from '../services/websocket-client.service';
+interface AgentCommandOptions {
+    apiKey?: string;
+    config?: boolean;
+    profiles?: boolean;
+    upgrade?: boolean;
+    log?: boolean;
+    version?: boolean;
+    run?: boolean;
+    stop?: boolean;
+    restart?: boolean;
+}
+export declare class AgentCommand extends CommandRunner implements OnModuleInit {
+    private apiService;
+    private configService;
+    private dockerService;
+    private eventService;
+    private logService;
+    private machineService;
+    private profileService;
+    private volumeSyncService;
+    private websocketClientService;
+    private readonly logger;
+    private isRunning;
+    constructor(apiService: ApiService, configService: ConfigService, dockerService: DockerService, eventService: EventService, logService: LogService, machineService: MachineService, profileService: ProfileService, volumeSyncService: VolumeSyncService, websocketClientService: WebSocketClientService);
+    onModuleInit(): Promise<void>;
+    run(passedParams: string[], options?: AgentCommandOptions): Promise<void>;
+    parseApiKey(val: string): string;
+    parseConfig(): boolean;
+    parseProfiles(): boolean;
+    parseUpgrade(): boolean;
+    parseLog(): boolean;
+    parseVersion(): boolean;
+    parseRun(): boolean;
+    parseStop(): boolean;
+    parseRestart(): boolean;
+    private handleApiKey;
+    private handleConfig;
+    private handleProfiles;
+    private handleUpgrade;
+    private handleLog;
+    private handleVersion;
+    private handleStop;
+    private handleRestart;
+    private printUsage;
+    runDaemon(): Promise<void>;
+}
+export {};

package/dist/agent/commands/agent.command.js

@@ -0,0 +1,417 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+var AgentCommand_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AgentCommand = void 0;
+const common_1 = require("@nestjs/common");
+const nest_commander_1 = require("nest-commander");
+const fs = require("fs-extra");
+const path = require("path");
+const api_service_1 = require("../services/api.service");
+const config_service_1 = require("../services/config.service");
+const docker_service_1 = require("../services/docker.service");
+const event_service_1 = require("../services/event.service");
+const log_service_1 = require("../services/log.service");
+const machine_service_1 = require("../services/machine.service");
+const profile_service_1 = require("../services/profile.service");
+const volume_sync_service_1 = require("../services/volume-sync.service");
+const websocket_client_service_1 = require("../services/websocket-client.service");
+let AgentCommand = AgentCommand_1 = class AgentCommand extends nest_commander_1.CommandRunner {
+    constructor(apiService, configService, dockerService, eventService, logService, machineService, profileService, volumeSyncService, websocketClientService) {
+        super();
+        this.apiService = apiService;
+        this.configService = configService;
+        this.dockerService = dockerService;
+        this.eventService = eventService;
+        this.logService = logService;
+        this.machineService = machineService;
+        this.profileService = profileService;
+        this.volumeSyncService = volumeSyncService;
+        this.websocketClientService = websocketClientService;
+        this.logger = new common_1.Logger(AgentCommand_1.name);
+        this.isRunning = false;
+    }
+    async onModuleInit() {
+        await this.apiService.initialize();
+    }
+    async run(passedParams, options) {
+        if (options?.apiKey) {
+            await this.handleApiKey(options.apiKey);
+            return;
+        }
+        if (options?.config) {
+            await this.handleConfig();
+            return;
+        }
+        if (options?.profiles) {
+            await this.handleProfiles();
+            return;
+        }
+        if (options?.upgrade) {
+            await this.handleUpgrade();
+            return;
+        }
+        if (options?.log) {
+            await this.handleLog();
+            return;
+        }
+        if (options?.version) {
+            this.handleVersion();
+            return;
+        }
+        if (options?.stop) {
+            await this.handleStop();
+            return;
+        }
+        if (options?.restart) {
+            await this.handleRestart();
+            return;
+        }
+        if (!options || Object.keys(options).length === 0) {
+            if (passedParams.length === 0) {
+                this.handleVersion();
+                return;
+            }
+        }
+        if (options?.run) {
+            await this.runDaemon();
+            return;
+        }
+        this.printUsage();
+    }
+    parseApiKey(val) {
+        return val;
+    }
+    parseConfig() {
+        return true;
+    }
+    parseProfiles() {
+        return true;
+    }
+    parseUpgrade() {
+        return true;
+    }
+    parseLog() {
+        return true;
+    }
+    parseVersion() {
+        return true;
+    }
+    parseRun() {
+        return true;
+    }
+    parseStop() {
+        return true;
+    }
+    parseRestart() {
+        return true;
+    }
+    async handleApiKey(apiKey) {
+        if (!apiKey) {
+            console.error('API key is required');
+            console.error('Example: agent --apiKey YOUR_API_KEY');
+            process.exit(1);
+        }
+        const existing = await this.configService.loadConfig();
+        const machineKey = existing.machineKey || this.machineService.generateMachineKey();
+        if (!existing.machineKey) {
+            console.log(`Generated new MACHINE_KEY: ${machineKey}`);
+        }
+        await this.configService.saveConfig({
+            apiKey,
+            machineKey,
+            serverUrl: existing.serverUrl,
+        });
+        await this.apiService.setApiKey(apiKey);
+        console.log(`API key saved to ${this.configService.configFilePath}`);
+        process.exit(0);
+    }
+    async handleConfig() {
+        const config = await this.configService.loadConfig();
+        console.log(`Config file  : ${this.configService.configFilePath}`);
+        console.log(`SERVER_URL   : ${this.configService.serverUrl}`);
+        console.log(`MACHINE_KEY  : ${config.machineKey || '<not set>'}`);
+        if (config.apiKey) {
+            console.log(`API_KEY      : (set, value hidden)`);
+        }
+        else {
+            console.log(`API_KEY      : <not set>`);
+        }
+        process.exit(0);
+    }
+    async handleProfiles() {
+        const workDir = this.configService.workDirPath;
+        const profiles = await fs.readdir(workDir);
+        console.log(`Profiles in ${workDir}:`);
+        console.log(`${'PROFILE_ID'.padEnd(30)} | ${'WORKSPACE_ID'.padEnd(36)} | ${'RUNNING'.padEnd(10)}`);
+        console.log('-'.repeat(80));
+        let found = 0;
+        for (const profileId of profiles) {
+            const profileDir = path.join(workDir, profileId);
+            if (!(await fs.stat(profileDir)).isDirectory()) {
+                continue;
+            }
+            found++;
+            const metadataFile = path.join(profileDir, '.profile_metadata');
+            let workspaceId = '<unknown>';
+            let running = 'no';
+            if (await fs.pathExists(metadataFile)) {
+                const metadata = await fs.readFile(metadataFile, 'utf-8');
+                const match = metadata.match(/^workspace_id=(.+)$/m);
+                if (match) {
+                    workspaceId = match[1].trim();
+                }
+            }
+            const composeFile = path.join(profileDir, `docker-compose_${profileId}.yaml`);
+            if (await fs.pathExists(composeFile)) {
+                running = 'yes';
+            }
+            console.log(`${profileId.padEnd(30)} | ${workspaceId.padEnd(36)} | ${running.padEnd(10)}`);
+        }
+        if (found === 0) {
+            console.log(`(no profiles found in ${workDir})`);
+        }
+        process.exit(0);
+    }
+    async handleUpgrade() {
+        const scriptUrl = `${this.configService.apiBaseUrl}/script`;
+        console.log(`Downloading latest agent from: ${scriptUrl}`);
+        console.log('Upgrade functionality to be implemented');
+        process.exit(0);
+    }
+    async handleLog() {
+        const logFile = this.configService.logFilePath;
+        if (await fs.pathExists(logFile)) {
+            console.log(`Following log from ${logFile}:`);
+            console.log('Press Ctrl+C to stop.');
+            console.log('Log viewing functionality to be implemented');
+        }
+        else {
+            console.log(`No log file found at ${logFile}`);
+            console.log('Suggestion: run agent as daemon with: agent --run');
+        }
+        process.exit(0);
+    }
+    handleVersion() {
+        console.log('DockerNet Agent version 0.1.2');
+        process.exit(0);
+    }
+    async handleStop() {
+        const pidFile = this.configService.pidFilePath;
+        if (await fs.pathExists(pidFile)) {
+            const pid = parseInt(await fs.readFile(pidFile, 'utf-8'), 10);
+            if (pid && !isNaN(pid)) {
+                try {
+                    process.kill(pid, 'SIGTERM');
+                    console.log(`Stopped DockerNet Agent (PID ${pid})`);
+                }
+                catch (error) {
+                    console.log(`PID ${pid} is not valid or process is not running`);
+                }
+            }
+            await fs.remove(pidFile);
+        }
+        else {
+            console.log(`No PID file found (${pidFile}). Agent may not be running.`);
+        }
+        process.exit(0);
+    }
+    async handleRestart() {
+        await this.handleStop();
+        await new Promise((resolve) => setTimeout(resolve, 2000));
+        await this.runDaemon();
+        process.exit(0);
+    }
+    printUsage() {
+        console.log(`
+DockerNet Agent CLI
+
+Usage:
+  agent --apiKey <API_KEY>   : Save/update API key in config
+  agent --config             : Show current config (mask API key)
+  agent --profiles           : List saved profiles on this machine
+  agent --upgrade            : Download latest agent script
+  agent --log                : Show log (if running as daemon)
+  agent --version            : Show agent version
+  agent --run                : Run agent as daemon
+  agent --stop               : Stop running daemon
+  agent --restart            : Restart agent daemon
+
+Run without arguments to see version.
+    `);
+        process.exit(0);
+    }
+    async runDaemon() {
+        if (this.isRunning) {
+            this.logger.warn('Agent is already running');
+            return;
+        }
+        this.isRunning = true;
+        this.logger.log('Starting DockerNet Agent daemon...');
+        try {
+            await this.dockerService.checkDependencies();
+        }
+        catch (error) {
+            this.logger.error(`Dependency check failed: ${error.message}`);
+            process.exit(1);
+        }
+        const config = await this.configService.loadConfig();
+        if (!config.apiKey) {
+            this.logger.error('API key not found. Run: agent --apiKey YOUR_API_KEY');
+            process.exit(1);
+        }
+        await this.apiService.setApiKey(config.apiKey);
+        if (!config.machineKey) {
+            const machineKey = this.machineService.generateMachineKey();
+            await this.configService.saveConfig({
+                apiKey: config.apiKey,
+                machineKey,
+                serverUrl: config.serverUrl,
+            });
+            this.logger.log(`Generated new machine key: ${machineKey}`);
+        }
+        await this.machineService.registerMachine();
+        this.logger.log('Checking and updating saved profiles status...');
+        try {
+            await this.profileService.updateAllProfilesStatus();
+            this.logger.log('Initial profile status check completed');
+        }
+        catch (error) {
+            this.logger.error(`Error during initial profile status check: ${error.message}`);
+        }
+        await this.logService.autoStreamLogsForAllContainers();
+        await this.eventService.startPolling();
+        const statusInterval = setInterval(async () => {
+            await this.machineService.updateMachineStatus('live');
+        }, 5000);
+        const logInterval = setInterval(async () => {
+            await this.logService.autoStreamLogsForAllContainers();
+        }, 10000);
+        process.on('SIGTERM', async () => {
+            this.logger.log('Received SIGTERM, shutting down...');
+            clearInterval(statusInterval);
+            clearInterval(logInterval);
+            this.eventService.stopPolling();
+            this.websocketClientService.disconnect();
+            await this.machineService.updateMachineStatus('exit');
+            process.exit(0);
+        });
+        process.on('SIGINT', async () => {
+            this.logger.log('Received SIGINT, shutting down...');
+            clearInterval(statusInterval);
+            clearInterval(logInterval);
+            this.eventService.stopPolling();
+            this.websocketClientService.disconnect();
+            await this.machineService.updateMachineStatus('exit');
+            process.exit(0);
+        });
+        this.logger.log('DockerNet Agent daemon is running...');
+    }
+};
+exports.AgentCommand = AgentCommand;
+__decorate([
+    (0, nest_commander_1.Option)({
+        flags: '--apiKey <apiKey>',
+        description: 'Save/update API key in config',
+    }),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [String]),
+    __metadata("design:returntype", String)
+], AgentCommand.prototype, "parseApiKey", null);
+__decorate([
+    (0, nest_commander_1.Option)({
+        flags: '--config',
+        description: 'Show current config (mask API key)',
+    }),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Boolean)
+], AgentCommand.prototype, "parseConfig", null);
+__decorate([
+    (0, nest_commander_1.Option)({
+        flags: '--profiles',
+        description: 'List saved profiles on this machine',
+    }),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Boolean)
+], AgentCommand.prototype, "parseProfiles", null);
+__decorate([
+    (0, nest_commander_1.Option)({
+        flags: '--upgrade',
+        description: 'Download latest agent script',
+    }),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Boolean)
+], AgentCommand.prototype, "parseUpgrade", null);
+__decorate([
+    (0, nest_commander_1.Option)({
+        flags: '--log',
+        description: 'Show log (if running as daemon)',
+    }),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Boolean)
+], AgentCommand.prototype, "parseLog", null);
+__decorate([
+    (0, nest_commander_1.Option)({
+        flags: '--version',
+        description: 'Show agent version',
+    }),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Boolean)
+], AgentCommand.prototype, "parseVersion", null);
+__decorate([
+    (0, nest_commander_1.Option)({
+        flags: '--run',
+        description: 'Run agent as daemon',
+    }),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Boolean)
+], AgentCommand.prototype, "parseRun", null);
+__decorate([
+    (0, nest_commander_1.Option)({
+        flags: '--stop',
+        description: 'Stop running daemon',
+    }),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Boolean)
+], AgentCommand.prototype, "parseStop", null);
+__decorate([
+    (0, nest_commander_1.Option)({
+        flags: '--restart',
+        description: 'Restart agent daemon',
+    }),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Boolean)
+], AgentCommand.prototype, "parseRestart", null);
+exports.AgentCommand = AgentCommand = AgentCommand_1 = __decorate([
+    (0, common_1.Injectable)(),
+    (0, nest_commander_1.Command)({
+        name: 'agent',
+        description: 'DockerNet Agent CLI',
+    }),
+    __metadata("design:paramtypes", [api_service_1.ApiService,
+        config_service_1.ConfigService,
+        docker_service_1.DockerService,
+        event_service_1.EventService,
+        log_service_1.LogService,
+        machine_service_1.MachineService,
+        profile_service_1.ProfileService,
+        volume_sync_service_1.VolumeSyncService,
+        websocket_client_service_1.WebSocketClientService])
+], AgentCommand);
+//# sourceMappingURL=agent.command.js.map