Using Database Schema Types

Author: ThomasCode92

src/app/drive-contents.tsx

@@ -4,14 +4,14 @@ import { ChevronRight, Upload } from "lucide-react";
 import Link from "next/link";
 
 import { Button } from "~/components/ui/button";
-import type { files_table, folders_table } from "~/server/db/schema";
+import type { File, Folder } from "~/server/db/schema";
 
 import { FileRow, FolderRow } from "./file-row";
 
 export default function DriveContents(props: {
-  files: (typeof files_table.$inferSelect)[];
-  folders: (typeof folders_table.$inferSelect)[];
-  parents: (typeof folders_table.$inferSelect)[];
+  files: File[];
+  folders: Folder[];
+  parents: Folder[];
 }) {
   const handleUpload = () => {
     alert("Upload functionality would be implemented here");

src/app/file-row.tsx

@@ -1,9 +1,9 @@
 import { FileIcon, Folder as FolderIcon } from "lucide-react";
 import Link from "next/link";
 
-import type { files_table, folders_table } from "~/server/db/schema";
+import type { File, Folder } from "~/server/db/schema";
 
-export function FileRow(props: { file: typeof files_table.$inferSelect }) {
+export function FileRow(props: { file: File }) {
   const { file } = props;
 
   return (
@@ -29,9 +29,7 @@ export function FileRow(props: { file: typeof files_table.$inferSelect }) {
   );
 }
 
-export function FolderRow(props: {
-  folder: typeof folders_table.$inferSelect;
-}) {
+export function FolderRow(props: { folder: Folder }) {
   const { folder } = props;
 
   return (

src/server/db/schema.ts

@@ -33,3 +33,6 @@ export const folders_table = createTable(
   },
   (table) => [index("parent_index").on(table.parent)],
 );
+
+export type File = typeof files_table.$inferSelect;
+export type Folder = typeof folders_table.$inferSelect;