From 06acb5978008c4d3672271b7c6faa8e0c7dbae5e Mon Sep 17 00:00:00 2001
From: chrismclarke <chris.m.clarke@live.co.uk>
Date: Thu, 15 Feb 2024 15:15:24 -0800
Subject: [PATCH] feat: forecast dashboard download and open

---
 .../pages/forecast/forecast.component.ts      | 43 ++++++++++++++++++-
 1 file changed, 41 insertions(+), 2 deletions(-)

diff --git a/apps/picsa-apps/dashboard/src/app/modules/climate/pages/forecast/forecast.component.ts b/apps/picsa-apps/dashboard/src/app/modules/climate/pages/forecast/forecast.component.ts
index 5cdb2d468..a2b005ae2 100644
--- a/apps/picsa-apps/dashboard/src/app/modules/climate/pages/forecast/forecast.component.ts
+++ b/apps/picsa-apps/dashboard/src/app/modules/climate/pages/forecast/forecast.component.ts
@@ -4,14 +4,31 @@ import { RouterModule } from '@angular/router';
 import { IDataTableOptions, PicsaDataTableComponent } from '@picsa/shared/features';
 import { SupabaseService } from '@picsa/shared/services/core/supabase';
 
+import { DashboardMaterialModule } from '../../../../material.module';
 import { ClimateService } from '../../climate.service';
 import { DashboardClimateApiStatusComponent, IApiStatusOptions } from '../../components/api-status/api-status';
 import { IForecastRow } from '../../types';
 
+const DISPLAY_COLUMNS: (keyof IForecastRow)[] = [
+  'country_code',
+  'district',
+  'type',
+  'language_code',
+  'filename',
+  'date_modified',
+  'storage_file',
+];
+
 @Component({
   selector: 'dashboard-climate-forecast',
   standalone: true,
-  imports: [CommonModule, DashboardClimateApiStatusComponent, RouterModule, PicsaDataTableComponent],
+  imports: [
+    CommonModule,
+    DashboardClimateApiStatusComponent,
+    RouterModule,
+    PicsaDataTableComponent,
+    DashboardMaterialModule,
+  ],
   templateUrl: './forecast.component.html',
   styleUrls: ['./forecast.component.scss'],
   changeDetection: ChangeDetectionStrategy.OnPush,
@@ -20,13 +37,15 @@ export class ClimateForecastPageComponent implements OnInit {
   public forecastData: IForecastRow[] = [];
 
   public tableOptions: IDataTableOptions = {
-    displayColumns: ['country', 'district', 'type', 'language', 'date_modified'],
+    displayColumns: DISPLAY_COLUMNS,
   };
   public apiStatusOptions: IApiStatusOptions = {
     events: { refresh: () => this.refreshData() },
     showStatusCode: false,
   };
 
+  public activeDownloads: Record<string, 'pending' | 'complete'> = {};
+
   constructor(private service: ClimateService, private supabase: SupabaseService, private cdr: ChangeDetectorRef) {}
 
   private get db() {
@@ -46,6 +65,26 @@ export class ClimateForecastPageComponent implements OnInit {
       await this.refreshData();
     }
   }
+
+  public async handleStorageClick(row: IForecastRow) {
+    // handle download if storage file doesn't exist or hasn't been downloaded
+    if (!row.storage_file && this.activeDownloads[row.filename] !== 'complete') {
+      await this.downloadStorageFile(row);
+    }
+    // handle open
+    const storagePath = `climate/forecasts/${row.filename}`;
+    const publicLink = this.supabase.storage.getPublicLink('mw', storagePath);
+    open(publicLink, '_blank');
+  }
+
+  private async downloadStorageFile(row: IForecastRow) {
+    this.activeDownloads[row.filename] = 'pending';
+    this.cdr.markForCheck();
+    await this.service.loadFromAPI.forecast_file(row);
+    this.activeDownloads[row.filename] = 'complete';
+    this.cdr.markForCheck();
+  }
+
   private loadForecastData(data: any[] = []) {
     this.forecastData = data;
     this.cdr.detectChanges();