From 3cd611ff3821ef88dc6a8c80a2a8b7e253429bca Mon Sep 17 00:00:00 2001
From: Zack Spear <hi@zackspear.com>
Date: Mon, 26 Jun 2023 17:40:13 -0700
Subject: [PATCH] refactor: purchase init callback

---
 store/purchase.ts |  6 +++---
 store/server.ts   | 14 ++++++++++++++
 types/server.ts   |  8 +++++++-
 3 files changed, 24 insertions(+), 4 deletions(-)

diff --git a/store/purchase.ts b/store/purchase.ts
index f15203096..9d16ad3a0 100644
--- a/store/purchase.ts
+++ b/store/purchase.ts
@@ -17,7 +17,7 @@ export const usePurchaseStore = defineStore('purchase', () => {
   // Actions
   const redeem = () => {
     console.debug('[redeem]');
-    callbackStore.send('https://unraid.ddev.site/init-purchase', [{
+    callbackStore.send('https://unraid.ddev.site/callback', [{
       server: {
         ...serverStore.serverPurchasePayload,
       },
@@ -26,7 +26,7 @@ export const usePurchaseStore = defineStore('purchase', () => {
   };
   const purchase = () => {
     console.debug('[purchase]');
-    callbackStore.send('https://unraid.ddev.site/init-purchase', [{
+    callbackStore.send('https://unraid.ddev.site/callback', [{
       server: {
         ...serverStore.serverPurchasePayload,
       },
@@ -35,7 +35,7 @@ export const usePurchaseStore = defineStore('purchase', () => {
   };
   const upgrade = () => {
     console.debug('[upgrade]');
-    callbackStore.send('https://unraid.ddev.site/init-purchase', [{
+    callbackStore.send('https://unraid.ddev.site/callback', [{
       server: {
         ...serverStore.serverPurchasePayload,
       },
diff --git a/store/server.ts b/store/server.ts
index b57b6a271..25a7f43ff 100644
--- a/store/server.ts
+++ b/store/server.ts
@@ -8,6 +8,7 @@ import { useThemeStore } from './theme';
 import type {
   Server,
   ServerAccountCallbackSendPayload,
+  ServerKeyTypeForPurchase,
   ServerPurchaseCallbackSendPayload,
   ServerState,
   ServerStateData,
@@ -33,6 +34,7 @@ export const useServerStore = defineStore('server', () => {
   const csrf = ref<string>(''); // required to make requests to Unraid webgui
   const description = ref<string>('');
   const deviceCount = ref<number>(0);
+  const email = ref<string>('');
   const expireTime = ref<number>(0);
   const flashProduct = ref<string>('');
   const flashVendor = ref<string>('');
@@ -73,6 +75,7 @@ export const useServerStore = defineStore('server', () => {
       avatar: avatar.value,
       description: description.value,
       deviceCount: deviceCount.value,
+      email: email.value,
       expireTime: expireTime.value,
       flashProduct: flashProduct.value,
       flashVendor: flashVendor.value,
@@ -96,11 +99,21 @@ export const useServerStore = defineStore('server', () => {
   });
 
   const serverPurchasePayload = computed((): ServerPurchaseCallbackSendPayload => {
+    let keyTypeForPurchase: ServerKeyTypeForPurchase = 'Trial';
+    switch (state.value) {
+      case 'BASIC': keyTypeForPurchase =  'Basic';
+      case 'PLUS': keyTypeForPurchase =  'Plus';
+      case 'PRO': keyTypeForPurchase =  'Pro';
+    }
     return {
       deviceCount: deviceCount.value,
+      email: email.value,
       guid: guid.value,
+      keyTypeForPurchase,
+      locale: locale.value,
       registered: registered.value ?? false,
       state: state.value,
+      site: site.value,
     }
   });
 
@@ -407,6 +420,7 @@ export const useServerStore = defineStore('server', () => {
     if (typeof data?.csrf !== 'undefined') csrf.value = data.csrf;
     if (typeof data?.description !== 'undefined') description.value = data.description;
     if (typeof data?.deviceCount !== 'undefined') deviceCount.value = data.deviceCount;
+    if (typeof data?.email !== 'undefined') email.value = data.email;
     if (typeof data?.expireTime !== 'undefined') expireTime.value = data.expireTime;
     if (typeof data?.flashProduct !== 'undefined') flashProduct.value = data.flashProduct;
     if (typeof data?.flashVendor !== 'undefined') flashVendor.value = data.flashVendor;
diff --git a/types/server.ts b/types/server.ts
index cb560e57a..ee6c247fe 100644
--- a/types/server.ts
+++ b/types/server.ts
@@ -26,6 +26,7 @@ export interface Server {
   csrf?: string;
   description?: string;
   deviceCount?: number;
+  email?: string;
   expireTime?: number;
   flashProduct?: string;
   flashVendor?: string;
@@ -68,12 +69,17 @@ export interface ServerAccountCallbackSendPayload {
   wanFQDN?: string;
 }
 
+export type ServerKeyTypeForPurchase = 'Basic'|'Plus'|'Pro'|'Trial';
+
 export interface ServerPurchaseCallbackSendPayload {
   deviceCount: number;
+  email: string;
   guid: string;
+  keyTypeForPurchase: ServerKeyTypeForPurchase;
+  locale: string;
   registered: boolean;
-  // state?: ServerState;
   state: string;
+  site: string;
 }
 
 export type ServerStateDataActionType = 'signIn'|'signOut'|'purchase'|'redeem'|'upgrade'|'recover'|'replace'|'trialExtend'|'trialStart';