feat: ref codes

src/api.ts

@@ -187,12 +187,13 @@ export class LNVpsApi {
     return data;
   }
 
-  async orderVm(template_id: number, image_id: number, ssh_key_id: number) {
+  async orderVm(template_id: number, image_id: number, ssh_key_id: number, ref_code?: string) {
     const { data } = await this.#handleResponse<ApiResponse<VmInstance>>(
       await this.#req("/api/v1/vm", "POST", {
         template_id,
         image_id,
         ssh_key_id,
+        ref_code
       }),
     );
     return data;

src/pages/layout.tsx

@@ -1,7 +1,10 @@
 import { Link, Outlet } from "react-router-dom";
 import LoginButton from "../components/login-button";
+import { saveRefCode } from "../ref";
+
 
 export default function Layout() {
+  saveRefCode();
   return (
     <div className="w-[700px] mx-auto m-2 p-2">
       <div className="flex items-center justify-between mb-4">

src/pages/order.tsx

@@ -8,6 +8,7 @@ import classNames from "classnames";
 import VpsResources from "../components/vps-resources";
 import OsImageName from "../components/os-image-name";
 import SSHKeySelector from "../components/ssh-keys";
+import { clearRefCode, getRefCode } from "../ref";
 
 export default function OrderPage() {
   const { state } = useLocation();
@@ -29,7 +30,9 @@ export default function OrderPage() {
 
     setOrderError("");
     try {
-      const newVm = await login.api.orderVm(template.id, useImage, useSshKey);
+      const ref = getRefCode();
+      const newVm = await login.api.orderVm(template.id, useImage, useSshKey, ref?.code);
+      clearRefCode();
       navigate("/vm/renew", {
         state: newVm,
       });

src/ref.ts

@@ -0,0 +1,34 @@
+export interface RefCode {
+    code: string;
+    saved: number;
+}
+
+export function saveRefCode() {
+    const search = new URLSearchParams(window.location.search);
+    const code = search.get("ref");
+    if (code) {
+        // save or overwrite new code from landing
+        window.localStorage.setItem("ref", JSON.stringify({
+            code,
+            saved: Math.floor(new Date().getTime() / 1000)
+        }));
+        window.location.search = "";
+    }
+}
+
+export function getRefCode() {
+    const ref = window.localStorage.getItem("ref");
+    if (ref) {
+        const refObj = JSON.parse(ref) as RefCode;
+        const now = Math.floor(new Date().getTime() / 1000);
+        // treat code as stale if > 7days old
+        if (Math.abs(refObj.saved - now) > 604800) {
+            window.localStorage.removeItem("ref");
+        }
+        return refObj;
+    }
+}
+
+export function clearRefCode() {
+    window.localStorage.removeItem("ref");
+}