apunkt/rowboat
1
0
Fork 0
mirror of https://github.com/rowboatlabs/rowboat.git synced 2026-06-03 19:25:19 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

check authz before consuming project action quota

Browse source
This commit is contained in:
Ramnique Singh 2025-08-05 17:05:04 +05:30
parent cd6ff9a46f
commit 7592df8068
4 changed files with 12 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

6
apps/rowboat/src/application/use-cases/conversations/create-cached-turn.use-case.ts
View file

@ -52,9 +52,6 @@ export class CreateCachedTurnUseCase implements ICreateCachedTurnUseCase {
// extract projectid from conversation // extract projectid from conversation
const { projectId } = conversation; const { projectId } = conversation;
// assert and consume quota
await this.usageQuotaPolicy.assertAndConsume(projectId);
// authz check // authz check
await this.projectActionAuthorizationPolicy.authorize({ await this.projectActionAuthorizationPolicy.authorize({
caller: data.caller, caller: data.caller,
@ -63,6 +60,9 @@ export class CreateCachedTurnUseCase implements ICreateCachedTurnUseCase {
projectId, projectId,
}); });
// assert and consume quota
await this.usageQuotaPolicy.assertAndConsume(projectId);
// create cache entry // create cache entry
const key = nanoid(); const key = nanoid();
const payload: z.infer<typeof CachedTurnRequest> = { const payload: z.infer<typeof CachedTurnRequest> = {

6
apps/rowboat/src/application/use-cases/conversations/create-conversation.use-case.ts
View file

@ -44,9 +44,6 @@ export class CreateConversationUseCase implements ICreateConversationUseCase {
let isLiveWorkflow = Boolean(data.isLiveWorkflow); let isLiveWorkflow = Boolean(data.isLiveWorkflow);
let workflow = data.workflow; let workflow = data.workflow;
// assert and consume quota
await this.usageQuotaPolicy.assertAndConsume(projectId);
// authz check // authz check
await this.projectActionAuthorizationPolicy.authorize({ await this.projectActionAuthorizationPolicy.authorize({
caller, caller,
@ -55,6 +52,9 @@ export class CreateConversationUseCase implements ICreateConversationUseCase {
projectId, projectId,
}); });
// assert and consume quota
await this.usageQuotaPolicy.assertAndConsume(projectId);
// if workflow is not provided, fetch workflow // if workflow is not provided, fetch workflow
if (!workflow) { if (!workflow) {
const project = await projectsCollection.findOne({ const project = await projectsCollection.findOne({

6
apps/rowboat/src/application/use-cases/conversations/fetch-cached-turn.use-case.ts
View file

@ -59,9 +59,6 @@ export class FetchCachedTurnUseCase implements IFetchCachedTurnUseCase {
// extract projectid from conversation // extract projectid from conversation
const { projectId } = conversation; const { projectId } = conversation;
// assert and consume quota
await this.usageQuotaPolicy.assertAndConsume(projectId);
// authz check // authz check
await this.projectActionAuthorizationPolicy.authorize({ await this.projectActionAuthorizationPolicy.authorize({
caller: data.caller, caller: data.caller,
@ -70,6 +67,9 @@ export class FetchCachedTurnUseCase implements IFetchCachedTurnUseCase {
projectId, projectId,
}); });
// assert and consume quota
await this.usageQuotaPolicy.assertAndConsume(projectId);
// delete from cache // delete from cache
await this.cacheService.delete(`turn-${data.key}`); await this.cacheService.delete(`turn-${data.key}`);

6
apps/rowboat/src/application/use-cases/conversations/run-conversation-turn.use-case.ts
View file

@ -51,9 +51,6 @@ export class RunConversationTurnUseCase implements IRunConversationTurnUseCase {
// extract projectid from conversation // extract projectid from conversation
const { id: conversationId, projectId } = conversation; const { id: conversationId, projectId } = conversation;
// assert and consume quota
await this.usageQuotaPolicy.assertAndConsume(projectId);
// authz check // authz check
await this.projectActionAuthorizationPolicy.authorize({ await this.projectActionAuthorizationPolicy.authorize({
caller: data.caller, caller: data.caller,
@ -62,6 +59,9 @@ export class RunConversationTurnUseCase implements IRunConversationTurnUseCase {
projectId, projectId,
}); });
// assert and consume quota
await this.usageQuotaPolicy.assertAndConsume(projectId);
// Check billing auth // Check billing auth
if (USE_BILLING) { if (USE_BILLING) {
// get billing customer id for project // get billing customer id for project