caprover/src/utils/TemplateHelperVersionPrinter.ts

import request = require('request')
import fs = require('fs-extra')
import ApiStatusCodes = require('../api/ApiStatusCodes')
import Logger = require('./Logger')
import CaptainConstants = require('./CaptainConstants')
import { ITemplate } from '../models/OtherTypes'

function getTagsForImage(
    imageBaseName: string,
    url: string | undefined,
    allTags: string[] | undefined
): Promise<string[]> {
    if (!url) {
        url =
            'https://hub.docker.com/v2/repositories/' + imageBaseName + '/tags'
    }

    return new Promise<string[]>(function(resolve, reject) {
        request(
            url!,

            function(error, response, body) {
                if (error || !body) {
                    Logger.e(error)
                    reject(error)
                    return
                }

                try {
                    // Sometimes Docker server is down and it crashes Captain!
                    body = JSON.parse(body)
                } catch (e) {
                    Logger.e(e)
                }

                let results: any

                if (body) {
                    results = body.results
                }

                if (!results) {
                    Logger.e('NO RESULT')
                    reject(new Error('NO RESULT'))
                    return
                }

                if (!allTags) {
                    allTags = []
                }

                for (let idx = 0; idx < results.length; idx++) {
                    allTags.push(results[idx].name)
                }

                if (body.next) {
                    resolve(getTagsForImage(imageBaseName, body.next, allTags))
                    return
                }

                resolve(allTags)
            }
        )
    })
}

function firstEndsWithSecond(str1: string, str2: string) {
    if (!str1 || !str2) {
        throw new Error('Str1 or Str2 are null ' + !str1 + ' ' + !str2)
    }
    const idx = str1.indexOf(str2)
    return idx >= 0 && idx + str2.length === str1.length
}

function isEmpty(obj: any) {
    for (const key in obj) {
        if (obj.hasOwnProperty(key)) {
            return false
        }
    }
    return true
}

type AnyHashMap = IHashMapGeneric<any>

class TemplateHelperVersionPrinter {
    private cachedImageTags: AnyHashMap

    constructor() {
        this.cachedImageTags = {}
    }

    getDockerVersionsForTemplateName(templateObj: ITemplate) {
        const self = this

        if (isEmpty(this.cachedImageTags)) {
            throw ApiStatusCodes.createError(
                ApiStatusCodes.STATUS_ERROR_GENERIC,
                'Please wait about 30 seconds, then try again.'
            )
        }

        const tags = self.cachedImageTags[templateObj.dockerHubImageName]

        const dockerVersions = []
        for (let i = 0; i < tags.length; i++) {
            const t = tags[i]
            if (firstEndsWithSecond(t, templateObj.tagSuffix)) {
                dockerVersions.push(
                    t.substring(0, t.length - templateObj.tagSuffix.length)
                )
            }
        }

        return dockerVersions
    }

    printAvailableImageTagsForReadme(templates: ITemplate[]) {
        const self = this
        self.cachedImageTags = {}

        const tempCache: AnyHashMap = {}
        for (let i = 0; i < templates.length; i++) {
            const currentImageName = templates[i].dockerHubImageName

            getTagsForImage(currentImageName, undefined, undefined)
                .then(function(tags) {
                    tempCache[currentImageName] = tags

                    let isAllDone = true

                    for (let j = 0; j < templates.length; j++) {
                        const imageName = templates[j].dockerHubImageName
                        if (!tempCache[imageName]) {
                            isAllDone = false
                        }
                    }

                    if (isAllDone) {
                        Logger.d('Template Cache Updated!')

                        self.cachedImageTags = tempCache

                        // Used for README
                        for (
                            let tempIdx = 0;
                            tempIdx < templates.length;
                            tempIdx++
                        ) {
                            Logger.d(' ')
                            Logger.d(templates[tempIdx].templateName + '/')
                            Logger.d(
                                self
                                    .getDockerVersionsForTemplateName(
                                        templates[tempIdx]
                                    )
                                    .join(', ')
                            )
                            Logger.d(' ')
                        }
                    }
                })
                .catch(function(error) {
                    Logger.e(error)
                })
        }
    }
}

export = TemplateHelperVersionPrinter