Daniel Gak Anagrov
77 lines
2.4 KiB
TypeScript
77 lines
2.4 KiB
TypeScript
import { CommandModule } from 'yargs'
|
|
import { cliLogger as logger } from '@/observability/logging'
|
|
import {
|
|
getWorkspaceBySlugOrIdFactory,
|
|
getWorkspaceFactory
|
|
} from '@/modules/workspaces/repositories/workspaces'
|
|
import { db } from '@/db/knex'
|
|
import {
|
|
getWorkspacePlanFactory,
|
|
getWorkspaceSubscriptionFactory,
|
|
upsertWorkspacePlanFactory
|
|
} from '@/modules/gatekeeper/repositories/billing'
|
|
import { WorkspaceNotFoundError } from '@/modules/workspaces/errors/workspace'
|
|
import { WorkspacePlans, WorkspacePlanStatuses } from '@speckle/shared'
|
|
import { getEventBus } from '@/modules/shared/services/eventBus'
|
|
import { updateWorkspacePlanFactory } from '@/modules/gatekeeper/services/workspacePlans'
|
|
|
|
const command: CommandModule<
|
|
unknown,
|
|
{
|
|
workspaceSlugOrId: string
|
|
// you need to know what you are doing, status and plan pairing validity is not ensured here
|
|
status: WorkspacePlanStatuses
|
|
plan: WorkspacePlans
|
|
}
|
|
> = {
|
|
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: WorkspacePlans.Team,
|
|
choices: Object.values(WorkspacePlans)
|
|
},
|
|
status: {
|
|
describe: 'Status to set for the workspace plan',
|
|
type: 'string',
|
|
default: 'valid',
|
|
choices: ['valid', 'paymentFailed', 'cancelationScheduled', 'canceled']
|
|
}
|
|
},
|
|
handler: async (args) => {
|
|
logger.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`
|
|
)
|
|
}
|
|
|
|
const updateWorkspacePlan = updateWorkspacePlanFactory({
|
|
getWorkspace: getWorkspaceFactory({ db }),
|
|
upsertWorkspacePlan: upsertWorkspacePlanFactory({ db }),
|
|
getWorkspacePlan: getWorkspacePlanFactory({ db }),
|
|
getWorkspaceSubscription: getWorkspaceSubscriptionFactory({ db }),
|
|
emitEvent: getEventBus().emit
|
|
})
|
|
await updateWorkspacePlan({
|
|
userId: null,
|
|
workspaceId: workspace.id,
|
|
name: args.plan,
|
|
status: args.status
|
|
})
|
|
|
|
logger.info(`Plan set!`)
|
|
}
|
|
}
|
|
|
|
export = command
|