From 69b32f43b5b42553a9129453ee314d8f38b49954 Mon Sep 17 00:00:00 2001
From: Aelita4 <kontakt@mikorosa.pl>
Date: Fri, 22 Nov 2024 15:26:28 +0100
Subject: [PATCH] Add simple mail system

---
 public/mail-notif.svg                    |  6 ++++
 public/mail.svg                          |  4 +++
 src/components/NavBar.astro              | 16 +++++++++-
 src/lib/classes/managers/FleetManager.ts | 33 ++++++++++++++++++++-
 src/lib/db/mails.ts                      | 33 +++++++++++++++++++++
 src/lib/db/mongodb.ts                    |  5 ++++
 src/pages/game/mail.astro                | 37 ++++++++++++++++++++++++
 src/types/db/DBMail.ts                   | 11 +++++++
 8 files changed, 143 insertions(+), 2 deletions(-)
 create mode 100644 public/mail-notif.svg
 create mode 100644 public/mail.svg
 create mode 100644 src/lib/db/mails.ts
 create mode 100644 src/pages/game/mail.astro
 create mode 100644 src/types/db/DBMail.ts

diff --git a/public/mail-notif.svg b/public/mail-notif.svg
new file mode 100644
index 0000000..8c6c015
--- /dev/null
+++ b/public/mail-notif.svg
@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="utf-8"?><!-- Uploaded to: SVG Repo, www.svgrepo.com, Generator: SVG Repo Mixer Tools -->
+<svg width="800px" height="800px" viewBox="0 0 24 24" fill="none" xmlns="http://www.w3.org/2000/svg">
+<path d="M20 4L3 9.31372L10.5 13.5M20 4L14.5 21L10.5 13.5M20 4L10.5 13.5" stroke="#FFFFFF" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"/>
+<circle cx="7" cy="7" r="4" fill="red" />
+<circle cx="7" cy="7" r="1.5" fill="white" />
+</svg>
\ No newline at end of file
diff --git a/public/mail.svg b/public/mail.svg
new file mode 100644
index 0000000..4c70db9
--- /dev/null
+++ b/public/mail.svg
@@ -0,0 +1,4 @@
+<?xml version="1.0" encoding="utf-8"?><!-- Uploaded to: SVG Repo, www.svgrepo.com, Generator: SVG Repo Mixer Tools -->
+<svg width="800px" height="800px" viewBox="0 0 24 24" fill="none" xmlns="http://www.w3.org/2000/svg">
+<path d="M20 4L3 9.31372L10.5 13.5M20 4L14.5 21L10.5 13.5M20 4L10.5 13.5" stroke="#FFFFFF" stroke-width="2" stroke-linecap="round" stroke-linejoin="round"/>
+</svg>
\ No newline at end of file
diff --git a/src/components/NavBar.astro b/src/components/NavBar.astro
index 8faf30c..eb93594 100644
--- a/src/components/NavBar.astro
+++ b/src/components/NavBar.astro
@@ -1,4 +1,6 @@
 ---
+import { ObjectId } from 'mongodb';
+import { checkForUnreadMails } from '../lib/db/mails';
 import { getHighestWeightedLanguage, getLocales, getName } from '../lib/utils/langDriver';
 
 interface Props {
@@ -133,6 +135,9 @@ const topRow = filteredList.filter(element => element.position === "top");
 const bottomRow = filteredList.filter(element => element.position === "bottom");
 
 const username = Astro.cookies.get('username')?.value ?? "";
+const userId = Astro.cookies.get('userid')?.value ?? 0;
+
+const hasUnreadMail = await checkForUnreadMails(new ObjectId(userId));
 ---
 <nav>
     <div class="row">
@@ -149,6 +154,7 @@ const username = Astro.cookies.get('username')?.value ?? "";
             )}
             <li class="nav-item"><span class="nav-username nav-keep-right">{username}</span></li>
             {loggedIn === "true" ? <li class="nav-item"><img src="/gargamel.png" class="avatar-icon" /></li> : ""}
+            {loggedIn === "true" ? <li class="nav-item"><a href="/game/mail"><img src={`/mail${hasUnreadMail ? "-notif" : ""}.svg?${new Date().getTime()}`} class="mail-icon" /></a></li> : ""}
         </ul>
     </div>
     <div class="row">
@@ -204,7 +210,7 @@ nav ul {
 
 .nav-keep-right {
     position: absolute;
-    right: 80px;
+    right: 140px;
 }
 
 .active {
@@ -216,6 +222,14 @@ nav ul {
     border-radius: 50%;
     border: 1px solid white;
     position: absolute;
+    right: 73px;
+}
+
+.mail-icon {
+    height: 50px;
+    border-radius: 50%;
+    /* border: 1px solid white; */
+    position: absolute;
     right: 13px;
 }
 </style>
\ No newline at end of file
diff --git a/src/lib/classes/managers/FleetManager.ts b/src/lib/classes/managers/FleetManager.ts
index 5a01f7b..bde8196 100644
--- a/src/lib/classes/managers/FleetManager.ts
+++ b/src/lib/classes/managers/FleetManager.ts
@@ -4,6 +4,7 @@ import DBFleet from "../../../types/db/DBFleet";
 import { updateFleet } from "../../db/fleet";
 import { Planet } from "./PlanetManager";
 import SystemManager, { System } from "./SystemManager";
+import { sendMail } from "../../db/mails";
 
 export type Fleet = {
     id: ObjectId,
@@ -77,6 +78,15 @@ export default class FleetManager {
             await this.data.source.resources.updateAmount(this.data.cargo);
             await this.data.source.ships.sync();
             await this.data.source.resources.sync();
+            await sendMail(
+                null,
+                this.data.source instanceof SystemManager ? this.data.source.data.ownedBy.id : this.data.source.system.data.ownedBy.id,
+                this.data.arrivalTime,
+                "Fleet Returned",
+                `Your fleet from ${this.data.destination instanceof SystemManager ? `${this.data.destination.data.name} system` : `planet ${this.data.destination.name}`} has returned.\n
+                Ships: ${this.data.ships.map(ship => `${ship.amount} ${ship.id}`).join(', ')}\n
+                Cargo: ${this.data.cargo.length > 0 ? this.data.cargo.map(cargo => `${cargo.amount} ${cargo.id}`).join(', ') : 'None'}`
+            );
             return true;
         } else {
             switch(this.data.mission) {
@@ -85,17 +95,38 @@ export default class FleetManager {
                 case 'TRANSPORT':
                     await this.data.destination.resources.updateAmount(this.data.cargo);
                     await this.data.destination.resources.sync();
+                    const cargo = JSON.parse(JSON.stringify(this.data.cargo)) as Array<{ id: string, amount: number }>;
                     this.data.cargo = [];
+                    const arrived = new Date(this.data.arrivalTime);
                     await this.initiateReturn();
+                    await sendMail(
+                        null,
+                        this.data.source instanceof SystemManager ? this.data.source.data.ownedBy.id : this.data.source.system.data.ownedBy.id,
+                        arrived,
+                        "Fleet Arrived",
+                        `Your fleet has arrived at ${this.data.destination instanceof SystemManager ? `${this.data.destination.data.name} system` : `planet ${this.data.destination.name}`}.\n
+                        Ships: ${this.data.ships.map(ship => `${ship.amount} ${ship.id}`).join(', ')}\n
+                        Cargo delivered: ${cargo.length > 0 ? cargo.map(cargo => `${cargo.amount} ${cargo.id}`).join(', ') : 'None'}\n
+                        Fleet will return at ${this.data.arrivalTime}`
+                    );
                     return false;
                 case 'TRANSFER':
                     await this.data.destination.resources.updateAmount(this.data.cargo);
                     await this.data.destination.resources.sync();
-                    this.data.cargo = [];
                     for(const ship of this.data.ships) {
                         this.data.destination.ships.addShips(ship.id, ship.amount);
                     }
                     await this.data.destination.ships.sync();
+                    await sendMail(
+                        null,
+                        this.data.source instanceof SystemManager ? this.data.source.data.ownedBy.id : this.data.source.system.data.ownedBy.id,
+                        this.data.arrivalTime,
+                        "Fleet Arrived",
+                        `Your fleet has arrived at ${this.data.destination instanceof SystemManager ? `${this.data.destination.data.name} system` : `planet ${this.data.destination.name}`}.\n
+                        Ships: ${this.data.ships.map(ship => `${ship.amount} ${ship.id}`).join(', ')}\n
+                        Cargo delivered: ${this.data.cargo.length > 0 ? this.data.cargo.map(cargo => `${cargo.amount} ${cargo.id}`).join(', ') : 'None'}\n
+                        Ships will stay at the destination.`
+                    );
                     return true;
             }
         }
diff --git a/src/lib/db/mails.ts b/src/lib/db/mails.ts
new file mode 100644
index 0000000..8e13301
--- /dev/null
+++ b/src/lib/db/mails.ts
@@ -0,0 +1,33 @@
+import { ObjectId } from "mongodb";
+import { Mails } from "./mongodb"
+import DBMail from "../../types/db/DBMail";
+
+export const getMailById = async (id: ObjectId) => {
+    return await (await Mails()).findOne({
+        _id: id
+    }) as DBMail;
+}
+
+export const getMailsByTo = async (to: ObjectId) => {
+    return await (await Mails()).find({
+        to
+    }).toArray() as DBMail[];
+}
+
+export const checkForUnreadMails = async (to: ObjectId) => {
+    return (await (await Mails()).find({
+        to,
+        read: false
+    }).toArray()).length > 0;
+}
+
+export const sendMail = async (from: ObjectId | null, to: ObjectId, date: Date, subject: string, body: string) => {
+    await (await Mails()).insertOne({
+        from,
+        to,
+        date,
+        subject,
+        body,
+        read: false
+    });
+}
\ No newline at end of file
diff --git a/src/lib/db/mongodb.ts b/src/lib/db/mongodb.ts
index 7782473..78f2881 100644
--- a/src/lib/db/mongodb.ts
+++ b/src/lib/db/mongodb.ts
@@ -91,4 +91,9 @@ export const Lang = async (language = "en") => {
         await db.collection('research'),
         await db.collection('structures')
     ]
+}
+
+export const Mails = async() => {
+    const db = await getDB();
+    return db.collection('mails');
 }
\ No newline at end of file
diff --git a/src/pages/game/mail.astro b/src/pages/game/mail.astro
new file mode 100644
index 0000000..698b669
--- /dev/null
+++ b/src/pages/game/mail.astro
@@ -0,0 +1,37 @@
+---
+import LoggedIn from '../../layouts/LoggedIn.astro';
+import locationManager from '../../lib/classes/managers/LocationManager';
+import { Planet } from '../../lib/classes/managers/PlanetManager';
+import SystemManager from '../../lib/classes/managers/SystemManager';
+import { getMailsByTo } from '../../lib/db/mails';
+
+const { user, token, lang } = Astro.locals;
+const active: SystemManager | Planet = Astro.locals.active;
+
+const mails = (await getMailsByTo(user.id)).sort((a, b) => b.date.getTime() - a.date.getTime());
+---
+<LoggedIn id="mail" title="Mail">
+	{mails.map(mail => <div class="mail-card">
+        <h1>{!mail.read && "(NEW) "}{mail.subject} | from: {mail.from === null ? "SYSTEM" : locationManager.getUser(mail.from)?.username ?? "Unknown"}</h1>
+        <p>{mail.date.toLocaleString()}</p>
+        <hr />
+        <p style="white-space: pre-line">{mail.body}</p>
+    </div>)}
+</LoggedIn>
+<style>
+.mail-card {
+    background: #333;
+    padding: 1em;
+    margin: 1em;
+    border-radius: 0.5em;
+}
+
+.mail-card h1 {
+    color: #fff;
+    font-size: 1.5em;
+}
+
+.mail-card p {
+    color: #fff;
+}
+</style>
\ No newline at end of file
diff --git a/src/types/db/DBMail.ts b/src/types/db/DBMail.ts
new file mode 100644
index 0000000..a986772
--- /dev/null
+++ b/src/types/db/DBMail.ts
@@ -0,0 +1,11 @@
+import { ObjectId } from "mongodb";
+
+export default interface DBMail {
+    _id: ObjectId;
+    from: ObjectId | null;
+    to: ObjectId;
+    date: Date;
+    subject: string;
+    body: string;
+    read: boolean;
+}
\ No newline at end of file