add user assertion and update User constructor

Author: joaotonaco

src/assertions/user.ts

@@ -0,0 +1,42 @@
+import { ApplicationLanguage, UserPlanName } from "@squarecloud/api-types/v2";
+import * as z from "zod";
+import { SquareCloudAPIError } from "..";
+
+const userSchema = z.object({
+  id: z.string(),
+  tag: z.string(),
+  locale: z.string(),
+  email: z.string(),
+  plan: z.object({
+    name: z.nativeEnum(UserPlanName),
+    memory: z.object({
+      limit: z.number(),
+      available: z.number(),
+      used: z.number(),
+    }),
+    duration: z.number().nullable(),
+  }),
+});
+
+const userApplicationSchema = z.object({
+  id: z.string(),
+  tag: z.string(),
+  desc: z.string().optional(),
+  ram: z.number(),
+  lang: z.nativeEnum(ApplicationLanguage),
+  cluster: z.string(),
+  isWebsite: z.boolean(),
+});
+
+const userInfoSchema = z.object({
+  user: userSchema,
+  applications: z.array(userApplicationSchema),
+});
+
+export function assertUserInfo(value: unknown): asserts value is z.infer<typeof userInfoSchema> {
+  try {
+    userInfoSchema.parse(value);
+  } catch {
+    throw new SquareCloudAPIError("INVALID_API_USER_INFO", "Invalid user info object received from API /user");
+  }
+}

src/structures/user.ts

@@ -1,3 +1,4 @@
+import { assertUserInfo } from "@/assertions/user";
 import { UserPlan } from "@/types/user";
 import { APIUserInfo } from "@squarecloud/api-types/v2";
 import { BaseApplication, Collection, SquareCloudAPI } from "..";
@@ -24,15 +25,21 @@ export class User {
   applications: Collection<string, BaseApplication>;
 
   constructor(client: SquareCloudAPI, data: APIUserInfo) {
-    this.id = data.user.id;
-    this.tag = data.user.tag;
-    this.locale = data.user.locale;
+    assertUserInfo(data);
+
+    const { user, applications } = data;
+    const { id, tag, locale, plan, email } = user;
+    const { duration } = plan;
+
+    this.id = id;
+    this.tag = tag;
+    this.locale = locale;
+    this.email = email;
     this.plan = {
-      ...data.user.plan,
-      expiresInTimestamp: data.user.plan.duration ?? undefined,
-      expiresIn: data.user.plan.duration ? new Date(data.user.plan.duration) : undefined,
+      ...plan,
+      expiresInTimestamp: duration ?? undefined,
+      expiresIn: duration ? new Date(duration) : undefined,
     };
-    this.email = data.user.email;
-    this.applications = new Collection(data.applications.map((app) => [app.id, new BaseApplication(client, app)]));
+    this.applications = new Collection(applications.map((app) => [app.id, new BaseApplication(client, app)]));
   }
 }