182 files

custom_components/maintenance_supporter/frontend-src/components/documents-section.ts

@@ -10,10 +10,12 @@
 import { LitElement, html, css, nothing } from "lit";
 import { isSafeHttpUrl } from "../helpers/url";
 import { property, state } from "lit/decorators.js";
-import { t, ensureLocale } from "../styles";
+import { t, ensureLocale, langOf } from "../styles";
 import { describeWsError } from "../ws-errors";
 import { downloadUrl } from "../helpers/download";
+import { downloadSignedDocument, openSignedDocument, signDocumentPath } from "../helpers/document-url";
 import { formatBytes } from "../helpers/format-bytes";
+import { CATEGORIES, CATEGORY_ICONS } from "../helpers/document-categories";
 import type { HomeAssistant } from "../types";
 
 interface MaintenanceDocument {
@@ -28,16 +30,6 @@ interface MaintenanceDocument {
   added_at?: string;
 }
 
-const CATEGORIES = ["manual", "warranty", "invoice", "spare_parts", "photo", "other"] as const;
-const CATEGORY_ICONS: Record<string, string> = {
-  manual: "mdi:book-open-variant",
-  warranty: "mdi:shield-check",
-  invoice: "mdi:receipt-text-outline",
-  spare_parts: "mdi:cog-outline",
-  photo: "mdi:image-outline",
-  other: "mdi:file-document-outline",
-};
-
 export class MaintenanceDocumentsSection extends LitElement {
   @property({ attribute: false }) public hass!: HomeAssistant;
   @property({ attribute: false }) public entryId!: string;
@@ -67,16 +59,11 @@ export class MaintenanceDocumentsSection extends LitElement {
   }
 
   private async _sign(doc: MaintenanceDocument): Promise<string> {
-    const signed = await this.hass.connection.sendMessagePromise<{ path: string }>({
-      type: "auth/sign_path",
-      path: `/api/maintenance_supporter/document/${doc.id}`,
-      expires: 300,
-    });
-    return signed.path;
+    return signDocumentPath(this.hass, doc.id);
   }
 
   private get _lang(): string {
-    return this.hass?.language || "en";
+    return langOf(this.hass);
   }
 
   updated(changed: Map<string, unknown>): void {
@@ -210,12 +197,7 @@ export class MaintenanceDocumentsSection extends LitElement {
 
   private async _download(doc: MaintenanceDocument): Promise<void> {
     try {
-      const signed = await this.hass.connection.sendMessagePromise<{ path: string }>({
-        type: "auth/sign_path",
-        path: `/api/maintenance_supporter/document/${doc.id}`,
-        expires: 30,
-      });
-      downloadUrl(signed.path, doc.filename || doc.title || "document");
+      await downloadSignedDocument(this.hass, doc.id, doc.filename || doc.title || "document");
     } catch (e) {
       this._error = describeWsError(e, this._lang);
     }
@@ -228,15 +210,9 @@ export class MaintenanceDocumentsSection extends LitElement {
       this._lightboxUrl = this._thumbs[doc.id] || (await this._sign(doc));
       return;
     }
-    // Open the tab synchronously (in the click gesture) so it isn't popup-blocked,
-    // then point it at the freshly signed URL once it resolves.
-    const win = window.open("about:blank", "_blank");
     try {
-      const url = await this._sign(doc);
-      // Absolute URL so it always resolves against the blank popup (about:blank).
-      if (win) win.location.href = new URL(url, window.location.origin).href;
+      await openSignedDocument(this.hass, doc.id);
     } catch (e) {
-      if (win) win.close();
       this._error = describeWsError(e, this._lang);
     }
   }