@wocker/cron-plugin
Version:
Cron plugin for wocker
220 lines (219 loc) • 9.31 kB
JavaScript
"use strict";
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
var desc = Object.getOwnPropertyDescriptor(m, k);
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
desc = { enumerable: true, get: function() { return m[k]; } };
}
Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
o["default"] = v;
});
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 __importStar = (this && this.__importStar) || (function () {
var ownKeys = function(o) {
ownKeys = Object.getOwnPropertyNames || function (o) {
var ar = [];
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
return ar;
};
return ownKeys(o);
};
return function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
__setModuleDefault(result, mod);
return result;
};
})();
var __metadata = (this && this.__metadata) || function (k, v) {
if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
};
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.CronService = void 0;
const core_1 = require("@wocker/core");
const Path = __importStar(require("path"));
const OS = __importStar(require("os"));
const makes_1 = require("../makes");
const spawn_1 = require("../utils/spawn");
let CronService = class CronService {
constructor(appService, pluginConfigService, dockerService) {
this.appService = appService;
this.pluginConfigService = pluginConfigService;
this.dockerService = dockerService;
this.name = "cron";
this.version = "1.0.15";
this.oldContainerNames = [
"cron.ws"
];
this.oldImages = [
"wocker-cron:latest",
"wocker-cron:1.0.9",
"wocker-cron:1.0.10",
"wocker-cron:1.0.11",
"wocker-cron:1.0.12",
"wocker-cron:1.0.13"
];
}
get fs() {
return this.pluginConfigService.fs;
}
get containerName() {
return "wocker-cron";
}
get imageName() {
return `wocker-cron:${this.version}`;
}
start(restart, rebuild) {
return __awaiter(this, void 0, void 0, function* () {
for (const containerName of this.oldContainerNames) {
yield this.dockerService.removeContainer(containerName);
}
if (restart || rebuild) {
yield this.dockerService.removeContainer(this.containerName);
}
let container = yield this.dockerService.getContainer(this.containerName);
if (!container) {
yield this.build(rebuild);
const TZ = Intl.DateTimeFormat().resolvedOptions().timeZone;
container = yield this.dockerService.createContainer({
name: this.containerName,
image: this.imageName,
networkMode: "host",
restart: "always",
env: {
TZ
},
volumes: [
"/var/run/docker.sock:/var/run/docker.sock:ro",
`${this.appService.fs.path("ws.log")}:/app/ws.log`,
`${this.fs.path("crontab.json")}:/app/crontab.json`
]
});
}
const { State: { Running } } = yield container.inspect();
if (!Running) {
yield container.start();
console.info("Cron started");
}
});
}
stop() {
return __awaiter(this, void 0, void 0, function* () {
yield this.dockerService.removeContainer(this.containerName);
});
}
build(rebuild) {
return __awaiter(this, void 0, void 0, function* () {
var _a;
for (const image of this.oldImages) {
yield this.dockerService.imageRm(image);
}
const images = yield this.dockerService.imageLs({
labels: {
"wocker.name": this.name,
"wocker.type": "service"
}
});
for (const image of images) {
if (image.Labels["wocker.type"] === "service" &&
image.Labels["wocker.name"] === this.name &&
image.Labels["wocker.version"] !== this.version &&
((_a = image.RepoTags) === null || _a === void 0 ? void 0 : _a[0])) {
yield this.dockerService.imageRm(image.RepoTags[0]);
}
}
if (!this.fs.exists("crontab.json")) {
this.fs.writeJSON("crontab.json", {});
}
if (yield this.dockerService.imageExists(this.imageName)) {
if (!rebuild) {
return;
}
yield this.dockerService.imageRm(this.imageName);
}
console.info(`Building ${this.imageName}...`);
yield this.dockerService.buildImage({
tag: this.imageName,
context: Path.join(__dirname, "../../plugin"),
src: "./Dockerfile",
labels: {
"wocker.name": this.name,
"wocker.version": this.version,
"wocker.type": "service"
}
});
});
}
edit(containerName) {
return __awaiter(this, void 0, void 0, function* () {
const tmp = new core_1.FileSystem(OS.tmpdir());
const crontab = this.getCrontab(containerName);
tmp.writeFile("ws-crontab.txt", crontab);
const editor = process.env.VISUAL || process.env.EDITOR || "nano";
yield (0, spawn_1.spawn)(editor, [tmp.path("ws-crontab.txt")]);
const res = tmp.readFile("ws-crontab.txt");
if (crontab === res.toString()) {
return;
}
this.setCrontab(containerName, res.toString());
const stream = yield this.dockerService.exec(this.containerName, ["supervisorctl", "restart", "docker-gen"]);
if (!stream) {
return;
}
console.info("Restarting...");
this.dockerService.docker.modem.demuxStream(stream, process.stdout, process.stderr);
});
}
getCrontab(containerName) {
if (!this.fs.exists("crontab.json")) {
return "";
}
const { [containerName]: crontab = "" } = this.fs.readJSON("crontab.json");
return crontab;
}
setCrontab(containerName, crontab) {
const c = new makes_1.CrontabEditor(crontab);
const data = this.fs.exists("crontab.json")
? this.fs.readJSON("crontab.json")
: {};
this.fs.writeJSON("crontab.json", Object.assign(Object.assign({}, data), { [containerName]: crontab }));
}
logs() {
return __awaiter(this, void 0, void 0, function* () {
const container = yield this.dockerService.getContainer(this.containerName);
if (!container) {
return;
}
yield this.dockerService.logs(container);
});
}
};
exports.CronService = CronService;
exports.CronService = CronService = __decorate([
(0, core_1.Injectable)(),
__metadata("design:paramtypes", [core_1.AppService,
core_1.PluginConfigService,
core_1.DockerService])
], CronService);