Refactor upload processing

1 files changed, 10 insertions, 0 deletions

diff --git a/src/metadata.ts b/src/metadata.ts
index 0877031..b7dbf19 100644
--- a/src/metadata.ts
+++ b/src/metadata.ts
@@ -3,6 +3,7 @@ import winston from "winston";
 
 import loggerConfig from "./util/logger";
 import { BranchNotFoundError } from "./errors";
+import { GradientStyle } from "./formats";
 
 interface Branch {
   head: string;
@@ -32,6 +33,8 @@ export interface EnvConfig {
   uploadLimit: number;
   hostDir: string;
   publicDir: string;
+  stage1: number;
+  stage2: number;
 }
 
 class Metadata {
@@ -59,6 +62,13 @@ class Metadata {
     return this.config.publicDir;
   }
 
+  getGradientStyle(): GradientStyle {
+    return {
+      stage1: this.config.stage1,
+      stage2: this.config.stage2
+    };
+  }
+
   async getHeadCommit(
     organization: string,
     repository: string,