Implemented rules

Author: firecow

src/default_cmd.ts

@@ -22,20 +22,27 @@ exports.handler = async(argv: any) => {
     if (argv.list !== undefined) {
         const pipelineIid = predefinedVariables.getPipelineIid(cwd);
         const parser = new Parser(cwd, pipelineIid);
+        await parser.initJobs();
+        parser.validateNeedsTags();
+
         await Commander.runList(parser);
         return;
     }
 
     if (argv.job) {
         const pipelineIid = predefinedVariables.getPipelineIid(cwd);
         const parser = new Parser(cwd, pipelineIid);
+        await parser.initJobs();
         parser.validateNeedsTags();
+
         await Commander.runSingleJob(parser, argv.job as string, argv.needs as boolean);
     } else {
         predefinedVariables.incrementPipelineIid(cwd);
         const pipelineIid = predefinedVariables.getPipelineIid(cwd);
         const parser = new Parser(cwd, pipelineIid);
+        await parser.initJobs();
         parser.validateNeedsTags();
+
         await Commander.runPipeline(parser, argv.manual as string[] || []);
     }
 };

src/index.ts

@@ -44,6 +44,7 @@ const argv = yargs
         const cwd = a.cwd as string || process.cwd();
         const pipelineIid = predefinedVariables.getPipelineIid(cwd);
         const parser = new Parser(cwd, pipelineIid, true);
+        await parser.initJobs();
         return parser.getJobNames();
     })
     .argv;

src/job.ts

@@ -1,5 +1,5 @@
 import * as c from "ansi-colors";
-import {spawn} from "child_process";
+import {spawn, execSync} from "child_process";
 import * as deepExtend from "deep-extend";
 import * as fs from "fs-extra";
 import * as glob from "glob";
@@ -19,8 +19,6 @@ export class Job {
     public readonly name: string;
     public readonly needs: string[] | null;
     public readonly stage: string;
-    public readonly allowFailure: boolean;
-    public readonly when: string;
     public readonly maxJobNameLength: number;
     public readonly stageIndex: number;
 
@@ -32,6 +30,10 @@ export class Job {
     private readonly scripts: string[] = [];
     private readonly variables: { [key: string]: string };
     private readonly predefinedVariables: { [key: string]: string };
+    private readonly rules: any;
+
+    public allowFailure: boolean;
+    public when: string;
 
     private prescriptsExitCode = 0;
     private afterScriptsExitCode = 0;
@@ -86,6 +88,7 @@ export class Job {
         this.allowFailure = jobData.allow_failure || false;
         this.variables = jobData.variables || {};
         this.needs = jobData.needs || null;
+        this.rules = jobData.rules || null;
 
         this.predefinedVariables = {
             CI_COMMIT_SHORT_SHA: "a33bd89c", // Changes
@@ -119,6 +122,27 @@ export class Job {
         };
     }
 
+    public async initRules() {
+        if (!this.rules) {
+            return
+        }
+        for (const rule of this.rules) {
+            try {
+                if (rule['if']) {
+                    const output = execSync(`[ ${rule['if']} ] && exit 0 || exit 1`, {cwd: this.cwd, env: this.getEnvs(), shell: 'bash'});
+                    if (output.length > 0) {
+                        process.stderr.write(`Rule output ${output}`);
+                    }
+                }
+                this.when = rule['when'] ? rule['when'] : this.when;
+                this.allowFailure = rule['allowFailure'] ? rule['allowFailure'] : this.allowFailure;
+                break;
+            } catch (e) {
+                // By pass rule on exit 1
+            }
+        }
+    }
+
     public getPrescriptsExitCode() {
         return this.prescriptsExitCode;
     }

src/parser.ts

@@ -19,13 +19,16 @@ export class Parser {
     private readonly stages: Map<string, Stage> = new Map();
     private readonly cwd: string;
     private readonly bashCompletionPhase: boolean;
+    private readonly pipelineIid: number;
+    private readonly gitlabData: any;
 
     public constructor(cwd: any, pipelineIid: number, bashCompletionPhase: boolean = false) {
         let path = '';
         let yamlDataList: any[] = [];
 
         this.bashCompletionPhase = bashCompletionPhase;
         this.cwd = cwd;
+        this.pipelineIid = pipelineIid;
 
         path = `${cwd}/.gitlab-ci.yml`;
         const gitlabCiData = Parser.loadYaml(path);
@@ -65,6 +68,14 @@ export class Parser {
             this.maxJobNameLength = Math.max(this.maxJobNameLength, key.length);
         }
 
+        this.gitlabData = gitlabData;
+    }
+
+    public async initJobs() {
+        const pipelineIid = this.pipelineIid;
+        const cwd = this.cwd;
+        const gitlabData = this.gitlabData;
+        const promises = [];
         // Generate jobs and put them into stages
         for (const [key, value] of Object.entries(gitlabData)) {
             if (this.illigalJobNames.includes(key) || key[0] === ".") {
@@ -73,6 +84,7 @@ export class Parser {
 
             const jobId = predefinedVariables.getJobId(cwd);
             const job = new Job(value, key, gitlabData.stages, cwd, gitlabData, pipelineIid, jobId, this.maxJobNameLength);
+            promises.push(job.initRules());
             const stage = this.stages.get(job.stage);
             if (stage) {
                 stage.addJob(job);
@@ -85,6 +97,8 @@ export class Parser {
 
             this.jobs.set(key, job);
         }
+
+        await Promise.all(promises);
     }
 
     public static loadYaml(filePath: string): any {