feat: the rest of the owl

packages/trpc/server/auth-router/router.ts

@@ -1,6 +1,8 @@
 import { TRPCError } from '@trpc/server';
 import { env } from 'next-runtime-env';
 
+import { IS_BILLING_ENABLED } from '@documenso/lib/constants/app';
+import { AppError, AppErrorCode } from '@documenso/lib/errors/app-error';
 import { ErrorCode } from '@documenso/lib/next-auth/error-codes';
 import { compareSync } from '@documenso/lib/server-only/auth/hash';
 import { createUser } from '@documenso/lib/server-only/user/create-user';
@@ -21,14 +23,29 @@ export const authRouter = router({
         });
       }
 
-      const { name, email, password, signature } = input;
+      const { name, email, password, signature, url } = input;
 
-      const user = await createUser({ name, email, password, signature });
+      if ((true || IS_BILLING_ENABLED()) && url && url.length <= 6) {
+        throw new AppError(
+          AppErrorCode.PREMIUM_PROFILE_URL,
+          'Only subscribers can have a username shorter than 6 characters',
+        );
+      }
+
+      const user = await createUser({ name, email, password, signature, url });
 
       await sendConfirmationToken({ email: user.email });
 
       return user;
     } catch (err) {
+      console.log(err);
+
+      const error = AppError.parseError(err);
+
+      if (error.code !== AppErrorCode.UNKNOWN_ERROR) {
+        throw AppError.parseErrorToTRPCError(error);
+      }
+
       let message =
         'We were unable to create your account. Please review the information you provided and try again.';
 

packages/trpc/server/auth-router/schema.ts

@@ -21,6 +21,7 @@ export const ZSignUpMutationSchema = z.object({
   email: z.string().email(),
   password: ZPasswordSchema,
   signature: z.string().min(1, { message: 'A signature is required.' }),
+  url: z.string().optional(),
 });
 
 export type TSignUpMutationSchema = z.infer<typeof ZSignUpMutationSchema>;

packages/trpc/server/profile-router/router.ts

@@ -1,6 +1,8 @@
 import { TRPCError } from '@trpc/server';
 
+import { IS_BILLING_ENABLED } from '@documenso/lib/constants/app';
 import { AppError, AppErrorCode } from '@documenso/lib/errors/app-error';
+import { getSubscriptionsByUserId } from '@documenso/lib/server-only/subscription/get-subscriptions-by-user-id';
 import { deleteUser } from '@documenso/lib/server-only/user/delete-user';
 import { findUserSecurityAuditLogs } from '@documenso/lib/server-only/user/find-user-security-audit-logs';
 import { forgotPassword } from '@documenso/lib/server-only/user/forgot-password';
@@ -11,6 +13,7 @@ import { updatePassword } from '@documenso/lib/server-only/user/update-password'
 import { updateProfile } from '@documenso/lib/server-only/user/update-profile';
 import { updatePublicProfile } from '@documenso/lib/server-only/user/update-public-profile';
 import { extractNextApiRequestMetadata } from '@documenso/lib/universal/extract-request-metadata';
+import { SubscriptionStatus } from '@documenso/prisma/client';
 
 import { adminProcedure, authenticatedProcedure, procedure, router } from '../trpc';
 import {
@@ -83,6 +86,21 @@ export const profileRouter = router({
       try {
         const { url } = input;
 
+        if (IS_BILLING_ENABLED() && url.length <= 6) {
+          const subscriptions = await getSubscriptionsByUserId({
+            userId: ctx.user.id,
+          }).then((subscriptions) =>
+            subscriptions.filter((s) => s.status === SubscriptionStatus.ACTIVE),
+          );
+
+          if (subscriptions.length === 0) {
+            throw new AppError(
+              AppErrorCode.PREMIUM_PROFILE_URL,
+              'Only subscribers can have a username shorter than 6 characters',
+            );
+          }
+        }
+
         const user = await updatePublicProfile({
           userId: ctx.user.id,
           url,

packages/trpc/server/profile-router/schema.ts

@@ -17,7 +17,14 @@ export const ZUpdateProfileMutationSchema = z.object({
 });
 
 export const ZUpdatePublicProfileMutationSchema = z.object({
-  url: z.string().min(1),
+  url: z
+    .string()
+    .trim()
+    .toLowerCase()
+    .min(1, { message: 'Please enter a valid username.' })
+    .regex(/^[a-z0-9-]+$/, {
+      message: 'Username can only container alphanumeric characters and dashes.',
+    }),
 });
 
 export const ZUpdatePasswordMutationSchema = z.object({