From 2304ff48a3186eb29fd87e068c3adb1a44f7c039 Mon Sep 17 00:00:00 2001
From: Aelita4 <kontakt@mikorosa.pl>
Date: Tue, 18 Jun 2024 15:19:52 +0200
Subject: [PATCH] Add planet support

---
 src/components/PlanetView.astro               | 36 +++++++++++
 src/components/ResourceBar.astro              |  4 +-
 src/layouts/Layout.astro                      |  3 +
 src/lib/db/planets.ts                         | 48 +++++++++++++++
 src/lib/db/users.ts                           | 60 -------------------
 src/lib/utils/resourceManager.ts              | 38 ++++++------
 src/pages/api/planets/getAllPlanets.ts        | 22 +++++++
 src/pages/api/planets/getPlanet/[planetId].ts | 44 ++++++++++++++
 src/pages/game/galaxyView.astro               | 13 ++--
 src/pages/game/index.astro                    |  4 +-
 src/pages/login.astro                         | 13 +++-
 src/pages/logout.astro                        |  4 ++
 src/types/Planet.ts                           | 15 +++++
 src/types/User.ts                             | 11 ++--
 14 files changed, 214 insertions(+), 101 deletions(-)
 create mode 100644 src/components/PlanetView.astro
 create mode 100644 src/lib/db/planets.ts
 create mode 100644 src/pages/api/planets/getAllPlanets.ts
 create mode 100644 src/pages/api/planets/getPlanet/[planetId].ts
 create mode 100644 src/types/Planet.ts

diff --git a/src/components/PlanetView.astro b/src/components/PlanetView.astro
new file mode 100644
index 0000000..3eaffa1
--- /dev/null
+++ b/src/components/PlanetView.astro
@@ -0,0 +1,36 @@
+---
+import { ObjectId } from 'mongodb';
+import { getHighestWeightedLanguage, getLocales } from '../lib/utils/langDriver';
+import { getUserWithPlanets } from '../lib/db/users';
+import { createPlanet } from '../lib/db/planets';
+
+// await createPlanet({
+//     name: "BiałePodNapletem2",
+//     ownerId: new ObjectId(Astro.cookies.get('userid')?.value ?? ""),
+//     fields: 69
+// })
+
+if(!Astro.cookies.get('userid')?.value) return;
+const planets = (await getUserWithPlanets(new ObjectId(Astro.cookies.get('userid')?.value ?? "")))?.planets.data;
+if(!planets) return;
+
+// console.log(planets);
+
+// const resourceLang = await getLocales(await getHighestWeightedLanguage(Astro.request.headers.get('accept-language')), 'resources');
+// const resBarLang = await getLocales(await getHighestWeightedLanguage(Astro.request.headers.get('accept-language')), 'resourcebar');
+
+
+---
+<div id="planetview">
+    {planets.map(planet => <div>
+        <b>{planet.name}</b><br />
+        <p>{planet.fields}</p>
+    </div>)}
+</div>
+
+<style>
+#planetview {
+    color: red;
+    background-color: pink;
+}
+</style>
\ No newline at end of file
diff --git a/src/components/ResourceBar.astro b/src/components/ResourceBar.astro
index e0616e8..e051905 100644
--- a/src/components/ResourceBar.astro
+++ b/src/components/ResourceBar.astro
@@ -1,6 +1,6 @@
 ---
 import { ObjectId } from 'mongodb';
-import { calculateCurrentAvailableResources, getUserResources } from '../lib/utils/resourceManager';
+import { calculateCurrentAvailableResources } from '../lib/utils/resourceManager';
 import { getHighestWeightedLanguage, getLocales } from '../lib/utils/langDriver';
 
 import resourceTypes from '../lib/data/resources.json';
@@ -8,7 +8,7 @@ import resourceTypes from '../lib/data/resources.json';
 const resourceLang = await getLocales(await getHighestWeightedLanguage(Astro.request.headers.get('accept-language')), 'resources');
 const resBarLang = await getLocales(await getHighestWeightedLanguage(Astro.request.headers.get('accept-language')), 'resourcebar');
 
-const resources = await calculateCurrentAvailableResources(new ObjectId(Astro.cookies.get('userid')?.value ?? ''));
+const resources = await calculateCurrentAvailableResources(new ObjectId(Astro.cookies.get('planetid')?.value ?? ''));
 
 const resourceArray = [];
 for(const key in resources) {
diff --git a/src/layouts/Layout.astro b/src/layouts/Layout.astro
index 7b552be..0018557 100644
--- a/src/layouts/Layout.astro
+++ b/src/layouts/Layout.astro
@@ -1,4 +1,6 @@
 ---
+import PlanetView from '../components/PlanetView.astro';
+
 interface Props {
 	title: string;
 }
@@ -18,6 +20,7 @@ const { title } = Astro.props;
 	</head>
 	<body>
 		<slot />
+		<!-- <PlanetView /> -->
 	</body>
 </html>
 <style is:global>
diff --git a/src/lib/db/planets.ts b/src/lib/db/planets.ts
new file mode 100644
index 0000000..0182b4b
--- /dev/null
+++ b/src/lib/db/planets.ts
@@ -0,0 +1,48 @@
+import { Planets } from '../db/mongodb';
+import type User from '../../types/User';
+import { ObjectId } from 'mongodb';
+import type Planet from '../../types/Planet';
+import { getUserById } from './users';
+import type Building from '../../types/Building';
+import type Ship from '../../types/Ship';
+import type DBResource from '../../types/DBResource';
+
+export const getAllPlanets = async (options?: { fetchUserPlanets: boolean }) => {
+    const planets = await Planets();
+    const fetched = (await planets.find({}).toArray()).map(async planet => {
+        planet.owner = await getUserById(planet.owner);
+        return planet;
+    })
+
+    return Promise.all(fetched) as unknown as Array<Planet>;
+}
+
+export const createPlanet = async (options: { name: string, owner?: User, ownerId?: ObjectId, fields?: number }) => {    
+    if(options.owner !== undefined && options.ownerId !== undefined) throw new Error("Duplicate identifier");
+
+    let user: ObjectId;
+    if(options.ownerId) user = options.ownerId;
+    else if(options.owner) user = options.owner._id;
+    else throw new Error("Unknown error");
+
+    if(options.fields === undefined) options.fields = 20;
+    
+    //@ts-ignore TODO: find another way of handling IDs from types before inserting to database
+    const planet = {
+        owner: user,
+        name: options.name,
+        fields: options.fields,
+        resources: new Array<DBResource>,
+        buildings: new Array<Building>,
+        ships: new Array<Ship>
+    }
+
+    await (await Planets()).insertOne(planet);
+}
+
+export const getPlanetById = async (id: ObjectId) => {
+    const planets = await Planets();
+    return planets.findOne({
+        _id: id
+    }) as Promise<Planet | null>;
+}
\ No newline at end of file
diff --git a/src/lib/db/users.ts b/src/lib/db/users.ts
index d0e1cfe..6c1677d 100644
--- a/src/lib/db/users.ts
+++ b/src/lib/db/users.ts
@@ -1,13 +1,10 @@
 import { Planets, Users } from '../db/mongodb';
 import type User from '../../types/User';
-import type Resources from '../../types/Resources';
-import type Building from '../../types/Building';
 import type AccessToken from '../../types/AccessToken';
 import { ObjectId } from 'mongodb';
 import { hash } from 'argon2'
 import { createInitialResources } from '../utils/resourceManager';
 import type Research from '../../types/Research';
-import type Ship from '../../types/Ship';
 
 export const getAllUsers = async () => {
     const users = await Users();
@@ -68,36 +65,6 @@ export const updateLastLogin = async (user: User) => {
     return users.updateOne({ username: user.username }, { $set: { lastLogin: new Date() } });
 }
 
-export const getUserBuildings = async (user: User): Promise<Array<Building>> => {
-    // const users = await Users();
-    // const user = await users.findOne({ username });
-
-    const buildings: Array<Building> = [];
-
-    if (user?.buildings !== undefined) {
-        user?.buildings.forEach((building: Building) => {
-            buildings.push(building);
-        });
-    }
-
-    return buildings;
-}
-
-export const createOrUpgradeBuilding = async (username: string, building: Building) => {
-    const users = await Users();
-
-    const result = await users.updateOne(
-        { username, "buildings.name": building.id },
-        { $set: { "buildings.$.level": building.level } }
-    );
-
-    if (result.modifiedCount === 0) {
-        await users.updateOne({ username },
-            { $push: { buildings: building } }
-        );
-    }
-}
-
 export const getUserResearch = async (user: User): Promise<Array<Research>> => {
     const research: Array<Research> = [];
 
@@ -125,33 +92,6 @@ export const createOrUpgradeResearch = async (username: string, research: Resear
     }
 }
 
-export const getUserShips = async (user: User): Promise<Array<Ship>> => {
-    const ships: Array<Ship> = [];
-
-    if (user?.ships !== undefined) {
-        user?.ships.forEach((ship: Ship) => {
-            ships.push(ship);
-        });
-    }
-
-    return ships;
-}
-
-export const createOrUpgradeShip = async (user: User, ship: Ship) => {
-    const users = await Users();
-
-    const result = await users.updateOne(
-        { username: user.username, "ships.id": ship.id },
-        { $set: { "ships.$.amount": ship.amount } }
-    );
-
-    if (result.modifiedCount === 0) {
-        await users.updateOne({ username: user.username },
-            { $push: { ships: ship } }
-        );
-    }
-}
-
 export const getUserWithPlanets = async (id: ObjectId): Promise<User | null> => {
     const users = await Users();
     const planets = await Planets();
diff --git a/src/lib/utils/resourceManager.ts b/src/lib/utils/resourceManager.ts
index 3ac8a05..8e2e002 100644
--- a/src/lib/utils/resourceManager.ts
+++ b/src/lib/utils/resourceManager.ts
@@ -1,10 +1,9 @@
 import { ObjectId } from "mongodb"
-import { getUserById } from "../db/users";
-import { Users } from "../db/mongodb";
+import { Planets } from "../db/mongodb";
 import type DBResource from "../../types/DBResource";
-import type Resource from "../../types/Resource";
+import { getPlanetById } from "../db/planets";
 
-export const createInitialResources = async (id: ObjectId) => {
+export const createInitialResources = async (planetId: ObjectId) => {
     const resources: Array<DBResource> = [
         {
             name: "coal",
@@ -62,34 +61,32 @@ export const createInitialResources = async (id: ObjectId) => {
         }
     ];
 
-    updateUserResources(id, resources);
+    updatePlanetResources(planetId, resources);
 }
 
-export const getUserResources = async (id: ObjectId): Promise<Array<DBResource>> => {
-    const user = await getUserById(id);
+export const getResourcesFromPlanet = async (planetId: ObjectId): Promise<Array<DBResource> | null> => {
+    const planet = await getPlanetById(planetId);
 
-    const defaultResources: Array<DBResource> = [];
+    if(!planet) return null;
 
-    return user?.resources === undefined ? defaultResources : user?.resources;
+    return planet.resources;
 }
 
-export const updateUserResources = async (id: ObjectId, resources: Array<DBResource>) => {
-    const users = await Users();
-    await users.updateOne({ _id: id }, {
+export const updatePlanetResources = async (planetId: ObjectId, resources: Array<DBResource>) => {
+    const planets = await Planets();
+    await planets.updateOne({ _id: planetId }, {
         $set: {
             resources
         }
     });
 }
 
-export const calculateCurrentAvailableResources = async (id: ObjectId): Promise<Array<DBResource>> => {
-    const user = await getUserById(id);
+export const calculateCurrentAvailableResources = async (planetId: ObjectId): Promise<Array<DBResource>> => {
+    const resources = await getResourcesFromPlanet(planetId);
 
-    if(user === null) return [];
+    if(resources === null) return [];
 
-    const userResources = await getUserResources(id);
-
-    userResources.forEach(res => {
+    resources.forEach(res => {
         const timeDiff = Math.abs((new Date()).getTime() - res.lastUpdated.getTime());
         const hours = timeDiff / (1000 * 60 * 60);
         const amountToAdd = hours * res.perHourMiningRate;
@@ -97,6 +94,7 @@ export const calculateCurrentAvailableResources = async (id: ObjectId): Promise<
         res.lastUpdated = new Date();
     });
 
-    updateUserResources(id, userResources);
-    return userResources;
+
+    updatePlanetResources(planetId, resources);
+    return resources;
 }
\ No newline at end of file
diff --git a/src/pages/api/planets/getAllPlanets.ts b/src/pages/api/planets/getAllPlanets.ts
new file mode 100644
index 0000000..c9c5ec8
--- /dev/null
+++ b/src/pages/api/planets/getAllPlanets.ts
@@ -0,0 +1,22 @@
+import { APIRoute } from "astro";
+import { getAllPlanets } from "../../../lib/db/planets";
+import Planet from "../../../types/Planet";
+import { ObjectId } from "mongodb";
+
+export const GET: APIRoute = async ({ request }) => {
+    const planets = await getAllPlanets();
+
+    const response: Array<{planetId: ObjectId, ownerId: ObjectId, name: string}> = [];
+
+    planets.forEach((planet: Planet) => {
+        response.push({
+            planetId: planet._id,
+            ownerId: planet.owner._id,
+            name: planet.name,
+        })
+    });
+
+    return new Response(
+        JSON.stringify(response)
+    );
+}
\ No newline at end of file
diff --git a/src/pages/api/planets/getPlanet/[planetId].ts b/src/pages/api/planets/getPlanet/[planetId].ts
new file mode 100644
index 0000000..3fa32a7
--- /dev/null
+++ b/src/pages/api/planets/getPlanet/[planetId].ts
@@ -0,0 +1,44 @@
+import { APIRoute } from "astro";
+import { getPlanetById } from "../../../../lib/db/planets";
+import { ObjectId } from "mongodb";
+import Planet from "../../../../types/Planet";
+import { calculateCurrentAvailableResources } from "../../../../lib/utils/resourceManager";
+
+export const GET: APIRoute = async ({ params, request }) => {
+    const planetId = params['planetId'];
+    if(!planetId) return new Response(
+        JSON.stringify({
+            code: 400,
+            message: "Bad Request"
+        }), { status: 400 }
+    );
+
+    let planet: Planet | null;
+
+    try {
+        planet = await getPlanetById(new ObjectId(planetId));
+        if(!planet) return new Response(
+            JSON.stringify({
+                code: 404,
+                message: "Not Found"
+            }), { status: 404 }
+        );
+    } catch(e) {
+        return new Response(
+            JSON.stringify({
+                code: 500,
+                message: "Internal Server Error"
+            }), { status: 500 }
+        );
+    }
+
+    await calculateCurrentAvailableResources(planet._id);
+
+    return new Response(
+        JSON.stringify({
+            code: 200,
+            message: "OK",
+            data: JSON.parse(JSON.stringify(planet).replace(/_id/g, 'id'))
+        })
+    )
+}
\ No newline at end of file
diff --git a/src/pages/game/galaxyView.astro b/src/pages/game/galaxyView.astro
index 33419b0..a2cc6d2 100644
--- a/src/pages/game/galaxyView.astro
+++ b/src/pages/game/galaxyView.astro
@@ -1,9 +1,8 @@
 ---
 import Layout from '../../layouts/Layout.astro';
 import NavBar from '../../components/NavBar.astro';
-import { getAllUsers, getUserByAccessToken } from '../../lib/db/users';
-import { getHighestWeightedLanguage, getLocales } from '../../lib/utils/langDriver';
-import format from '../../lib/utils/format';
+import { getUserByAccessToken } from '../../lib/db/users';
+import { getAllPlanets } from '../../lib/db/planets';
 
 const loggedToken = Astro.cookies.get('sessionToken')?.value ?? null;
 const username = Astro.cookies.get('username')?.value ?? "";
@@ -12,17 +11,13 @@ if(loggedToken === null || username === "") return Astro.redirect('/logout');
 const checkUser = await getUserByAccessToken(loggedToken);
 if(checkUser === null || checkUser.username !== username) return Astro.redirect('/logout');
 
-const allUsers = await getAllUsers();
-
-const langGame = await getLocales(await getHighestWeightedLanguage(Astro.request.headers.get('accept-language')), 'game');
+const allPlanets = await getAllPlanets();
 ---
 
 <Layout title="Galaxy view">
 	<NavBar loggedIn="true" active="galaxyView" />
-    <h1>{format(langGame['Header_user'], username)}</h1>
-    <h1>Galaxy view</h1>
     <ul>
-		{allUsers.map(user => <li>{user.username}</li>)}
+		{allPlanets.map(planet => <li>{planet.name} ({planet.owner.username})</li>)}
     </ul>
 </Layout>
 
diff --git a/src/pages/game/index.astro b/src/pages/game/index.astro
index c4d464d..5cd2a17 100644
--- a/src/pages/game/index.astro
+++ b/src/pages/game/index.astro
@@ -3,7 +3,6 @@ import Layout from '../../layouts/Layout.astro';
 import NavBar from '../../components/NavBar.astro';
 import ResourceBar from '../../components/ResourceBar.astro';
 import { getUserByAccessToken } from '../../lib/db/users';
-// import { createInitialResources } from '../../lib/utils/resourceManager';
 
 const loggedToken = Astro.cookies.get('sessionToken')?.value ?? null;
 const username = Astro.cookies.get('username')?.value ?? "";
@@ -11,10 +10,9 @@ if(loggedToken === null || username === "") return Astro.redirect('/logout');
 
 const checkUser = await getUserByAccessToken(loggedToken);
 if(checkUser === null || checkUser.username !== username) return Astro.redirect('/logout');
-// createInitialResources(checkUser._id);
 ---
 
-<Layout title="chujów sto">
+<Layout title="Overview">
 	<NavBar loggedIn="true" active="overview" />
 	<ResourceBar />
 </Layout>
diff --git a/src/pages/login.astro b/src/pages/login.astro
index ba27f2d..6b99c65 100644
--- a/src/pages/login.astro
+++ b/src/pages/login.astro
@@ -2,7 +2,7 @@
 import Layout from '../layouts/Layout.astro';
 import NavBar from '../components/NavBar.astro';
 
-import { getUserByNickOrEmail, updateLastLogin } from '../lib/db/users';
+import { getUserByNickOrEmail, getUserWithPlanets, updateLastLogin } from '../lib/db/users';
 
 import config from '../../config.json';
 
@@ -68,6 +68,17 @@ if(Astro.request.method === "POST") {
             secure: true
         });
 
+        const userWithPlanets = await getUserWithPlanets(user._id);
+        if(!userWithPlanets) return;
+        const planetId = userWithPlanets.planets.data[0]._id;
+
+        Astro.cookies.set("planetid", planetId.toString(), {
+            path: "/",
+            maxAge: sessionTime,
+            sameSite: "lax",
+            secure: true
+        });
+
         return Astro.redirect("/game");
     } else {
         error = "invalid username or password";
diff --git a/src/pages/logout.astro b/src/pages/logout.astro
index 7557ec1..0ebd29c 100644
--- a/src/pages/logout.astro
+++ b/src/pages/logout.astro
@@ -11,5 +11,9 @@ if(Astro.cookies.has('userid')) {
     Astro.cookies.delete('userid');
 }
 
+if(Astro.cookies.has('planetid')) {
+    Astro.cookies.delete('planetid');
+}
+
 return Astro.redirect('/');
 ---
\ No newline at end of file
diff --git a/src/types/Planet.ts b/src/types/Planet.ts
new file mode 100644
index 0000000..c71d418
--- /dev/null
+++ b/src/types/Planet.ts
@@ -0,0 +1,15 @@
+import type { ObjectId } from "mongodb";
+import type DBResource from "./DBResource";
+import type Building from "./Building";
+import type Ship from "./Ship";
+import type User from "./User";
+
+export default interface Planet {
+    _id: ObjectId,
+    owner: User,
+    name: string,
+    fields: number,
+    resources: Array<DBResource>,
+    buildings: Array<Building>,
+    ships: Array<Ship>,
+}
\ No newline at end of file
diff --git a/src/types/User.ts b/src/types/User.ts
index fa7f6c0..c1d4244 100644
--- a/src/types/User.ts
+++ b/src/types/User.ts
@@ -1,8 +1,6 @@
 import type { ObjectId } from "mongodb";
-import type Building from "./Building";
-import type DBResource from "./DBResource";
 import type Research from "./Research";
-import type Ship from "./Ship";
+import type Planet from "./Planet";
 
 export default interface User {
     _id: ObjectId;
@@ -10,10 +8,11 @@ export default interface User {
     email: string;
     password: string;
     lastLogin: Date;
-    resources: Array<DBResource>;
-    buildings: Array<Building>;
     research: Array<Research>;
-    ships: Array<Ship>
+    planets: {
+        partial: boolean;
+        data: Array<Planet>;
+    };
     createdAt: Date;
     updatedAt: Date;
 }
\ No newline at end of file