33 lines
878 B
TypeScript
33 lines
878 B
TypeScript
import { Prisma } from "@prisma/client";
|
|
|
|
import LicenseKeyService, { UsageEvent } from "@calcom/ee/common/server/LicenseKeyService";
|
|
|
|
export function usageTrackingExtention() {
|
|
return Prisma.defineExtension({
|
|
query: {
|
|
booking: {
|
|
async create({ args, query }) {
|
|
try {
|
|
const licenseKeyService = await LicenseKeyService.create();
|
|
await licenseKeyService.incrementUsage();
|
|
} catch (e) {
|
|
console.log(e);
|
|
}
|
|
return query(args);
|
|
},
|
|
},
|
|
user: {
|
|
async create({ args, query }) {
|
|
try {
|
|
const licenseKeyService = await LicenseKeyService.create();
|
|
await licenseKeyService.incrementUsage(UsageEvent.USER);
|
|
} catch (e) {
|
|
console.log(e);
|
|
}
|
|
return query(args);
|
|
},
|
|
},
|
|
},
|
|
});
|
|
}
|