From 41797f5ce4203fc5afeed82b9f2ad590673c867d Mon Sep 17 00:00:00 2001
From: Kyle Spearrin <kyle.spearrin@gmail.com>
Date: Tue, 23 Jan 2018 17:29:30 -0500
Subject: [PATCH] icon component

---
 src/app/app.module.ts               |  2 +
 src/app/services/services.module.ts |  1 +
 src/app/vault/icon.component.html   |  4 ++
 src/app/vault/icon.component.ts     | 88 +++++++++++++++++++++++++++++
 src/app/vault/vault.component.html  |  1 +
 5 files changed, 96 insertions(+)
 create mode 100644 src/app/vault/icon.component.html
 create mode 100644 src/app/vault/icon.component.ts

diff --git a/src/app/app.module.ts b/src/app/app.module.ts
index e88960e0..a7c285e1 100644
--- a/src/app/app.module.ts
+++ b/src/app/app.module.ts
@@ -8,6 +8,7 @@ import { NgModule } from '@angular/core';
 import { ServicesModule } from './services/services.module';
 
 import { AppComponent } from './app.component';
+import { IconComponent } from './vault/icon.component';
 import { LoginComponent } from './accounts/login.component';
 import { VaultComponent } from './vault/vault.component';
 
@@ -20,6 +21,7 @@ import { VaultComponent } from './vault/vault.component';
     ],
     declarations: [
         AppComponent,
+        IconComponent,
         LoginComponent,
         VaultComponent,
     ],
diff --git a/src/app/services/services.module.ts b/src/app/services/services.module.ts
index f5f78a68..49d473b2 100644
--- a/src/app/services/services.module.ts
+++ b/src/app/services/services.module.ts
@@ -97,6 +97,7 @@ environmentService.setUrlsFromStorage().then(() => {
         { provide: CipherServiceAbstraction, useValue: cipherService },
         { provide: FolderServiceAbstraction, useValue: folderService },
         { provide: CollectionServiceAbstraction, useValue: collectionService },
+        { provide: EnvironmentServiceAbstraction, useValue: environmentService },
     ],
 })
 export class ServicesModule {
diff --git a/src/app/vault/icon.component.html b/src/app/vault/icon.component.html
new file mode 100644
index 00000000..6a18b042
--- /dev/null
+++ b/src/app/vault/icon.component.html
@@ -0,0 +1,4 @@
+<div class="icon">
+    <img [src]="image" *ngIf="imageEnabled && image" alt="" />
+    <i class="fa fa-fw fa-lg {{icon}}" *ngIf="!imageEnabled || !image"></i>
+</div>
diff --git a/src/app/vault/icon.component.ts b/src/app/vault/icon.component.ts
new file mode 100644
index 00000000..1a57b1f6
--- /dev/null
+++ b/src/app/vault/icon.component.ts
@@ -0,0 +1,88 @@
+import * as template from './icon.component.html';
+
+import {
+    Component,
+    Input,
+    OnChanges,
+} from '@angular/core';
+
+import { CipherType } from 'jslib/enums/cipherType';
+
+import { EnvironmentService } from 'jslib/abstractions/environment.service';
+
+@Component({
+    selector: 'app-vault-icon',
+    template: template,
+})
+export class IconComponent implements OnChanges {
+    @Input() cipher: any;
+    icon: string;
+    image: string;
+    fallbackImage: string;
+    imageEnabled: boolean;
+
+    private iconsUrl: string;
+
+    constructor(private environmentService: EnvironmentService) {
+        this.imageEnabled = true; // TODO
+
+        this.iconsUrl = environmentService.iconsUrl;
+        if (!this.iconsUrl) {
+            if (environmentService.baseUrl) {
+                this.iconsUrl = environmentService.baseUrl + '/icons';
+            } else {
+                this.iconsUrl = 'https://icons.bitwarden.com';
+            }
+        }
+    }
+
+    ngOnChanges() {
+        switch (this.cipher.type) {
+            case CipherType.Login:
+                this.icon = 'fa-globe';
+                this.setLoginIcon();
+                break;
+            case CipherType.SecureNote:
+                this.icon = 'fa-sticky-note-o';
+                break;
+            case CipherType.Card:
+                this.icon = 'fa-credit-card';
+                break;
+            case CipherType.Identity:
+                this.icon = 'fa-id-card-o';
+                break;
+            default:
+                break;
+        }
+    }
+
+    private setLoginIcon() {
+        if (this.cipher.login.uri) {
+            let hostnameUri = this.cipher.login.uri;
+            let isWebsite = false;
+
+            if (hostnameUri.indexOf('androidapp://') === 0) {
+                this.icon = 'fa-android';
+                this.image = null;
+            } else if (hostnameUri.indexOf('iosapp://') === 0) {
+                this.icon = 'fa-apple';
+                this.image = null;
+            } else if (this.imageEnabled && hostnameUri.indexOf('://') === -1 && hostnameUri.indexOf('.') > -1) {
+                hostnameUri = 'http://' + hostnameUri;
+                isWebsite = true;
+            } else if (this.imageEnabled) {
+                isWebsite = hostnameUri.indexOf('http') === 0 && hostnameUri.indexOf('.') > -1;
+            }
+
+            if (this.imageEnabled && isWebsite) {
+                try {
+                    const url = new URL(hostnameUri);
+                    this.image = this.iconsUrl + '/' + url.hostname + '/icon.png';
+                    this.fallbackImage = '../images/fa-globe.png'; // TODO?
+                } catch (e) { }
+            }
+        } else {
+            this.image = null;
+        }
+    }
+}
diff --git a/src/app/vault/vault.component.html b/src/app/vault/vault.component.html
index 116cfe9e..ee646e6a 100644
--- a/src/app/vault/vault.component.html
+++ b/src/app/vault/vault.component.html
@@ -53,6 +53,7 @@
                 <div class="list-section" style="padding-top: 0; padding-bottom: 0;">
                     <a *ngFor="let cipher of vaultCiphers"
                        href="#" class="list-section-item condensed" title="View Item">
+                        <app-vault-icon [cipher]="cipher"></app-vault-icon>
                         <span class="text">
                             {{cipher.name}}
                             <i class="fa fa-share-alt text-muted" *ngIf="cipher.organizationId"></i>