From 326a2d3c271587976097e1a33f66fb8046a4338b Mon Sep 17 00:00:00 2001
From: Sidhin S Thomas <thomas.sidhin@outlook.com>
Date: Wed, 14 Aug 2024 02:20:27 +0530
Subject: [PATCH] Integrate link budget

---
 ui/components/header.tsx            | 20 ++++++-
 ui/components/link_budget_modal.tsx | 87 +++++++++++++++++++++++++++++
 ui/services/share_service.ts        | 11 +++-
 3 files changed, 114 insertions(+), 4 deletions(-)
 create mode 100644 ui/components/link_budget_modal.tsx

diff --git a/ui/components/header.tsx b/ui/components/header.tsx
index b515c3d..1b31ad1 100644
--- a/ui/components/header.tsx
+++ b/ui/components/header.tsx
@@ -25,6 +25,7 @@ import { connect } from "react-redux";
 import { budgetSlice, headerSlice, navigate, store, View } from "../store";
 import { GetScreenSize, ScreenSize } from "../utils";
 import { ShareBudgetModal } from "./share_budget_modal";
+import { LinkBudgetModal } from "./link_budget_modal";
 
 export interface HeaderBudgetDetails {
     name: string;
@@ -42,6 +43,7 @@ export interface HeaderProps {
 interface HeaderState {
     is_budget_selector_visible: boolean;
     open_share_modal: boolean;
+    open_link_modal: boolean;
 }
 
 class Header extends React.Component<HeaderProps, HeaderState> {
@@ -73,6 +75,7 @@ class Header extends React.Component<HeaderProps, HeaderState> {
         this.state = {
             is_budget_selector_visible: false,
             open_share_modal: false,
+            open_link_modal: false,
         };
     }
 
@@ -96,15 +99,20 @@ class Header extends React.Component<HeaderProps, HeaderState> {
             store.dispatch(budgetSlice.actions.updateSelection({ index: null }));
             store.dispatch(navigate(View.NoBudgetAvailable));
         };
+        const linkBudget = (e: React.MouseEvent<HTMLButtonElement | HTMLAnchorElement>) => {
+            this.setState({ open_link_modal: true });
+            e.preventDefault();
+        }
         const onSelectionChanged = (value: number) => {
             store.dispatch(budgetSlice.actions.updateSelection({ index: value }));
-        }
+        };
         const onSettingsClicked = () => {
             this.setState({ is_budget_selector_visible: false });
             store.dispatch(navigate(View.NoBudgetAvailable));
-        }
+        };
 
         return (
+            <>
             <Select
                 size="large"
                 style={{ width: 300 }}
@@ -123,7 +131,7 @@ class Header extends React.Component<HeaderProps, HeaderState> {
                             <Button type="text" icon={<PlusOutlined />} onClick={addItem}>
                                 New
                             </Button>
-                            <Button type="text" icon={<LinkOutlined />} disabled onClick={addItem}>
+                            <Button type="text" icon={<LinkOutlined />}  onClick={linkBudget}>
                                 Link
                             </Button>
                         </Flex>
@@ -136,6 +144,12 @@ class Header extends React.Component<HeaderProps, HeaderState> {
                 )}
                 variant="outlined"
             />
+            <LinkBudgetModal 
+                isOpen={this.state.open_link_modal} 
+                onDone={() => { store.dispatch(navigate(View.Overview)); }}
+                onClose={() => { this.setState({ open_link_modal: false }) }}
+            />
+            </>
         );
     }
 
diff --git a/ui/components/link_budget_modal.tsx b/ui/components/link_budget_modal.tsx
new file mode 100644
index 0000000..2987a08
--- /dev/null
+++ b/ui/components/link_budget_modal.tsx
@@ -0,0 +1,87 @@
+/* 
+ * BudgetBug - Budgeting and Expense Tracker with WebUI and API server
+ * Copyright (C) 2024  Sidhin S Thomas <sidhin.thomas@gmail.com>
+ * 
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as published
+ * by the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ * 
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ * 
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <https://www.gnu.org/licenses/>.
+ * 
+ * The source is available at: https://github.com/ParadoxZero/budgetbud
+ */
+
+import { Button, Flex, Input, message, Modal, Typography } from "antd";
+import { useState } from "react";
+import { LinkBudget } from "../services/share_service";
+import { InputStatus } from "antd/es/_util/statusUtils";
+
+export function LinkBudgetModal(props: { isOpen: boolean, onDone: () => void, onClose: () => void }) {
+    const [confirmLoading, setConfirmLoading] = useState(false);
+    const [validationStatus, setValidationStatus] = useState("");
+    const [messageApi, contextHolder] = message.useMessage();
+
+    const handle_ok = () => {
+        setConfirmLoading(true);
+        const code = (document.getElementById('share_code_id') as HTMLInputElement).value
+        if (code.length == 0) {
+            messageApi.error("Please enter the share code.", 5);
+            setValidationStatus("error");
+            setConfirmLoading(false);
+            return;
+        } 
+        LinkBudget(code).then(() => {
+            props.onDone();
+            messageApi.error("Succesfully linked the budget.", 5);
+        }).catch(()=>{
+            messageApi.error("Failed to link budget.", 5);
+        }).finally(() => {
+            setConfirmLoading(false);
+            props.onClose();
+        });
+       
+    }
+    let status: InputStatus = "";
+    switch (validationStatus) {
+        case "error":
+            status = "error";
+            break;
+        case "warning":
+            status = "warning";
+            break;
+    }
+    return (
+        <>
+            {contextHolder}
+            <Modal
+                 centered
+                 closable={true}
+                 destroyOnClose
+                 keyboard
+                 mask
+                 width={300}
+                 maskClosable
+                 confirmLoading={confirmLoading}
+                 onCancel={props.onClose}
+                 title="Link Budget"
+                 footer={(
+                     <Flex style={{ width: "100%" }} align="center" justify="center">
+                         <Button type="primary" size="large" onClick={handle_ok}>Ok</Button>
+                     </Flex>
+                 )}
+                 open={props.isOpen}>
+                <Flex vertical align="center" justify="center">
+                    <Typography.Paragraph>Linking budget will allow you to share the budget with others.</Typography.Paragraph>
+                    <Input id="share_code_id" size="large" placeholder="Enter Share Code" status={status} />
+                </Flex>
+                </Modal>  
+        </>
+    );
+}
\ No newline at end of file
diff --git a/ui/services/share_service.ts b/ui/services/share_service.ts
index 2e257c7..71632c9 100644
--- a/ui/services/share_service.ts
+++ b/ui/services/share_service.ts
@@ -26,9 +26,18 @@ export interface ShareKeyResponse {
 
 export async function GetShareKey(budget_id: string): Promise<ShareKeyResponse> {
     if (import.meta.env.VITE_USE_LOCAL_DATA_SERVICE === 'true') {
-        await new Promise((resolve,_reject) => _reject("Failed"));
+        await new Promise((resolve,_reject) => setTimeout(resolve, 3000));
         return { shareKey: "2U0T0Y05" };
     }
     const response = await fetchData(`/api/Social/share_code/${budget_id}`, {method:"POST"});
     return await response.json();
+}
+
+export async function LinkBudget(share_key: string): Promise<void> {
+    if (import.meta.env.VITE_USE_LOCAL_DATA_SERVICE === 'true') {
+        await new Promise((resolve,_reject) => setTimeout(resolve, 3000));
+        return;
+    }
+    await fetchData(`/api/Social/link/${share_key}`, {method:"POST"});
+    return;
 }
\ No newline at end of file