@cto.ai/ops/lib/services/Publish.js

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
const tslib_1 = require("tslib");
const sdk_1 = require("@cto.ai/sdk");
const debug_1 = tslib_1.__importDefault(require("debug"));
const json = tslib_1.__importStar(require("JSONStream"));
const through = tslib_1.__importStar(require("through2"));
const CustomErrors_1 = require("../errors/CustomErrors");
const get_docker_1 = tslib_1.__importDefault(require("../utils/get-docker"));
const debug = debug_1.default('ops:ImageService');
class Publish {
    constructor() {
        this.publishOpToAPI = async (op, version, teamID, accessToken, api, isGlueCode = false) => {
            try {
                const res = await api.create('ops', Object.assign({}, op, { version, teamID, isGlueCode, isPublic: op.isPublic }), {
                    headers: {
                        Authorization: accessToken,
                    },
                });
                return res;
            }
            catch (err) {
                throw new CustomErrors_1.CouldNotCreateOp(err.message);
            }
        };
        this.publishOpToRegistry = async (apiOp, registryAuth, teamName, accessToken, registryAuthService, version) => {
            const imageUniqueId = `${registryAuth.projectFullName}/${apiOp.id.toLowerCase()}`;
            const imageName = `${registryAuth.projectFullName}/${apiOp.name}`;
            const self = this;
            const docker = await get_docker_1.default(self, 'publish');
            try {
                if (!docker) {
                    throw new Error('Could not initialize Docker.');
                }
                // getImage always returns an image. Must listImages
                const image = docker.getImage(imageName);
                if (!image) {
                    throw new CustomErrors_1.DockerPublishNoImageFound(apiOp.name, teamName);
                }
                console.log(`🔋 Creating release ${sdk_1.ux.colors.callOutCyan(imageUniqueId)}... \n`);
                const all = [];
                let size = 0;
                const seenChunks = {};
                const parser = through.obj(function (chunk, _enc, cb) {
                    this.push(chunk.status);
                    if (chunk.aux) {
                        console.log(`\n🚀 ${sdk_1.ux.colors.white('Publishing...')}\n`);
                        console.log(`${sdk_1.ux.colors.green('>')} Tag: ${sdk_1.ux.colors.multiBlue(chunk.aux.Tag)}`);
                        console.log(`${sdk_1.ux.colors.green('>')} Size: ${sdk_1.ux.colors.multiBlue(chunk.aux.Size)}`);
                        console.log(`${sdk_1.ux.colors.green('>')} Digest: ${sdk_1.ux.colors.multiBlue(chunk.aux.Digest)}\n`);
                        size = chunk.aux.Size;
                    }
                    else if (chunk.id) {
                        const chunkString = `${chunk.status}: ${sdk_1.ux.colors.white(chunk.id)}`;
                        if (!seenChunks[chunkString]) {
                            console.log(`${chunk.status}: ${sdk_1.ux.colors.white(chunk.id)}`);
                            seenChunks[chunkString] = true;
                        }
                    }
                    cb();
                });
                const _pipe = parser.pipe;
                parser.pipe = function (dest) {
                    return _pipe(dest);
                };
                await new Promise((res, rej) => {
                    image.tag({ repo: imageUniqueId }, (err, _data) => {
                        if (err) {
                            rej(new CustomErrors_1.ImageTagError(err));
                        }
                        const image = docker.getImage(imageUniqueId);
                        image.push({
                            tag: 'latest',
                            authconfig: registryAuth.authconfig,
                        }, (err, stream) => {
                            if (err) {
                                rej(new CustomErrors_1.ImagePushError(err));
                            }
                            stream
                                .pipe(json.parse())
                                .pipe(parser)
                                .on('data', (d) => {
                                all.push(d);
                            })
                                .on('end', async function () {
                                const bar = sdk_1.ux.progress.init();
                                bar.start(100, 0);
                                for (let i = 0; i < size; i++) {
                                    bar.update(100 - size / i);
                                    await sdk_1.ux.wait(5);
                                }
                                bar.update(100);
                                bar.stop();
                                console.log(`\n🙌 ${sdk_1.ux.colors.callOutCyan(imageUniqueId)} has been published! \n`);
                                await registryAuthService.delete(accessToken, registryAuth.robotID, teamName, apiOp.name, version);
                                res();
                            });
                        });
                    });
                });
            }
            catch (err) {
                debug('%O', err);
                process.exit(1);
            }
        };
    }
}
exports.Publish = Publish;