Union of schema with relations

import { array, nativeEnum, string, union } from "zod";
import { createInsertSchema, createSelectSchema } from 'drizzle-zod';

export const roles = mysqlTable(
  "role",
  {
    id: bigint("id", { mode: "number" }).primaryKey().autoincrement(),
    slug: varchar("slug", { length: 256 }).notNull().notNull(),
    name: varchar("name", { length: 256 }).unique().notNull(),
    updatedAt: timestamp("updatedAt", { mode: "date" }).defaultNow().onUpdateNow().notNull(),
  },
  (role) => ({
    roleNameIndex: index("name_idx").on(role.name),
    roleIdIdx: index("roleId_idx").on(role.id),
    roleSlug: index("roleSlug_idx").on(role.slug),
  })
);

export const rolesRelations = relations(roles, ({ one }) => ({
  permissions: one(rolePermissions, { fields: [roles.name], references: [rolePermissions.role] }),
}));

export const roleInsertSchema = createInsertSchema(roles);

export const rolePermissions = mysqlTable(
  "role-permission",
  {
    role: varchar("role", { length: 256 }).notNull(),
    permissions: varchar("permissions", { length: 256 }).$type<Permission[]>().notNull(),
  },
  (rolePermission) => ({
    compositeKey: primaryKey({ columns: [rolePermission.role, rolePermission.permissions] }),
  })
);

export const rolePermissionRelations = relations(rolePermissions, ({ one }) => ({
  role: one(roles, { fields: [rolePermissions.role], references: [roles.name] }),
}));

export const rolePermissionInsertSchema = createInsertSchema(rolePermissions, {
  permissions: array(nativeEnum(Permission)),
});
export const rolePermissionSelectSchema = createSelectSchema(rolePermissions, {
  permissions: array(nativeEnum(Permission)),
});