Fix issues with null token codes, return code with getAgent

src/resolvers/agent.resolver.ts

@@ -33,6 +33,7 @@ export class AgentResolver {
     ) {
         const agent = await this.prismaService.agent.findFirst({
             where: { slug: code },
+            include: { tokenCode: true },
         });
 
         if (!agent) throw new NotFoundException();

src/resolvers/token-code.resolver.ts

-import { UseGuards, NotFoundException, UnauthorizedException, ForbiddenException } from '@nestjs/common';
+import {
+    UseGuards,
+    NotFoundException,
+    UnauthorizedException,
+    ForbiddenException,
+} from '@nestjs/common';
 import {
     Args,
     Context,
@@ -42,11 +47,13 @@ export class TokenCodeResolver {
 
         if (!agent) throw new ForbiddenException();
 
-        let tokenCode = await this.prismaService.tokenCode.findFirst({
-            where: {
-                id: agent.tokenCodeId,
-            },
-        });
+        let tokenCode =
+            agent.tokenCodeId &&
+            (await this.prismaService.tokenCode.findFirst({
+                where: {
+                    id: agent.tokenCodeId,
+                },
+            }));
 
         if (!tokenCode) {
             tokenCode = await this.prismaService.tokenCode.create({