feat(common): adds sql support

Author: bc-chaz

lib/dbs/mysql.ts

@@ -1,6 +1,7 @@
 import * as mysql from 'mysql';
 import { promisify } from 'util';
 import { SessionProps, StoreData } from '../../types';
+import { trialDays } from '../checkout';
 
 const MYSQL_CONFIG = {
     host: process.env.MYSQL_HOST,
@@ -26,12 +27,17 @@ export async function setUser({ user }: SessionProps) {
 }
 
 export async function setStore(session: SessionProps) {
-    const { access_token: accessToken, context, scope } = session;
+    const {
+        access_token: accessToken,
+        context,
+        scope,
+        user: { id },
+    } = session;
     // Only set on app install or update
     if (!accessToken || !scope) return null;
 
     const storeHash = context?.split('/')[1] || '';
-    const storeData: StoreData = { accessToken, scope, storeHash };
+    const storeData: StoreData = { accessToken, adminId: id, scope, storeHash };
 
     await query('REPLACE INTO stores SET ?', storeData);
 }
@@ -91,3 +97,47 @@ export async function getStoreToken(storeHash: string) {
 export async function deleteStore({ store_hash: storeHash }: SessionProps) {
     await query('DELETE FROM stores WHERE storeHash = ?', storeHash);
 }
+
+// CHECKOUT Functions
+export async function setStorePlan(session: SessionProps) {
+    const { access_token: accessToken, context, plan, sub } = session;
+    // Only set on app install or subscription verification (load)
+    if ((!accessToken && !plan?.pid) || (plan && !plan.isPaidApp)) return null;
+
+    const contextString = context ?? sub;
+    const storeHash = contextString?.split('/')[1] || '';
+    const defaultEnd = Date.now() + (trialDays * 24 * 60 * 60 * 1000);
+    const data = { pid: '', isPaidApp: false, showPaidWelcome: false, storeHash, trialEndDate: defaultEnd, ...plan };
+
+    await query('REPLACE INTO plan SET ?', data);
+}
+
+export async function getStorePlan(storeHash: string) {
+    if (!storeHash) return null;
+
+    const results = await query('SELECT * FROM plan WHERE storeHash = ? LIMIT 1', storeHash);
+
+    return results.length ? results[0] : null;
+}
+
+export async function setStoreWelcome(storeHash: string, show: boolean) {
+    if (!storeHash) return null;
+
+    const values = [show, storeHash];
+
+    await query('UPDATE plan SET showPaidWelcome = ? WHERE storeHash = ?', values);
+}
+
+export async function setCheckoutId(pid: string, checkoutId: string) {
+    if (!pid || !checkoutId) return null;
+
+    await query('REPLACE INTO checkout SET ?', { pid, checkoutId });
+}
+
+export async function getCheckoutId(pid: string) {
+    if (!pid) return null;
+
+    const results = await query('SELECT checkoutId FROM checkout WHERE pid = ?', pid);
+
+    return results.length ? results[0].checkoutId : null;
+}

scripts/db.js

@@ -26,6 +26,7 @@ const storesCreate = query('CREATE TABLE `stores` (\n' +
     '  `id` int(11) unsigned NOT NULL AUTO_INCREMENT,\n' +
     '  `storeHash` varchar(10) NOT NULL,\n' +
     '  `accessToken` text,\n' +
+    '  `adminId` int(11) NOT NULL,\n' +
     '  `scope` text,\n' +
     '  PRIMARY KEY (`id`),\n' +
     '  UNIQUE KEY `storeHash` (`storeHash`)\n' +
@@ -42,6 +43,27 @@ const storeUsersCreate = query('CREATE TABLE `storeUsers` (\n' +
     ') ENGINE=InnoDB AUTO_INCREMENT=5 DEFAULT CHARSET=utf8;\n'
 );
 
-Promise.all([usersCreate, storesCreate]).then(() => {
+const planCreate = query('CREATE TABLE `plan` (\n' +
+    '  `id` int(11) unsigned NOT NULL AUTO_INCREMENT,\n' +
+    '  `storeHash` varchar(10) NOT NULL,\n' +
+    '  `pid` varchar(20) NOT NULL,\n' +
+    '  `isPaidApp` boolean,\n' +
+    '  `showPaidWelcome` boolean,\n' +
+    '  `trialEndDate` bigint(20),\n' +
+    '  PRIMARY KEY (`id`),\n' +
+    '  UNIQUE KEY `storeHash` (`storeHash`)\n' +
+    ') ENGINE=InnoDB AUTO_INCREMENT=5 DEFAULT CHARSET=utf8;'
+);
+
+const checkoutCreate = query('CREATE TABLE `checkout` (\n' +
+    '  `id` int(11) unsigned NOT NULL AUTO_INCREMENT,\n' +
+    '  `pid` varchar(20) NOT NULL,\n' +
+    '  `checkoutId` varchar(40) NOT NULL,\n' +
+    '  PRIMARY KEY (`id`),\n' +
+    '  UNIQUE KEY `pid` (`pid`)\n' +
+    ') ENGINE=InnoDB AUTO_INCREMENT=5 DEFAULT CHARSET=utf8;\n'
+);
+
+Promise.all([usersCreate, storesCreate, storeUsersCreate, planCreate, checkoutCreate]).then(() => {
     connection.end();
 });

types/db.ts

@@ -4,6 +4,7 @@ import { SessionProps } from './index';
 
 export interface StoreData {
     accessToken?: string;
+    adminId: number;
     scope?: string;
     storeHash: string;
 }