export using cache

Author: perryr16

src/@seed/api/inventory/inventory.service.ts

@@ -352,19 +352,9 @@ export class InventoryService {
     )
   }
 
-  startInventoryExport(orgId: number): Observable<ProgressResponse> {
+  startInventoryExport(orgId: number, data: InventoryExportData): Observable<ProgressResponse> {
     const url = `/api/v3/tax_lot_properties/start_export/?organization_id=${orgId}`
-    return this._httpClient.get<ProgressResponse>(url).pipe(
-      take(1),
-      catchError((error: HttpErrorResponse) => {
-        return this._errorService.handleError(error, 'Error starting export')
-      }),
-    )
-  }
-
-  exportInventory(orgId: number, type: InventoryType, data: InventoryExportData): Observable<Blob> {
-    const url = `/api/v3/tax_lot_properties/export/?inventory_type=${type}&organization_id=${orgId}`
-    return this._httpClient.post(url, data, { responseType: 'blob' }).pipe(
+    return this._httpClient.post<ProgressResponse>(url, data).pipe(
       take(1),
       catchError((error: HttpErrorResponse) => {
         return this._errorService.handleError(error, 'Error starting export')

src/@seed/services/uploader/uploader.service.ts

@@ -7,6 +7,7 @@ import type { ProgressResponse } from '@seed/api'
 import { ErrorService } from '../error'
 import type {
   CheckProgressLoopParams,
+  ExportDataType,
   GreenButtonMeterPreview,
   MeterPreviewResponse,
   ProgressBarObj,
@@ -176,6 +177,26 @@ export class UploaderService {
     )
   }
 
+  stringToBlob(data: string, exportType: ExportDataType) {
+    const base64ToBlob = (base64: string): Blob => {
+      const binary = atob(base64)
+      const bytes = new Uint8Array(binary.length)
+      for (let i = 0; i < binary.length; i++) {
+        bytes[i] = binary.charCodeAt(i)
+      }
+
+      return new Blob([bytes], { type: 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet' })
+    }
+
+    const blobMap: Record<ExportDataType, () => Blob> = {
+      csv: () => new Blob([data], { type: 'text/csv' }),
+      xlsx: () => base64ToBlob(data),
+      geojson: () => new Blob([JSON.stringify(data, null, '\t')], { type: 'application/geo+json' }),
+    }
+
+    return blobMap[exportType]()
+  }
+
   /*
    * Updates the progress bar object with incoming progress data.
    */

src/@seed/services/uploader/uploader.types.ts

@@ -68,3 +68,5 @@ export type MeterPreviewResponse = {
   unlinkable_pm_ids: number[];
   validated_type_units: ValidatedTypeUnit[];
 }
+
+export type ExportDataType = 'csv' | 'xlsx' | 'geojson'

src/app/modules/datasets/data-mappings/data-mapping.component.ts

@@ -223,6 +223,7 @@ export class DataMappingComponent implements OnDestroy, OnInit {
             this.mappingResultsResponse = response as MappingResultsResponse
             this.nextStep(2)
           }),
+          take(1),
         )
         .subscribe()
     }

src/app/modules/inventory-detail/ubids/modal/form-modal.component.ts

@@ -63,7 +63,6 @@ export class FormModalComponent implements OnInit {
         .update(this.data.orgId, this.data.viewId, this.data.ubid.id, ubidDetails, this.data.type)
         .pipe(
           tap((response) => {
-            console.log('response', response)
             this.close(preferred)
           }),
         )
@@ -75,7 +74,6 @@ export class FormModalComponent implements OnInit {
         .create(this.data.orgId, this.data.viewId, ubidDetails, this.data.type)
         .pipe(
           tap((response) => {
-            console.log('response', response)
             this.close(preferred)
           }),
         )

src/app/modules/inventory/actions/export-modal.component.html

@@ -16,12 +16,11 @@
           <mat-button-toggle class="flex-1" value="geojson">GeoJSON</mat-button-toggle>
         </mat-button-toggle-group>
 
+        <mat-checkbox formControlName="include_notes">Include Notes</mat-checkbox>
         @if (form.value.export_type === 'geojson') {
           <mat-checkbox formControlName="include_meter_readings"
             >Include Meter Readings <span class="text-secondary"> (Only recommended for small exports)</span>
           </mat-checkbox>
-        } @else if (form.value.export_type === 'csv') {
-          <mat-checkbox formControlName="include_meter_readings"> Include Label Header</mat-checkbox>
         } @else {
           <div class="h-10"></div>
         }

src/app/modules/inventory/actions/export-modal.component.ts

@@ -3,8 +3,8 @@ import { Component, inject, ViewChild } from '@angular/core'
 import { FormControl, FormGroup, FormsModule, ReactiveFormsModule, Validators } from '@angular/forms'
 import { MAT_DIALOG_DATA, MatDialogRef } from '@angular/material/dialog'
 import type { MatStepper } from '@angular/material/stepper'
-import { catchError, combineLatest, EMPTY, finalize, Subject, switchMap, takeUntil, tap } from 'rxjs'
-import { InventoryService } from '@seed/api'
+import { catchError, EMPTY, Subject, switchMap, take, takeUntil, tap } from 'rxjs'
+import { CacheService, InventoryService, ProgressResponse } from '@seed/api'
 import { ModalHeaderComponent, ProgressBarComponent } from '@seed/components'
 import { MaterialImports } from '@seed/materials'
 import { UploaderService } from '@seed/services'
@@ -18,6 +18,7 @@ import type { InventoryExportData, InventoryType } from '../inventory.types'
 })
 export class ExportModalComponent implements OnDestroy {
   @ViewChild('stepper') stepper!: MatStepper
+  private _cacheService = inject(CacheService)
   private _dialogRef = inject(MatDialogRef<ExportModalComponent>)
   private _inventoryService = inject(InventoryService)
   private _snackBar = inject(SnackBarService)
@@ -36,41 +37,48 @@ export class ExportModalComponent implements OnDestroy {
   }
 
   form = new FormGroup({
-    name: new FormControl<string>(null, Validators.required),
-    include_notes: new FormControl(true),
     export_type: new FormControl<'csv' | 'xlsx' | 'geojson'>('csv', Validators.required),
-    include_label_header: new FormControl(false),
+    include_notes: new FormControl(false),
     include_meter_readings: new FormControl(false),
+    name: new FormControl<string>(null, Validators.required),
   })
 
   export() {
-    this._inventoryService.startInventoryExport(this.data.orgId)
+    const successFn = ({ unique_id }: ProgressResponse) => {
+      this._cacheService.getCacheEntry(this.data.orgId, unique_id).pipe(
+        tap((response: { data: string }) => {
+          const blob = this.getBlob(response.data)
+          this.downloadData(blob)
+          this.close()
+        }),
+        take(1),
+      ).subscribe()
+    }
+
+    this.initExport()
+    this._inventoryService.startInventoryExport(this.data.orgId, this.exportData)
       .pipe(
-        tap(({ progress_key }) => { this.initExport(progress_key) }),
-        switchMap(() => this.pollExport()),
-        tap((response) => { this.downloadData(response[0]) }),
+        switchMap(({ progress_key }) => this._uploaderService.checkProgressLoop({
+          progressKey: progress_key,
+          successFn,
+          failureFn: () => { this.close() },
+          progressBarObj: this.progressBarObj,
+        })),
         takeUntil(this._unsubscribeAll$),
         catchError(() => { return EMPTY }),
-        finalize(() => { this.close() }),
       )
       .subscribe()
   }
 
-  initExport(progress_key: string) {
+  initExport() {
     this.stepper.next()
     this.formatFilename()
-    this.formatExportData(this.filename, progress_key)
+    this.formatExportData(this.filename)
   }
 
-  pollExport() {
-    const { orgId, type } = this.data
-    return combineLatest([
-      this._inventoryService.exportInventory(orgId, type, this.exportData),
-      this._uploaderService.checkProgressLoop({
-        progressKey: this.exportData.progress_key,
-        progressBarObj: this.progressBarObj,
-      }),
-    ])
+  getBlob(data: string): Blob {
+    const exportType = this.form.value.export_type
+    return this._uploaderService.stringToBlob(data, exportType)
   }
 
   downloadData(data: Blob) {
@@ -91,15 +99,14 @@ export class ExportModalComponent implements OnDestroy {
     }
   }
 
-  formatExportData(filename: string, progress_key: string) {
+  formatExportData(filename: string) {
     this.exportData = {
       export_type: this.form.value.export_type,
       filename,
       ids: this.data.viewIds,
       include_meter_readings: this.form.value.include_meter_readings,
       include_notes: this.form.value.include_notes,
       profile_id: this.data.profileId,
-      progress_key,
     }
   }
 

src/app/modules/inventory/inventory.types.ts

@@ -263,5 +263,4 @@ export type InventoryExportData = {
   include_meter_readings: boolean;
   include_notes: boolean;
   profile_id: number;
-  progress_key: string;
 }