dreamlab pro

Author: CodedRedGIT

client/src/auth.ts

@@ -8,6 +8,7 @@ export type AuthToken = {
   playerId: string;
   token: string;
   guest?: boolean;
+  isPro?: boolean;
 };
 
 const authToken = async (): Promise<AuthToken> => {
@@ -55,6 +56,7 @@ const TokenSchema = z.object({
   instance_id: z.uuid(),
   nickname: z.string(),
   player_id: z.string(),
+  is_pro: z.boolean().optional(),
 
   // this is not functional currently
   // world: z.string(),
@@ -67,6 +69,7 @@ const decodeToken = (token: string): AuthToken => {
     token,
     nickname: claims.nickname,
     playerId: claims.player_id,
+    isPro: claims.is_pro,
   };
 };
 
@@ -75,7 +78,7 @@ const devAuth = (nickname: string): AuthToken => {
   const playerId = window.localStorage.getItem(PLAYER_ID) ?? createId("ply");
   window.localStorage.setItem(PLAYER_ID, playerId);
 
-  return { nickname, playerId, token: "" } satisfies AuthToken;
+  return { nickname, playerId, token: "", isPro: true } satisfies AuthToken;
 };
 
 export const generateMigrateUrl = (guestPlayerId: string): string => {

client/src/init.tsx

@@ -139,6 +139,7 @@ export const init = async () => {
   }
 
   const info = await auth(nickname);
+
   if (info.guest) {
     const span = document.createElement("span");
     span.textContent = "Guest User ";
@@ -188,7 +189,8 @@ export const init = async () => {
 
       const gameName = (
         <div id="game-info">
-          <code data-instance={game.instanceId}>{game.worldId.split('/').at(1)}</code> {serverButton}
+          <code data-instance={game.instanceId}>{game.worldId.split("/").at(1)}</code>{" "}
+          {serverButton}
         </div>
       );
 

editor/client/main.ts

@@ -44,6 +44,7 @@ import { UndoRedoManager } from "./undo-redo.ts";
 
 const nickname = "Player" + Math.floor(Math.random() * 999) + 1;
 const info = await auth(nickname);
+const isPro = info.isPro ?? false;
 
 const urlParams = new URLSearchParams(window.location.search);
 const isPopout = urlParams.get("popout") === "true";
@@ -369,7 +370,7 @@ if (editModeFlag) {
   const appMenu = new AppMenu(uiRoot, games);
   appMenu.setup(inspector);
 
-  const bottomTabs = new BottomTabs(games);
+  const bottomTabs = new BottomTabs(games, isPro);
   bottomTabs.setup(inspector);
   bottomTabs.show(uiRoot);
 }

editor/client/ui/assistant/assistant.tsx

@@ -7,10 +7,12 @@ export class Assistant {
   game: ClientGame;
   container: HTMLElement;
   ui: InspectorUI | undefined;
+  isPro: boolean;
 
-  constructor(game: ClientGame, container: HTMLElement) {
+  constructor(game: ClientGame, container: HTMLElement, isPro: boolean) {
     this.game = game;
     this.container = container;
+    this.isPro = isPro;
   }
 
   setup(ui: InspectorUI) {
@@ -28,7 +30,7 @@ export class Assistant {
 
       const iframeUrl = `${chatbotUIUrl}/?directory=${decodeURIComponent(
         serviceId,
-      )}&baseUrl=${coderBaseUrl}`;
+      )}&baseUrl=${coderBaseUrl}&isPro=${this.isPro}`;
 
       // 5. Create an iframe to show that coder instance
       const iframe = document.createElement("iframe");

editor/client/ui/bottom-tabs.tsx

@@ -21,7 +21,7 @@ export class BottomTabs implements InspectorUIWidget {
   #assistantContent: HTMLElement;
   #tilemapContent: HTMLElement;
 
-  constructor(games: { edit: ClientGame; play?: ClientGame }) {
+  constructor(games: { edit: ClientGame; play?: ClientGame }, isPro: boolean) {
     this.#container = elem("div", { className: "bottom-tabs" });
 
     this.#logContent = elem("div", { id: "log-viewer-content" });
@@ -31,7 +31,7 @@ export class BottomTabs implements InspectorUIWidget {
 
     this.#logViewer = new LogViewer(this.#logContent, games);
     this.#prefabViewer = new PrefabViewer(games.edit, this.#prefabContent);
-    this.#assistant = new Assistant(games.edit, this.#assistantContent);
+    this.#assistant = new Assistant(games.edit, this.#assistantContent, isPro);
     this.#tilemapViewer = new TileMapViewer(games.edit, this.#tilemapContent);
   }