From 9274020aea13394ecf18ebc5320de4f96c675128 Mon Sep 17 00:00:00 2001
From: butschster <butschster@gmail.com>
Date: Sun, 5 May 2024 11:03:22 +0400
Subject: [PATCH] Refactors SMTP attachments fetching from HTTP endpoint

---
 pages/smtp/[id].vue                           | 18 ++++++--
 src/entities/smtp/lib/use-smtp.ts             | 11 +++--
 src/screens/smtp/ui/smtp-page/smtp-page.vue   | 30 ++++++--------
 src/shared/lib/io/index.ts                    |  1 +
 src/shared/lib/io/use-smtp-requests.ts        | 41 +++++++++++++++++++
 src/shared/lib/use-smtp/index.ts              |  1 +
 src/shared/lib/use-smtp/use-smtp-api.ts       | 16 ++++++++
 src/shared/types/events.ts                    |  9 ++++
 .../ui/file-attachment/file-attachment.vue    |  2 +-
 9 files changed, 105 insertions(+), 24 deletions(-)
 create mode 100644 src/shared/lib/io/use-smtp-requests.ts
 create mode 100644 src/shared/lib/use-smtp/index.ts
 create mode 100644 src/shared/lib/use-smtp/use-smtp-api.ts

diff --git a/pages/smtp/[id].vue b/pages/smtp/[id].vue
index 6781dba3..36791bea 100644
--- a/pages/smtp/[id].vue
+++ b/pages/smtp/[id].vue
@@ -7,7 +7,7 @@ import { useSmtp } from "~/src/entities/smtp";
 import type { SMTP } from "~/src/entities/smtp/types";
 import { REST_API_URL } from "~/src/shared/lib/io";
 import { useEvents } from "~/src/shared/lib/use-events";
-import type { EventId, ServerEvent } from "~/src/shared/types";
+import type { Attachment, EventId, ServerEvent } from "~/src/shared/types";
 
 const { normalizeSmtpEvent } = useSmtp();
 
@@ -21,9 +21,10 @@ useHead({
 });
 
 const { events } = useEvents();
-
+const { smtp } = useSmtp();
 const isLoading = ref(false);
 const serverEvent = ref<Event | null>(null);
+const serverAttachments = ref<Attachment[]>([]);
 
 const event = computed(() =>
   serverEvent.value
@@ -31,6 +32,8 @@ const event = computed(() =>
     : null
 );
 
+const attachments = computed(() => serverAttachments.value);
+
 const html = computed(
   () => `<iframe src="${REST_API_URL}/api/smtp/${eventId}/html"/>`
 );
@@ -57,6 +60,11 @@ const getEvent = async () => {
       router.push("/404");
     },
   });
+
+  await smtp.getAttachments(eventId)
+    .then((attachments: Attachment[]) => {
+      serverAttachments.value = attachments;
+    });
 };
 
 onMounted(getEvent);
@@ -81,7 +89,11 @@ onMounted(getEvent);
     </div>
 
     <div class="smtp-event__body">
-      <SmtpPage v-if="event" :event="event" :html-source="html" />
+      <SmtpPage v-if="event"
+                :event="event"
+                :attachments="attachments"
+                :html-source="html"
+      />
     </div>
   </main>
 </template>
diff --git a/src/entities/smtp/lib/use-smtp.ts b/src/entities/smtp/lib/use-smtp.ts
index 561fce99..dae1c372 100644
--- a/src/entities/smtp/lib/use-smtp.ts
+++ b/src/entities/smtp/lib/use-smtp.ts
@@ -1,11 +1,16 @@
 import type { ServerEvent, NormalizedEvent } from '~/src/shared/types';
 import type { SMTP } from "../types";
 import { normalizeSmtpEvent } from "./normalize-smtp-event";
+import { type TUseSmtpApi, useSmtpApi } from "~/src/shared/lib/use-smtp";
 
 type TUseSmtp = () => {
   normalizeSmtpEvent: (event: ServerEvent<SMTP>) => NormalizedEvent<SMTP>
+  smtp: TUseSmtpApi
 }
 
-export const useSmtp: TUseSmtp = () => ({
-  normalizeSmtpEvent
-})
+export const useSmtp: TUseSmtp = () => {
+  return {
+    normalizeSmtpEvent,
+    smtp: useSmtpApi()
+  }
+}
diff --git a/src/screens/smtp/ui/smtp-page/smtp-page.vue b/src/screens/smtp/ui/smtp-page/smtp-page.vue
index 5359b948..a6004809 100644
--- a/src/screens/smtp/ui/smtp-page/smtp-page.vue
+++ b/src/screens/smtp/ui/smtp-page/smtp-page.vue
@@ -15,10 +15,13 @@ import { SmtpPagePreview } from "../smtp-page-preview";
 
 type Props = {
   event: NormalizedEvent<SMTP>;
+  attachments: Attachment[];
   htmlSource: string;
 };
 
-const props = defineProps<Props>();
+const props = withDefaults(defineProps<Props>(), {
+  attachments: [],
+});
 
 const htmlSource = ref(props.htmlSource || props.event.payload.html);
 
@@ -60,12 +63,6 @@ const mail = computed(() => props.event.payload);
 const date = computed(() =>
   moment(props.event.date).format("DD.MM.YYYY HH:mm:ss")
 );
-
-const attachments = computed(() =>
-  Array.isArray(props.event.payload.attachments)
-    ? props.event.payload.attachments
-    : (Object.values(props.event.payload.attachments || {}) as Attachment[])
-);
 </script>
 
 <template>
@@ -108,7 +105,7 @@ const attachments = computed(() =>
             suffix="<span class='smtp-page__body-tab-badge'>HTML</span>"
           >
             <SmtpPagePreview device="tablet">
-              <div v-html="htmlSource" />
+              <div v-html="htmlSource"/>
             </SmtpPagePreview>
           </Tab>
           <Tab v-if="isHtml" name="HTML">
@@ -127,14 +124,13 @@ const attachments = computed(() =>
           </Tab>
           <Tab
             v-if="attachments.length"
-            name="Attachments"
-            :suffix="`<span class='smtp-page__body-tab-badge'>${attachments.length}</span>`"
+            :name="`Attachments (${attachments.length})`"
           >
             <section class="mb-5">
               <div class="flex gap-x-3">
                 <FileAttachment
                   v-for="a in attachments"
-                  :key="a.id"
+                  :key="a.uuid"
                   :event-id="event.id"
                   :event="event"
                   :attachment="a"
@@ -143,7 +139,7 @@ const attachments = computed(() =>
             </section>
           </Tab>
           <Tab name="Raw">
-            <CodeSnippet language="html" :code="event.payload.raw" />
+            <CodeSnippet language="html" :code="event.payload.raw"/>
           </Tab>
           <Tab name="Tech Info">
             <section>
@@ -156,22 +152,22 @@ const attachments = computed(() =>
                   {{ event.payload.subject }}
                 </TableBaseRow>
                 <TableBaseRow title="From">
-                  <SmtpPageAddresses :addresses="event.payload.from" />
+                  <SmtpPageAddresses :addresses="event.payload.from"/>
                 </TableBaseRow>
                 <TableBaseRow title="To">
-                  <SmtpPageAddresses :addresses="event.payload.to" />
+                  <SmtpPageAddresses :addresses="event.payload.to"/>
                 </TableBaseRow>
                 <TableBaseRow v-if="event.payload.cc.length" title="Cc">
-                  <SmtpPageAddresses :addresses="event.payload.cc" />
+                  <SmtpPageAddresses :addresses="event.payload.cc"/>
                 </TableBaseRow>
                 <TableBaseRow v-if="event.payload.bcc.length" title="Bcc">
-                  <SmtpPageAddresses :addresses="event.payload.bcc" />
+                  <SmtpPageAddresses :addresses="event.payload.bcc"/>
                 </TableBaseRow>
                 <TableBaseRow
                   v-if="event.payload.reply_to.length"
                   title="Reply to"
                 >
-                  <SmtpPageAddresses :addresses="event.payload.reply_to" />
+                  <SmtpPageAddresses :addresses="event.payload.reply_to"/>
                 </TableBaseRow>
               </TableBase>
             </section>
diff --git a/src/shared/lib/io/index.ts b/src/shared/lib/io/index.ts
index b7f0cfcc..e7cd4297 100644
--- a/src/shared/lib/io/index.ts
+++ b/src/shared/lib/io/index.ts
@@ -1,5 +1,6 @@
 export * from './constants';
 export * from './types';
 export * from './use-events-requests';
+export * from './use-smtp-requests';
 export * from './logger';
 export * from './centrifuge';
diff --git a/src/shared/lib/io/use-smtp-requests.ts b/src/shared/lib/io/use-smtp-requests.ts
new file mode 100644
index 00000000..d60babbc
--- /dev/null
+++ b/src/shared/lib/io/use-smtp-requests.ts
@@ -0,0 +1,41 @@
+import { Attachment, EventId } from "../../types";
+import { type NuxtApp, useNuxtApp } from "#app"; // eslint-disable-line @conarti/feature-sliced/layers-slices
+import { REST_API_URL } from "./constants";
+
+type TUseSmtpRequests = () => {
+  getAttachments: (id: EventId) => Promise<Attachment[]>
+}
+
+// TODO: add 403 response handling
+
+export const useSmtpRequests: TUseSmtpRequests = () => {
+
+  const app: NuxtApp = useNuxtApp()
+  const { token } = app.$authToken ?? { token: null }
+  const headers = { "X-Auth-Token": token || '' }
+
+  const getAttachmentsRestUrl = (id: EventId): string => `${REST_API_URL}/api/smtp/${id}/attachments`
+
+  const getAttachments = (id: EventId) => fetch(getAttachmentsRestUrl(id), { headers })
+    .then((response) => response.json())
+    .then((response) => {
+      if (response?.data) {
+        return response.data as Attachment[]
+      }
+
+      if (response?.code === 403) {
+        console.error('Forbidden')
+        return [];
+      }
+
+      console.error('Fetch Error')
+
+      return [];
+    })
+    .then((attachments: Attachment[]) => attachments)
+
+  return {
+    getAttachmentsRestUrl,
+    getAttachments
+  }
+}
diff --git a/src/shared/lib/use-smtp/index.ts b/src/shared/lib/use-smtp/index.ts
new file mode 100644
index 00000000..af9ec530
--- /dev/null
+++ b/src/shared/lib/use-smtp/index.ts
@@ -0,0 +1 @@
+export * from "./use-smtp-api";
diff --git a/src/shared/lib/use-smtp/use-smtp-api.ts b/src/shared/lib/use-smtp/use-smtp-api.ts
new file mode 100644
index 00000000..3ffc8b7c
--- /dev/null
+++ b/src/shared/lib/use-smtp/use-smtp-api.ts
@@ -0,0 +1,16 @@
+import type { Attachment, EventId } from '../../types';
+import { useSmtpRequests } from "../io";
+
+export type TUseSmtpApi = {
+  getAttachments: (id: EventId) => Promise<Attachment[]>
+}
+
+export const useSmtpApi: TUseSmtpApi = () => {
+  const {
+    getAttachments
+  } = useSmtpRequests()
+
+  return {
+    getAttachments
+  }
+}
diff --git a/src/shared/types/events.ts b/src/shared/types/events.ts
index 2d120086..2de5aab6 100644
--- a/src/shared/types/events.ts
+++ b/src/shared/types/events.ts
@@ -14,6 +14,7 @@ export enum EVENT_TYPES {
 }
 
 export type EventId = string;
+export type Uuid = string;
 
 export type EventType = OneOfValues<typeof EVENT_TYPES>;
 
@@ -35,3 +36,11 @@ export interface NormalizedEvent<T> {
   date: Date | null,
   payload: T
 }
+
+export interface Attachment {
+  uuid: Uuid,
+  name: string,
+  path: string,
+  size: number,
+  mime: string,
+}
diff --git a/src/shared/ui/file-attachment/file-attachment.vue b/src/shared/ui/file-attachment/file-attachment.vue
index bc9d0cf4..f5601de5 100644
--- a/src/shared/ui/file-attachment/file-attachment.vue
+++ b/src/shared/ui/file-attachment/file-attachment.vue
@@ -15,7 +15,7 @@ const props = defineProps<Props>();
 const size = computed(() => formatFileSize(props.attachment.size || 0));
 const downloadUrl = computed(
   () =>
-    `${REST_API_URL}/api/smtp/${props.eventId}/attachment/${props.attachment.id}`
+    `${REST_API_URL}/api/smtp/${props.eventId}/attachments/${props.attachment.uuid}`
 );
 </script>