@atomist/org-visualizer/lib/job/createFingerprintJob.js

Organization Visualizer using Atomist project scanning

"use strict";
/*
 * Copyright © 2019 Atomist, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
    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) : new P(function (resolve) { resolve(result.value); }).then(fulfilled, rejected); }
        step((generator = generator.apply(thisArg, _arguments || [])).next());
    });
};
Object.defineProperty(exports, "__esModule", { value: true });
const automation_client_1 = require("@atomist/automation-client");
const sdm_1 = require("@atomist/sdm");
const slack_messages_1 = require("@atomist/slack-messages");
const fingerprintTask_1 = require("./fingerprintTask");
exports.CreateFingerprintJobCommand = {
    name: "CreateFingerprintJob",
    intent: "calculate fingerprints",
    description: "Trigger a background job to calculate all fingerprints across a given org",
    listener: (ci) => __awaiter(this, void 0, void 0, function* () {
        yield fingerprintGitHubResourceProvider(ci.context, true);
    }),
};
exports.CreateFingerprintJob = {
    name: "CreateFingerprintJob",
    description: "Creates a job that calculates the fingerprints on every repo of an org",
    subscription: automation_client_1.GraphQL.subscription("OnDiscoveryJob"),
    listener: (e, ctx) => __awaiter(this, void 0, void 0, function* () {
        const job = e.data.AtmJob[0];
        if (job.name.startsWith("RepositoryDiscovery")) {
            const event = JSON.parse(job.data);
            if (!!event.data.GitHubAppInstallation) {
                yield fingerprintGitHubAppInstallation(event.data, ctx);
            }
            else {
                yield fingerprintGitHubResourceProvider(ctx);
            }
        }
        return automation_client_1.Success;
    }),
};
function fingerprintGitHubAppInstallation(event, ctx) {
    return __awaiter(this, void 0, void 0, function* () {
        const org = event.GitHubAppInstallation[0];
        const provider = event.GitHubAppInstallation[0].gitHubAppResourceProvider;
        const result = yield ctx.graphClient.query({
            name: "ReposByProvider",
            variables: {
                providerId: provider.providerId,
                org: org.owner,
            },
        });
        const repos = {
            providerId: provider.providerId,
            name: org.owner,
            tasks: result.Org[0].repos.map(repo => {
                return {
                    providerId: provider.providerId,
                    repoId: repo.id,
                    owner: repo.owner,
                    name: repo.name,
                    branch: repo.defaultBranch || "master",
                };
            }),
        };
        const prefs = automation_client_1.configurationValue("sdm.preferenceStoreFactory")(ctx);
        const analyzed = yield prefs.get(preferenceKey(org.owner), { scope: sdm_1.PreferenceScope.Sdm, defaultValue: false });
        if (!analyzed) {
            try {
                yield sdm_1.createJob({
                    command: fingerprintTask_1.calculateFingerprintTask([], []),
                    parameters: repos.tasks,
                    name: `OrganizationAnalysis/${provider.providerId}/${org.owner}`,
                    description: `Analyzing repositories in ${slack_messages_1.bold(org.owner)}`,
                }, ctx);
                yield prefs.put(preferenceKey(org.owner), true, { scope: sdm_1.PreferenceScope.Sdm });
            }
            catch (e) {
                automation_client_1.logger.warn("Failed to create job for org '%s': %s", org.owner, e.message);
            }
        }
    });
}
function fingerprintGitHubResourceProvider(ctx, rerun = false) {
    return __awaiter(this, void 0, void 0, function* () {
        const result = yield ctx.graphClient.query({
            name: "ReposByProvider",
            variables: {
                providerId: "zjlmxjzwhurspem",
            },
        });
        const orgs = result.Org.map(org => {
            const provider = org.scmProvider;
            return {
                providerId: provider.providerId,
                name: org.owner,
                tasks: org.repos.map(repo => {
                    return {
                        providerId: provider.providerId,
                        repoId: repo.id,
                        owner: repo.owner,
                        name: repo.name,
                        branch: repo.defaultBranch || "master",
                    };
                }),
            };
        });
        const prefs = automation_client_1.configurationValue("sdm.preferenceStoreFactory")(ctx);
        for (const org of orgs) {
            const analyzed = yield prefs.get(preferenceKey(org.name), { scope: sdm_1.PreferenceScope.Sdm, defaultValue: false });
            if (!analyzed || rerun) {
                try {
                    yield sdm_1.createJob({
                        command: fingerprintTask_1.calculateFingerprintTask([], []),
                        parameters: org.tasks,
                        name: `OrganizationAnalysis/${org.providerId}/${org.name}`,
                        description: `Analyzing repositories in ${slack_messages_1.bold(org.name)}`,
                    }, ctx);
                    yield prefs.put(preferenceKey(org.name), true, { scope: sdm_1.PreferenceScope.Sdm });
                }
                catch (e) {
                    automation_client_1.logger.warn("Failed to create job for org '%s': %s", org.name, e.message);
                }
            }
        }
    });
}
function preferenceKey(org) {
    return `analyzed/${org}`;
}
//# sourceMappingURL=createFingerprintJob.js.map