Package: Add repositoryName

dmstern · Aug 27, 2018 · 2dd1aef · 2dd1aef
1 parent 326ef2e
commit 2dd1aef
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/src/model/Package.ts b/src/model/Package.ts
@@ -56,6 +56,7 @@ export default class Package implements PackageMetaDataDTO {
   public readonly dependenciesCount: number;
   public readonly scope: string | undefined;
   public readonly mainCode: string | undefined;
+  protected repositoryNameIsBeautified: boolean = false;
 
   constructor(packageMetaData: PackageMetaDataDTO) {
     Object.assign(this, packageMetaData);
@@ -101,4 +102,18 @@ export default class Package implements PackageMetaDataDTO {
     }
   }
 
+  public get repositoryName() {
+    if (this.repositoryUrl) {
+      if (this.repositoryUrl.includes('github')) {
+        this.repositoryNameIsBeautified = true;
+        return 'github';
+      }
+      if (this.repositoryUrl.includes('gitlab')) {
+        this.repositoryNameIsBeautified = true;
+        return 'gitlab';
+      }
+      return this.repositoryUrl.split('/')[2];
+    }
+  }
+
 }