Kristaps Fabians Geikins
68 lines
1.9 KiB
TypeScript
68 lines
1.9 KiB
TypeScript
import { CommandModule } from 'yargs'
|
|
import { cliLogger } from '@/logging/logging'
|
|
import { getWorkspaceBySlugOrIdFactory } from '@/modules/workspaces/repositories/workspaces'
|
|
import { db } from '@/db/knex'
|
|
import { upsertPaidWorkspacePlanFactory } from '@/modules/gatekeeper/repositories/billing'
|
|
import { WorkspaceNotFoundError } from '@/modules/workspaces/errors/workspace'
|
|
import { PaidWorkspacePlans, PaidWorkspacePlanStatuses } from '@speckle/shared'
|
|
|
|
const command: CommandModule<
|
|
unknown,
|
|
{
|
|
workspaceSlugOrId: string
|
|
status: PaidWorkspacePlanStatuses
|
|
plan: PaidWorkspacePlans
|
|
}
|
|
> = {
|
|
command: 'set-plan <workspaceSlugOrId> [plan] [status]',
|
|
describe: 'Set a plan for a workspace.',
|
|
builder: {
|
|
workspaceSlugOrId: {
|
|
describe: 'Workspace ID or slug',
|
|
type: 'string'
|
|
},
|
|
plan: {
|
|
describe: 'Plan to set the status for',
|
|
type: 'string',
|
|
default: 'business',
|
|
choices: ['business', 'starter', 'plus']
|
|
},
|
|
status: {
|
|
describe: 'Status to set for the workspace plan',
|
|
type: 'string',
|
|
default: 'valid',
|
|
choices: [
|
|
'valid',
|
|
'trial',
|
|
'expired',
|
|
'paymentFailed',
|
|
'cancelationScheduled',
|
|
'canceled'
|
|
]
|
|
}
|
|
},
|
|
handler: async (args) => {
|
|
cliLogger.info(
|
|
`Setting plan for workspace '${args.workspaceSlugOrId}' to '${args.plan}' with status '${args.status}'`
|
|
)
|
|
const workspace = await getWorkspaceBySlugOrIdFactory({ db })(args)
|
|
if (!workspace) {
|
|
throw new WorkspaceNotFoundError(
|
|
`Workspace w/ slug or id '${args.workspaceSlugOrId}' not found`
|
|
)
|
|
}
|
|
|
|
await upsertPaidWorkspacePlanFactory({ db })({
|
|
workspacePlan: {
|
|
createdAt: new Date(),
|
|
workspaceId: workspace.id,
|
|
name: args.plan,
|
|
status: args.status
|
|
}
|
|
})
|
|
cliLogger.info(`Plan set!`)
|
|
}
|
|
}
|
|
|
|
export = command
|