chore: add pricing plan lock for machine types

This commit is contained in:
bennykok
2024-01-28 14:36:36 +08:00
parent 42aaf1acb9
commit 9de266fbab
8 changed files with 177 additions and 14 deletions
+11 -7
View File
@@ -183,6 +183,7 @@ export const columns: ColumnDef<Machine>[] = [
cell: ({ row }) => {
const machine = row.original;
const [open, setOpen] = useState(false);
const sub = useCurrentPlan();
return (
<DropdownMenu>
@@ -291,7 +292,10 @@ export const columns: ColumnDef<Machine>[] = [
fieldType: "models",
},
gpu: {
inputProps: {},
fieldType: "gpuPicker",
inputProps: {
sub: sub,
},
},
}}
/>
@@ -319,14 +323,14 @@ export const columns: ColumnDef<Machine>[] = [
},
];
import { useCurrentPlan } from "./useCurrentPlan";
export function MachineList({
data,
userMetadata,
sub,
}: {
data: Machine[];
userMetadata: z.infer<typeof AccessType>;
sub: Awaited<ReturnType<typeof getCurrentPlanWithAuth>>;
}) {
const [sorting, setSorting] = React.useState<SortingState>([]);
const [columnFilters, setColumnFilters] = React.useState<ColumnFiltersState>(
@@ -336,6 +340,8 @@ export function MachineList({
React.useState<VisibilityState>({});
const [rowSelection, setRowSelection] = React.useState({});
const sub = useCurrentPlan();
const table = useReactTable({
data,
columns,
@@ -420,11 +426,9 @@ export function MachineList({
},
},
gpu: {
fieldType: !userMetadata.betaFeaturesAccess
? "fallback"
: "select",
fieldType: "gpuPicker",
inputProps: {
disabled: !userMetadata.betaFeaturesAccess,
sub: sub,
},
},
}}