@adobe/spacecat-shared-data-access 2.57.0 → 2.58.0

package/CHANGELOG.md

@@ -1,3 +1,17 @@
+# [@adobe/spacecat-shared-data-access-v2.58.0](https://github.com/adobe/spacecat-shared/compare/@adobe/spacecat-shared-data-access-v2.57.1...@adobe/spacecat-shared-data-access-v2.58.0) (2025-08-28)
+
+
+### Features
+
+* **dataAccess:** Add entities for LLMO trial onboarding ([#928](https://github.com/adobe/spacecat-shared/issues/928)) ([b5acc3c](https://github.com/adobe/spacecat-shared/commit/b5acc3c3af49d5fd45b8abf91b851fe23da60bc7))
+
+# [@adobe/spacecat-shared-data-access-v2.57.1](https://github.com/adobe/spacecat-shared/compare/@adobe/spacecat-shared-data-access-v2.57.0...@adobe/spacecat-shared-data-access-v2.57.1) (2025-08-26)
+
+
+### Bug Fixes
+
+* use DELIVERY_TYPES from shared-utils ([#933](https://github.com/adobe/spacecat-shared/issues/933)) ([3592540](https://github.com/adobe/spacecat-shared/commit/3592540fc3cee05333fbf854447755b727dafae1))
+
 # [@adobe/spacecat-shared-data-access-v2.57.0](https://github.com/adobe/spacecat-shared/compare/@adobe/spacecat-shared-data-access-v2.56.1...@adobe/spacecat-shared-data-access-v2.57.0) (2025-08-22)
 
 

package/README.md

@@ -42,6 +42,49 @@ npm install @adobe/spacecat-shared-data-access
 - **geo** (String): Geo of the top page.
 - **importedAt** (String): Timestamp of the import.
 
+### Organization
+- **id** (String): Unique identifier for an organization.
+- **createdAt** (String): Timestamp of creation.
+- **updatedAt** (String): Timestamp of the last update.
+
+### OrganizationIdentityProvider
+- **id** (String): Unique identifier for the identity provider.
+- **metadata** (Map): Metadata for the identity provider.
+- **provider** (String): Type of identity provider. (IMS, MICROSOFT, GOOGLE)
+- **externalId** (String): External identifier from the provider.
+- **createdAt** (String): Timestamp of creation.
+
+### TrialUser
+- **id** (String): Unique identifier for the trial user.
+- **externalUserId** (String): External user identifier.
+- **status** (String): Status of the trial user. (REGISTERED, VERIFIED, BLOCKED, DELETED)
+- **provider** (String): Type of identity provider. (IMS, MICROSOFT, GOOGLE)
+- **lastSeenAt** (String): Timestamp of last activity.
+- **createdAt** (String): Timestamp of creation.
+- **metadata** (Map): Metadata for the trial user.
+- **updatedAt** (String): Timestamp of the last update.
+
+### TrialUserActivity
+- **id** (String): Unique identifier for the trial user activity.
+- **type** (String): Type of activity performed. (SIGN_UP, SIGN_IN, CREATE_SITE, RUN_AUDIT, PROMPT_RUN, DOWNLOAD)
+- **details** (Map): Details of the activity.
+- **createdAt** (String): Timestamp of creation.
+- **productCode** (String): Product code associated with the activity. (LLMO, ASO, etc.)
+
+### Entitlement
+- **id** (String): Unique identifier for the entitlement.
+- **productCode** (String): Product code for the entitlement. (LLMO, ASO, etc.)
+- **tier** (String): Tier level of the entitlement. (FREE_TRIAL, PAID)
+- **status** (String): Status of the entitlement. (ACTIVE, SUSPENDED, ENDED)
+- **createdAt** (String): Timestamp of creation.
+- **updatedAt** (String): Timestamp of the last update.
+- **quotas** (Map): Quota information for the entitlement.
+
+### SiteEnrollment
+- **id** (String): Unique identifier for the site enrollment.
+- **status** (String): Status of the enrollment. (ACTIVE, SUSPENDED, ENDED)
+- **createdAt** (String): Timestamp of creation.
+
 ## DynamoDB Data Model
 
 The module is designed to work with the following DynamoDB tables:

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@adobe/spacecat-shared-data-access",
-  "version": "2.57.0",
+  "version": "2.58.0",
   "description": "Shared modules of the Spacecat Services - Data Access",
   "type": "module",
   "engines": {
@@ -35,7 +35,7 @@
     "access": "public"
   },
   "dependencies": {
-    "@adobe/spacecat-shared-utils": "1.39.1",
+    "@adobe/spacecat-shared-utils": "1.49.0",
     "@aws-sdk/client-dynamodb": "3.864.0",
     "@aws-sdk/lib-dynamodb": "3.864.0",
     "@types/joi": "17.2.3",

package/src/models/base/entity.registry.js

@@ -18,6 +18,7 @@ import AsyncJobCollection from '../async-job/async-job.collection.js';
 import AuditCollection from '../audit/audit.collection.js';
 import ConfigurationCollection from '../configuration/configuration.collection.js';
 import ExperimentCollection from '../experiment/experiment.collection.js';
+import EntitlementCollection from '../entitlement/entitlement.collection.js';
 import FixEntityCollection from '../fix-entity/fix-entity.collection.js';
 import ImportJobCollection from '../import-job/import-job.collection.js';
 import ImportUrlCollection from '../import-url/import-url.collection.js';
@@ -25,20 +26,25 @@ import KeyEventCollection from '../key-event/key-event.collection.js';
 import LatestAuditCollection from '../latest-audit/latest-audit.collection.js';
 import OpportunityCollection from '../opportunity/opportunity.collection.js';
 import OrganizationCollection from '../organization/organization.collection.js';
+import OrganizationIdentityProviderCollection from '../organization-identity-provider/organization-identity-provider.collection.js';
 import ScrapeJobCollection from '../scrape-job/scrape-job.collection.js';
 import ScrapeUrlCollection from '../scrape-url/scrape-url.collection.js';
 import SiteCandidateCollection from '../site-candidate/site-candidate.collection.js';
 import SiteCollection from '../site/site.collection.js';
+import SiteEnrollmentCollection from '../site-enrollment/site-enrollment.collection.js';
 import SiteTopFormCollection from '../site-top-form/site-top-form.collection.js';
 import SiteTopPageCollection from '../site-top-page/site-top-page.collection.js';
 import SuggestionCollection from '../suggestion/suggestion.collection.js';
 import PageIntentCollection from '../page-intent/page-intent.collection.js';
 import ReportCollection from '../report/report.collection.js';
+import TrialUserCollection from '../trial-user/trial-user.collection.js';
+import TrialUserActivityCollection from '../trial-user-activity/trial-user-activity.collection.js';
 
 import ApiKeySchema from '../api-key/api-key.schema.js';
 import AsyncJobSchema from '../async-job/async-job.schema.js';
 import AuditSchema from '../audit/audit.schema.js';
 import ConfigurationSchema from '../configuration/configuration.schema.js';
+import EntitlementSchema from '../entitlement/entitlement.schema.js';
 import FixEntitySchema from '../fix-entity/fix-entity.schema.js';
 import ExperimentSchema from '../experiment/experiment.schema.js';
 import ImportJobSchema from '../import-job/import-job.schema.js';
@@ -47,15 +53,19 @@ import KeyEventSchema from '../key-event/key-event.schema.js';
 import LatestAuditSchema from '../latest-audit/latest-audit.schema.js';
 import OpportunitySchema from '../opportunity/opportunity.schema.js';
 import OrganizationSchema from '../organization/organization.schema.js';
+import OrganizationIdentityProviderSchema from '../organization-identity-provider/organization-identity-provider.schema.js';
 import ScrapeJobSchema from '../scrape-job/scrape-job.schema.js';
 import ScrapeUrlSchema from '../scrape-url/scrape-url.schema.js';
 import SiteSchema from '../site/site.schema.js';
 import SiteCandidateSchema from '../site-candidate/site-candidate.schema.js';
+import SiteEnrollmentSchema from '../site-enrollment/site-enrollment.schema.js';
 import SiteTopFormSchema from '../site-top-form/site-top-form.schema.js';
 import SiteTopPageSchema from '../site-top-page/site-top-page.schema.js';
 import SuggestionSchema from '../suggestion/suggestion.schema.js';
 import PageIntentSchema from '../page-intent/page-intent.schema.js';
 import ReportSchema from '../report/report.schema.js';
+import TrialUserSchema from '../trial-user/trial-user.schema.js';
+import TrialUserActivitySchema from '../trial-user-activity/trial-user-activity.schema.js';
 
 /**
  * EntityRegistry - A registry class responsible for managing entities, their schema and collection.
@@ -142,6 +152,7 @@ EntityRegistry.registerEntity(ApiKeySchema, ApiKeyCollection);
 EntityRegistry.registerEntity(AsyncJobSchema, AsyncJobCollection);
 EntityRegistry.registerEntity(AuditSchema, AuditCollection);
 EntityRegistry.registerEntity(ConfigurationSchema, ConfigurationCollection);
+EntityRegistry.registerEntity(EntitlementSchema, EntitlementCollection);
 EntityRegistry.registerEntity(FixEntitySchema, FixEntityCollection);
 EntityRegistry.registerEntity(ExperimentSchema, ExperimentCollection);
 EntityRegistry.registerEntity(ImportJobSchema, ImportJobCollection);
@@ -150,14 +161,21 @@ EntityRegistry.registerEntity(KeyEventSchema, KeyEventCollection);
 EntityRegistry.registerEntity(LatestAuditSchema, LatestAuditCollection);
 EntityRegistry.registerEntity(OpportunitySchema, OpportunityCollection);
 EntityRegistry.registerEntity(OrganizationSchema, OrganizationCollection);
+EntityRegistry.registerEntity(
+  OrganizationIdentityProviderSchema,
+  OrganizationIdentityProviderCollection,
+);
 EntityRegistry.registerEntity(ScrapeJobSchema, ScrapeJobCollection);
 EntityRegistry.registerEntity(ScrapeUrlSchema, ScrapeUrlCollection);
 EntityRegistry.registerEntity(SiteSchema, SiteCollection);
 EntityRegistry.registerEntity(SiteCandidateSchema, SiteCandidateCollection);
+EntityRegistry.registerEntity(SiteEnrollmentSchema, SiteEnrollmentCollection);
 EntityRegistry.registerEntity(SiteTopFormSchema, SiteTopFormCollection);
 EntityRegistry.registerEntity(SiteTopPageSchema, SiteTopPageCollection);
 EntityRegistry.registerEntity(SuggestionSchema, SuggestionCollection);
 EntityRegistry.registerEntity(PageIntentSchema, PageIntentCollection);
 EntityRegistry.registerEntity(ReportSchema, ReportCollection);
+EntityRegistry.registerEntity(TrialUserSchema, TrialUserCollection);
+EntityRegistry.registerEntity(TrialUserActivitySchema, TrialUserActivityCollection);
 
 export default EntityRegistry;

package/src/models/entitlement/entitlement.collection.js

@@ -0,0 +1,26 @@
+/*
+ * Copyright 2024 Adobe. All rights reserved.
+ * This file is licensed to you under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License. You may obtain a copy
+ * of the License at http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under
+ * the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR REPRESENTATIONS
+ * OF ANY KIND, either express or implied. See the License for the specific language
+ * governing permissions and limitations under the License.
+ */
+
+import BaseCollection from '../base/base.collection.js';
+
+/**
+ * EntitlementCollection - A collection class responsible for managing Entitlement entities.
+ * Extends the BaseCollection to provide specific methods for interacting with Entitlement records.
+ *
+ * @class EntitlementCollection
+ * @extends BaseCollection
+ */
+class EntitlementCollection extends BaseCollection {
+  // add custom methods here
+}
+
+export default EntitlementCollection;

package/src/models/entitlement/entitlement.model.js

@@ -0,0 +1,36 @@
+/*
+ * Copyright 2024 Adobe. All rights reserved.
+ * This file is licensed to you under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License. You may obtain a copy
+ * of the License at http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under
+ * the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR REPRESENTATIONS
+ * OF ANY KIND, either express or implied. See the License for the specific language
+ * governing permissions and limitations under the License.
+ */
+
+import BaseModel from '../base/base.model.js';
+
+/**
+ * Entitlement - A class representing an Entitlement entity.
+ * Provides methods to access and manipulate Entitlement-specific data.
+ *
+ * @class Entitlement
+ * @extends BaseModel
+ */
+class Entitlement extends BaseModel {
+  static TIERS = {
+    FREE_TRIAL: 'FREE_TRIAL',
+    PAID: 'PAID',
+  };
+
+  static PRODUCT_CODES = {
+    LLMO: 'LLMO',
+    ASO: 'ASO',
+  };
+
+  // add your custom methods or overrides here
+}
+
+export default Entitlement;

package/src/models/entitlement/entitlement.schema.js

@@ -0,0 +1,49 @@
+/*
+ * Copyright 2025 Adobe. All rights reserved.
+ * This file is licensed to you under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License. You may obtain a copy
+ * of the License at http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under
+ * the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR REPRESENTATIONS
+ * OF ANY KIND, either express or implied. See the License for the specific language
+ * governing permissions and limitations under the License.
+ */
+
+import SchemaBuilder from '../base/schema.builder.js';
+import Entitlement from './entitlement.model.js';
+import EntitlementCollection from './entitlement.collection.js';
+
+/*
+Schema Doc: https://electrodb.dev/en/modeling/schema/
+Attribute Doc: https://electrodb.dev/en/modeling/attributes/
+Indexes Doc: https://electrodb.dev/en/modeling/indexes/
+ */
+
+const schema = new SchemaBuilder(Entitlement, EntitlementCollection)
+  // Reference to Organization (many-to-one relationship)
+  .addReference('belongs_to', 'Organization')
+  // Reference to SiteEnrollments (one-to-many relationship)
+  .addReference('has_many', 'SiteEnrollments')
+  .addReference('has_many', 'TrialUserActivities')
+  .addAttribute('productCode', {
+    type: Object.values(Entitlement.PRODUCT_CODES),
+    required: true,
+  })
+  .addAttribute('tier', {
+    type: Object.values(Entitlement.TIERS),
+    required: true,
+  })
+  .addAttribute('quotas', {
+    type: 'map',
+    required: false,
+    properties: {
+      llmo_trial_prompts: { type: 'number' },
+    },
+  })
+  .addIndex(
+    { composite: ['organizationId'] },
+    { composite: ['productCode'] },
+  );
+
+export default schema.build();

package/src/models/entitlement/index.d.ts

@@ -0,0 +1,47 @@
+/*
+ * Copyright 2025 Adobe. All rights reserved.
+ * This file is licensed to you under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License. You may obtain a copy
+ * of the License at http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under
+ * the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR REPRESENTATIONS
+ * OF ANY KIND, either express or implied. See the License for the specific language
+ * governing permissions and limitations under the License.
+ */
+
+import type {
+  BaseCollection, BaseModel, Organization, SiteEnrollment,
+} from '../index';
+
+export type EntitlementStatus = 'ACTIVE' | 'SUSPENDED' | 'ENDED';
+export type EntitlementTier = 'FREE_TRIAL' | 'PAID';
+export type EntitlementProductCode = 'LLMO' | 'ASO';
+
+export interface Entitlement extends BaseModel {
+  getProductCode(): EntitlementProductCode;
+  getTier(): EntitlementTier;
+  getStatus(): EntitlementStatus;
+  getQuotas(): object | null;
+  getOrganization(): Promise<Organization>;
+  getSiteEnrollments(): Promise<SiteEnrollment[]>;
+  setProductCode(productCode: EntitlementProductCode): Entitlement;
+  setTier(tier: EntitlementTier): Entitlement;
+  setStatus(status: EntitlementStatus): Entitlement;
+  setQuotas(quotas: object): Entitlement;
+}
+
+export interface EntitlementCollection extends
+    BaseCollection<Entitlement> {
+  allByOrganizationId(organizationId: string): Promise<Entitlement[]>;
+  allByOrganizationIdAndProductCode(
+    organizationId: string,
+    productCode: EntitlementProductCode,
+  ): Promise<Entitlement[]>;
+
+  findByOrganizationId(organizationId: string): Promise<Entitlement[]>;
+  findByOrganizationIdAndProductCode(
+    organizationId: string,
+    productCode: EntitlementProductCode,
+  ): Promise<Entitlement[]>;
+}

package/src/models/entitlement/index.js

@@ -0,0 +1,19 @@
+/*
+ * Copyright 2024 Adobe. All rights reserved.
+ * This file is licensed to you under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License. You may obtain a copy
+ * of the License at http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under
+ * the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR REPRESENTATIONS
+ * OF ANY KIND, either express or implied. See the License for the specific language
+ * governing permissions and limitations under the License.
+ */
+
+import Entitlement from './entitlement.model.js';
+import EntitlementCollection from './entitlement.collection.js';
+
+export {
+  Entitlement,
+  EntitlementCollection,
+};

package/src/models/index.d.ts

@@ -16,17 +16,22 @@ export type * from './configuration';
 export type * from './base';
 export type * from './fix-entity';
 export type * from './experiment';
+export type * from './entitlement';
 export type * from './import-job';
 export type * from './import-url';
 export type * from './key-event';
 export type * from './latest-audit';
 export type * from './opportunity';
 export type * from './organization';
+export type * from './organization-identity-provider';
 export type * from './scrape-job';
 export type * from './scrape-url';
 export type * from './site';
 export type * from './site-candidate';
+export type * from './site-enrollment';
 export type * from './site-top-form';
 export type * from './site-top-page';
 export type * from './suggestion';
 export type * from './report';
+export type * from './trial-user';
+export type * from './trial-user-activity';

package/src/models/index.js

@@ -15,6 +15,7 @@ export * from './async-job/index.js';
 export * from './audit/index.js';
 export * from './base/index.js';
 export * from './configuration/index.js';
+export * from './entitlement/index.js';
 export * from './fix-entity/index.js';
 export * from './experiment/index.js';
 export * from './import-job/index.js';
@@ -23,12 +24,16 @@ export * from './key-event/index.js';
 export * from './latest-audit/index.js';
 export * from './opportunity/index.js';
 export * from './organization/index.js';
+export * from './organization-identity-provider/index.js';
 export * from './scrape-job/index.js';
 export * from './scrape-url/index.js';
 export * from './site-candidate/index.js';
+export * from './site-enrollment/index.js';
 export * from './site-top-form/index.js';
 export * from './site-top-page/index.js';
 export * from './site/index.js';
 export * from './suggestion/index.js';
 export * from './page-intent/index.js';
 export * from './report/index.js';
+export * from './trial-user/index.js';
+export * from './trial-user-activity/index.js';

package/src/models/organization/index.d.ts

@@ -10,7 +10,9 @@
  * governing permissions and limitations under the License.
  */
 
-import type { BaseCollection, BaseModel, Site } from '../index';
+import type {
+  BaseCollection, BaseModel, Site, Entitlement, OrganizationIdentityProvider, TrialUser,
+} from '../index';
 
 export interface Organization extends BaseModel {
   getConfig(): object;
@@ -18,6 +20,9 @@ export interface Organization extends BaseModel {
   getImsOrgId(): string;
   getName(): string;
   getSites(): Promise<Site[]>;
+  getEntitlements(): Promise<Entitlement[]>;
+  getOrganizationIdentityProviders(): Promise<OrganizationIdentityProvider[]>;
+  getTrialUsers(): Promise<TrialUser[]>;
   setConfig(config: object): Organization;
   setFulfillableItems(fulfillableItems: object): Organization;
   setImsOrgId(imsOrgId: string): Organization;

package/src/models/organization/organization.schema.js

@@ -28,6 +28,9 @@ Indexes Doc: https://electrodb.dev/en/modeling/indexes/
 const schema = new SchemaBuilder(Organization, OrganizationCollection)
   // this will add an attribute 'organizationId' as well as an index 'byOrganizationId'
   .addReference('has_many', 'Sites')
+  .addReference('has_many', 'Entitlements')
+  .addReference('has_many', 'OrganizationIdentityProviders')
+  .addReference('has_many', 'TrialUsers')
   .addAttribute('config', {
     type: 'any',
     required: true,

package/src/models/organization-identity-provider/index.d.ts

@@ -0,0 +1,40 @@
+/*
+ * Copyright 2025 Adobe. All rights reserved.
+ * This file is licensed to you under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License. You may obtain a copy
+ * of the License at http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under
+ * the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR REPRESENTATIONS
+ * OF ANY KIND, either express or implied. See the License for the specific language
+ * governing permissions and limitations under the License.
+ */
+
+import type {
+  BaseCollection, BaseModel, Organization, TrialUser,
+} from '../index';
+
+export type ProviderType = 'IMS' | 'MICROSOFT' | 'GOOGLE';
+
+export interface OrganizationIdentityProvider extends BaseModel {
+  getMetadata(): object | null;
+  getProvider(): ProviderType;
+  getExternalId(): string;
+  getOrganization(): Promise<Organization>;
+  getTrialUsers(): Promise<TrialUser[]>;
+  setMetadata(metadata: object): OrganizationIdentityProvider;
+  setProvider(provider: ProviderType): OrganizationIdentityProvider;
+  setExternalId(externalId: string): OrganizationIdentityProvider;
+}
+
+export interface OrganizationIdentityProviderCollection extends
+    BaseCollection<OrganizationIdentityProvider> {
+  allByProvider(provider: ProviderType): Promise<OrganizationIdentityProvider[]>;
+  allByProviderAndExternalId(provider: ProviderType, externalId: string):
+    Promise<OrganizationIdentityProvider[]>;
+  allByOrganizationId(organizationId: string): Promise<OrganizationIdentityProvider[]>;
+  findByProvider(provider: ProviderType): Promise<OrganizationIdentityProvider[]>;
+  findByProviderAndExternalId(provider: ProviderType, externalId: string):
+    Promise<OrganizationIdentityProvider | null>;
+  findByOrganizationId(organizationId: string): Promise<OrganizationIdentityProvider[]>;
+}

package/src/models/organization-identity-provider/index.js

@@ -0,0 +1,19 @@
+/*
+ * Copyright 2025 Adobe. All rights reserved.
+ * This file is licensed to you under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License. You may obtain a copy
+ * of the License at http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under
+ * the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR REPRESENTATIONS
+ * OF ANY KIND, either express or implied. See the License for the specific language
+ * governing permissions and limitations under the License.
+ */
+
+import OrganizationIdentityProvider from './organization-identity-provider.model.js';
+import OrganizationIdentityProviderCollection from './organization-identity-provider.collection.js';
+
+export {
+  OrganizationIdentityProvider,
+  OrganizationIdentityProviderCollection,
+};

package/src/models/organization-identity-provider/organization-identity-provider.collection.js

@@ -0,0 +1,28 @@
+/*
+ * Copyright 2025 Adobe. All rights reserved.
+ * This file is licensed to you under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License. You may obtain a copy
+ * of the License at http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under
+ * the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR REPRESENTATIONS
+ * OF ANY KIND, either express or implied. See the License for the specific language
+ * governing permissions and limitations under the License.
+ */
+
+import BaseCollection from '../base/base.collection.js';
+
+/**
+ * OrganizationIdentityProviderCollection - A collection class responsible for
+ * managing OrganizationIdentityProvider entities.
+ * Extends the BaseCollection to provide specific methods for interacting with
+ * OrganizationIdentityProvider records.
+ *
+ * @class OrganizationIdentityProviderCollection
+ * @extends BaseCollection
+ */
+class OrganizationIdentityProviderCollection extends BaseCollection {
+  // add custom methods here
+}
+
+export default OrganizationIdentityProviderCollection;

package/src/models/organization-identity-provider/organization-identity-provider.model.js

@@ -0,0 +1,32 @@
+/*
+ * Copyright 2025 Adobe. All rights reserved.
+ * This file is licensed to you under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License. You may obtain a copy
+ * of the License at http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under
+ * the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR REPRESENTATIONS
+ * OF ANY KIND, either express or implied. See the License for the specific language
+ * governing permissions and limitations under the License.
+ */
+
+import BaseModel from '../base/base.model.js';
+
+/**
+ * OrganizationIdentityProvider - A class representing an OrganizationIdentityProvider entity.
+ * Provides methods to access and manipulate OrganizationIdentityProvider-specific data.
+ *
+ * @class OrganizationIdentityProvider
+ * @extends BaseModel
+ */
+class OrganizationIdentityProvider extends BaseModel {
+  static PROVIDER_TYPES = {
+    IMS: 'IMS',
+    MICROSOFT: 'MICROSOFT',
+    GOOGLE: 'GOOGLE',
+  };
+
+  // add your custom methods or overrides here
+}
+
+export default OrganizationIdentityProvider;

package/src/models/organization-identity-provider/organization-identity-provider.schema.js

@@ -0,0 +1,48 @@
+/*
+ * Copyright 2025 Adobe. All rights reserved.
+ * This file is licensed to you under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License. You may obtain a copy
+ * of the License at http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under
+ * the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR REPRESENTATIONS
+ * OF ANY KIND, either express or implied. See the License for the specific language
+ * governing permissions and limitations under the License.
+ */
+
+import { isObject } from '@adobe/spacecat-shared-utils';
+import SchemaBuilder from '../base/schema.builder.js';
+import OrganizationIdentityProvider from './organization-identity-provider.model.js';
+import OrganizationIdentityProviderCollection from './organization-identity-provider.collection.js';
+
+/*
+Schema Doc: https://electrodb.dev/en/modeling/schema/
+Attribute Doc: https://electrodb.dev/en/modeling/attributes/
+Indexes Doc: https://electrodb.dev/en/modeling/indexes/
+ */
+
+const schema = new SchemaBuilder(
+  OrganizationIdentityProvider,
+  OrganizationIdentityProviderCollection,
+)
+  // Reference to Organization (many-to-one relationship)
+  .addReference('belongs_to', 'Organization')
+  .addAttribute('metadata', {
+    type: 'any',
+    required: false,
+    validate: (value) => !value || isObject(value),
+  })
+  .addAttribute('provider', {
+    type: Object.values(OrganizationIdentityProvider.PROVIDER_TYPES),
+    required: true,
+  })
+  .addAttribute('externalId', {
+    type: 'string',
+    required: true,
+  })
+  .addIndex(
+    { composite: ['provider'] },
+    { composite: ['externalId'] },
+  );
+
+export default schema.build();

package/src/models/site/index.d.ts

@@ -20,6 +20,7 @@ import type {
   Opportunity,
   Organization,
   SiteCandidate,
+  SiteEnrollment,
   SiteTopPage,
 } from '../index.js';
 
@@ -214,6 +215,7 @@ export interface Site extends BaseModel {
   getOrganization(): Promise<Organization>;
   getOrganizationId(): string;
   getSiteCandidates(): Promise<SiteCandidate[]>;
+  getSiteEnrollments(): Promise<SiteEnrollment[]>;
   getSiteTopPages(): Promise<SiteTopPage[]>;
   getSiteTopPagesBySource(source: string): Promise<SiteTopPage[]>;
   getSiteTopPagesBySourceAndGeo(source: string, geo: string): Promise<SiteTopPage[]>;

package/src/models/site/site.model.js

@@ -10,7 +10,12 @@
  * governing permissions and limitations under the License.
  */
 
-import { composeAuditURL, hasText, isValidUrl } from '@adobe/spacecat-shared-utils';
+import {
+  composeAuditURL,
+  hasText,
+  isValidUrl,
+  DELIVERY_TYPES,
+} from '@adobe/spacecat-shared-utils';
 import BaseModel from '../base/base.model.js';
 
 const HLX_HOST = /\.(?:aem|hlx)\.(?:page|live)$/i;
@@ -64,12 +69,7 @@ export const getAuthoringType = (hostname, authoringTypes) => {
  * @extends BaseModel
  */
 class Site extends BaseModel {
-  static DELIVERY_TYPES = {
-    AEM_CS: 'aem_cs',
-    AEM_EDGE: 'aem_edge',
-    AEM_AMS: 'aem_ams',
-    OTHER: 'other',
-  };
+  static DELIVERY_TYPES = DELIVERY_TYPES;
 
   static DEFAULT_DELIVERY_TYPE = Site.DELIVERY_TYPES.AEM_EDGE;
 

package/src/models/site/site.schema.js

@@ -42,8 +42,10 @@ const schema = new SchemaBuilder(Site, SiteCollection)
   .addReference('has_one', 'LatestAudit', ['auditType'], { required: false })
   .addReference('has_many', 'Opportunities')
   .addReference('has_many', 'SiteCandidates')
+  .addReference('has_many', 'SiteEnrollments')
   .addReference('has_many', 'SiteTopForms')
   .addReference('has_many', 'SiteTopPages')
+  .addReference('has_many', 'TrialUserActivities')
   .addReference('has_many', 'PageIntents')
   .addAttribute('baseURL', {
     type: 'string',

package/src/models/site-enrollment/index.d.ts

@@ -0,0 +1,33 @@
+/*
+ * Copyright 2025 Adobe. All rights reserved.
+ * This file is licensed to you under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License. You may obtain a copy
+ * of the License at http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under
+ * the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR REPRESENTATIONS
+ * OF ANY KIND, either express or implied. See the License for the specific language
+ * governing permissions and limitations under the License.
+ */
+
+import type {
+  BaseCollection, BaseModel, Site, Entitlement,
+} from '../index';
+
+export type SiteEnrollmentStatus = 'ACTIVE' | 'SUSPENDED' | 'ENDED';
+
+export interface SiteEnrollment extends BaseModel {
+  getStatus(): SiteEnrollmentStatus;
+  getSite(): Promise<Site>;
+  getEntitlement(): Promise<Entitlement>;
+  setStatus(status: SiteEnrollmentStatus): SiteEnrollment;
+}
+
+export interface SiteEnrollmentCollection extends
+    BaseCollection<SiteEnrollment> {
+  allBySiteId(siteId: string): Promise<SiteEnrollment[]>;
+  allByEntitlementId(entitlementId: string): Promise<SiteEnrollment[]>;
+
+  findBySiteId(siteId: string): Promise<SiteEnrollment[]>;
+  findByEntitlementId(entitlementId: string): Promise<SiteEnrollment[]>;
+}

package/src/models/site-enrollment/index.js

@@ -0,0 +1,15 @@
+/*
+ * Copyright 2025 Adobe. All rights reserved.
+ * This file is licensed to you under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License. You may obtain a copy
+ * of the License at http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under
+ * the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR REPRESENTATIONS
+ * OF ANY KIND, either express or implied. See the License for the specific language
+ * governing permissions and limitations under the License.
+ */
+
+export { default as SiteEnrollment } from './site-enrollment.model.js';
+export { default as SiteEnrollmentCollection } from './site-enrollment.collection.js';
+export { default as SiteEnrollmentSchema } from './site-enrollment.schema.js';

package/src/models/site-enrollment/site-enrollment.collection.js

@@ -0,0 +1,26 @@
+/*
+ * Copyright 2025 Adobe. All rights reserved.
+ * This file is licensed to you under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License. You may obtain a copy
+ * of the License at http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under
+ * the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR REPRESENTATIONS
+ * OF ANY KIND, either express or implied. See the License for the specific language
+ * governing permissions and limitations under the License.
+ */
+
+import BaseCollection from '../base/base.collection.js';
+
+/**
+ * SiteEnrollmentCollection - A class representing a collection of SiteEnrollment entities.
+ * Provides methods to query and manipulate collections of SiteEnrollment data.
+ *
+ * @class SiteEnrollmentCollection
+ * @extends BaseCollection
+ */
+class SiteEnrollmentCollection extends BaseCollection {
+  // add your custom collection methods here
+}
+
+export default SiteEnrollmentCollection;

package/src/models/site-enrollment/site-enrollment.model.js

@@ -0,0 +1,26 @@
+/*
+ * Copyright 2025 Adobe. All rights reserved.
+ * This file is licensed to you under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License. You may obtain a copy
+ * of the License at http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under
+ * the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR REPRESENTATIONS
+ * OF ANY KIND, either express or implied. See the License for the specific language
+ * governing permissions and limitations under the License.
+ */
+
+import BaseModel from '../base/base.model.js';
+
+/**
+ * SiteEnrollment - A class representing a SiteEnrollment entity.
+ * Provides methods to access and manipulate SiteEnrollment-specific data.
+ *
+ * @class SiteEnrollment
+ * @extends BaseModel
+ */
+class SiteEnrollment extends BaseModel {
+  // add your custom methods or overrides here
+}
+
+export default SiteEnrollment;

package/src/models/site-enrollment/site-enrollment.schema.js

@@ -0,0 +1,29 @@
+/*
+ * Copyright 2025 Adobe. All rights reserved.
+ * This file is licensed to you under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License. You may obtain a copy
+ * of the License at http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under
+ * the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR REPRESENTATIONS
+ * OF ANY KIND, either express or implied. See the License for the specific language
+ * governing permissions and limitations under the License.
+ */
+
+import SchemaBuilder from '../base/schema.builder.js';
+import SiteEnrollment from './site-enrollment.model.js';
+import SiteEnrollmentCollection from './site-enrollment.collection.js';
+
+/*
+Schema Doc: https://electrodb.dev/en/modeling/schema/
+Attribute Doc: https://electrodb.dev/en/modeling/attributes/
+Indexes Doc: https://electrodb.dev/en/modeling/indexes/
+ */
+
+const schema = new SchemaBuilder(SiteEnrollment, SiteEnrollmentCollection)
+  // Reference to Site (many-to-one relationship)
+  .addReference('belongs_to', 'Site')
+  // Reference to Entitlement (many-to-one relationship)
+  .addReference('belongs_to', 'Entitlement');
+
+export default schema.build();

package/src/models/trial-user/index.d.ts

@@ -0,0 +1,51 @@
+/*
+ * Copyright 2025 Adobe. All rights reserved.
+ * This file is licensed to you under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License. You may obtain a copy
+ * of the License at http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under
+ * the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR REPRESENTATIONS
+ * OF ANY KIND, either express or implied. See the License for the specific language
+ * governing permissions and limitations under the License.
+ */
+
+import type {
+  BaseCollection, BaseModel, Organization,
+} from '../index';
+import type { ProviderType } from '../organization-identity-provider';
+
+export type Status = 'INVITED' | 'REGISTERED' | 'BLOCKED' | 'DELETED';
+
+export interface TrialUser extends BaseModel {
+  getExternalUserId(): string;
+  getStatus(): Status;
+  getProvider(): ProviderType;
+  getLastSeenAt(): Date | null;
+  getMetadata(): object | null;
+  getOrganization(): Promise<Organization>;
+  getEmailId(): string;
+  getFirstName(): string | null;
+  getLastName(): string | null;
+  getTrialUserActivities(): Promise<TrialUserActivity[]>;
+  setExternalUserId(externalUserId: string): TrialUser;
+  setStatus(status: Status): TrialUser;
+  setProvider(provider: ProviderType): TrialUser;
+  setLastSeenAt(lastSeenAt: Date): TrialUser;
+  setMetadata(metadata: object): TrialUser;
+  setEmailId(emailId: string): TrialUser;
+  setFirstName(firstName: string): TrialUser;
+  setLastName(lastName: string): TrialUser;
+}
+
+export interface TrialUserCollection extends BaseCollection<TrialUser> {
+  allByProvider(provider: ProviderType): Promise<TrialUser[]>;
+  allByProviderAndExternalUserId(provider: ProviderType, externalId: string): Promise<TrialUser[]>;
+  allByOrganizationId(organizationId: string): Promise<TrialUser[]>;
+  allByEmailId(emailId: string): Promise<TrialUser[]>;
+  findByProvider(provider: ProviderType): Promise<TrialUser[]>;
+  findByProviderAndExternalUserId(provider: ProviderType, externalId: string):
+    Promise<TrialUser[]>;
+  findByOrganizationId(organizationId: string): Promise<TrialUser[]>;
+  findByEmailId(emailId: string): Promise<TrialUser[]>;
+}

package/src/models/trial-user/index.js

@@ -0,0 +1,19 @@
+/*
+ * Copyright 2025 Adobe. All rights reserved.
+ * This file is licensed to you under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License. You may obtain a copy
+ * of the License at http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under
+ * the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR REPRESENTATIONS
+ * OF ANY KIND, either express or implied. See the License for the specific language
+ * governing permissions and limitations under the License.
+ */
+
+import TrialUser from './trial-user.model.js';
+import TrialUserCollection from './trial-user.collection.js';
+
+export {
+  TrialUser,
+  TrialUserCollection,
+};

package/src/models/trial-user/trial-user.collection.js

@@ -0,0 +1,26 @@
+/*
+ * Copyright 2025 Adobe. All rights reserved.
+ * This file is licensed to you under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License. You may obtain a copy
+ * of the License at http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under
+ * the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR REPRESENTATIONS
+ * OF ANY KIND, either express or implied. See the License for the specific language
+ * governing permissions and limitations under the License.
+ */
+
+import BaseCollection from '../base/base.collection.js';
+
+/**
+ * TrialUserCollection - A collection class responsible for managing TrialUser entities.
+ * Extends the BaseCollection to provide specific methods for interacting with TrialUser records.
+ *
+ * @class TrialUserCollection
+ * @extends BaseCollection
+ */
+class TrialUserCollection extends BaseCollection {
+  // add custom methods here
+}
+
+export default TrialUserCollection;

package/src/models/trial-user/trial-user.model.js

@@ -0,0 +1,35 @@
+/*
+ * Copyright 2025 Adobe. All rights reserved.
+ * This file is licensed to you under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License. You may obtain a copy
+ * of the License at http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under
+ * the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR REPRESENTATIONS
+ * OF ANY KIND, either express or implied. See the License for the specific language
+ * governing permissions and limitations under the License.
+ */
+
+import BaseModel from '../base/base.model.js';
+
+/**
+ * TrialUser - A class representing a trial user entity.
+ * Provides methods to access and manipulate trial user-specific data.
+ *
+ * @class TrialUser
+ * @extends BaseModel
+ */
+class TrialUser extends BaseModel {
+  /**
+   * Trial user status types.
+   * Any change to this object needs to be reflected in the index.d.ts file as well.
+   */
+  static STATUSES = {
+    INVITED: 'INVITED',
+    REGISTERED: 'REGISTERED', // User has logged in at least once
+    BLOCKED: 'BLOCKED',
+    DELETED: 'DELETED',
+  };
+}
+
+export default TrialUser;

package/src/models/trial-user/trial-user.schema.js

@@ -0,0 +1,68 @@
+/*
+ * Copyright 2025 Adobe. All rights reserved.
+ * This file is licensed to you under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License. You may obtain a copy
+ * of the License at http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under
+ * the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR REPRESENTATIONS
+ * OF ANY KIND, either express or implied. See the License for the specific language
+ * governing permissions and limitations under the License.
+ */
+
+import { isIsoDate, isObject } from '@adobe/spacecat-shared-utils';
+import SchemaBuilder from '../base/schema.builder.js';
+import TrialUser from './trial-user.model.js';
+import TrialUserCollection from './trial-user.collection.js';
+import OrganizationIdentityProvider from '../organization-identity-provider/organization-identity-provider.model.js';
+
+/*
+Schema Doc: https://electrodb.dev/en/modeling/schema/
+Attribute Doc: https://electrodb.dev/en/modeling/attributes/
+Indexes Doc: https://electrodb.dev/en/modeling/indexes/
+*/
+
+const schema = new SchemaBuilder(TrialUser, TrialUserCollection)
+  // Reference to Organization (many-to-one relationship)
+  .addReference('belongs_to', 'Organization')
+  // Reference to TrialUserActivity (one-to-many relationship)
+  .addReference('has_many', 'TrialUserActivities')
+  .addAttribute('externalUserId', {
+    type: 'string',
+    required: false,
+  })
+  .addAttribute('status', {
+    type: Object.values(TrialUser.STATUSES),
+    required: true,
+  })
+  .addAttribute('provider', {
+    type: Object.values(OrganizationIdentityProvider.PROVIDER_TYPES),
+    required: false,
+  })
+  .addAttribute('lastSeenAt', {
+    type: 'string',
+    validate: (value) => !value || isIsoDate(value),
+  })
+  .addAttribute('emailId', {
+    type: 'string',
+    required: true,
+  })
+  .addAttribute('firstName', {
+    type: 'string',
+    required: false,
+  })
+  .addAttribute('lastName', {
+    type: 'string',
+    required: false,
+  })
+  .addAttribute('metadata', {
+    type: 'any',
+    validate: (value) => !value || isObject(value),
+  })
+  .addAllIndex(['emailId'])
+  .addIndex(
+    { composite: ['provider'] },
+    { composite: ['externalUserId'] },
+  );
+
+export default schema.build();

package/src/models/trial-user-activity/index.d.ts

@@ -0,0 +1,44 @@
+/*
+ * Copyright 2025 Adobe. All rights reserved.
+ * This file is licensed to you under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License. You may obtain a copy
+ * of the License at http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under
+ * the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR REPRESENTATIONS
+ * OF ANY KIND, either express or implied. See the License for the specific language
+ * governing permissions and limitations under the License.
+ */
+
+import type {
+  BaseCollection, BaseModel,
+} from '../base';
+import type { EntitlementProductCode } from '../entitlement';
+
+export type Type = 'SIGN_UP' | 'SIGN_IN' | 'CREATE_SITE' | 'RUN_AUDIT' | 'PROMPT_RUN' | 'DOWNLOAD';
+
+export interface TrialUserActivity extends BaseModel {
+  getType(): Type;
+  getDetails(): object | null;
+  getProductCode(): EntitlementProductCode;
+  getEntitlementId(): string;
+  getTrialUserId(): string;
+  getSiteId(): string;
+  setType(type: Type): TrialUserActivity;
+  setDetails(details: object): TrialUserActivity;
+  setProductCode(productCode: string): TrialUserActivity;
+}
+
+export interface TrialUserActivityCollection extends BaseCollection<TrialUserActivity> {
+  allByEntitlementId(entitlementId: string): Promise<TrialUserActivity[]>;
+  allByProductCode(productCode: EntitlementProductCode): Promise<TrialUserActivity[]>;
+  allByProductCodeAndCreatedAt(productCode: EntitlementProductCode, createdAt: string):
+    Promise<TrialUserActivity[]>;
+  allBySiteId(siteId: string): Promise<TrialUserActivity[]>;
+
+  findByEntitlementId(entitlementId: string): Promise<TrialUserActivity[]>;
+  findByProductCode(productCode: EntitlementProductCode): Promise<TrialUserActivity[]>;
+  findByProductCodeAndCreatedAt(productCode: EntitlementProductCode, createdAt: string):
+    Promise<TrialUserActivity[]>;
+  findBySiteId(siteId: string): Promise<TrialUserActivity[]>;
+}

package/src/models/trial-user-activity/index.js

@@ -0,0 +1,19 @@
+/*
+ * Copyright 2025 Adobe. All rights reserved.
+ * This file is licensed to you under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License. You may obtain a copy
+ * of the License at http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under
+ * the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR REPRESENTATIONS
+ * OF ANY KIND, either express or implied. See the License for the specific language
+ * governing permissions and limitations under the License.
+ */
+
+import TrialUserActivity from './trial-user-activity.model.js';
+import TrialUserActivityCollection from './trial-user-activity.collection.js';
+
+export {
+  TrialUserActivity,
+  TrialUserActivityCollection,
+};

package/src/models/trial-user-activity/trial-user-activity.collection.js

@@ -0,0 +1,28 @@
+/*
+ * Copyright 2025 Adobe. All rights reserved.
+ * This file is licensed to you under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License. You may obtain a copy
+ * of the License at http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under
+ * the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR REPRESENTATIONS
+ * OF ANY KIND, either express or implied. See the License for the specific language
+ * governing permissions and limitations under the License.
+ */
+
+import BaseCollection from '../base/base.collection.js';
+
+/**
+ * TrialUserActivityCollection - A collection class responsible for managing TrialUserActivity
+ * entities.
+ * Extends the BaseCollection to provide specific methods for interacting with TrialUserActivity
+ * records.
+ *
+ * @class TrialUserActivityCollection
+ * @extends BaseCollection
+ */
+class TrialUserActivityCollection extends BaseCollection {
+  // add custom methods here
+}
+
+export default TrialUserActivityCollection;

package/src/models/trial-user-activity/trial-user-activity.model.js

@@ -0,0 +1,37 @@
+/*
+ * Copyright 2025 Adobe. All rights reserved.
+ * This file is licensed to you under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License. You may obtain a copy
+ * of the License at http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under
+ * the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR REPRESENTATIONS
+ * OF ANY KIND, either express or implied. See the License for the specific language
+ * governing permissions and limitations under the License.
+ */
+
+import BaseModel from '../base/base.model.js';
+
+/**
+ * TrialUserActivity - A class representing a trial user activity entity.
+ * Provides methods to access and manipulate trial user activity-specific data.
+ *
+ * @class TrialUserActivity
+ * @extends BaseModel
+ */
+class TrialUserActivity extends BaseModel {
+  /**
+   * Trial user activity types.
+   * Any change to this object needs to be reflected in the index.d.ts file as well.
+   */
+  static TYPES = {
+    SIGN_UP: 'SIGN_UP',
+    SIGN_IN: 'SIGN_IN',
+    CREATE_SITE: 'CREATE_SITE',
+    RUN_AUDIT: 'RUN_AUDIT',
+    PROMPT_RUN: 'PROMPT_RUN',
+    DOWNLOAD: 'DOWNLOAD',
+  };
+}
+
+export default TrialUserActivity;

package/src/models/trial-user-activity/trial-user-activity.schema.js

@@ -0,0 +1,47 @@
+/*
+ * Copyright 2025 Adobe. All rights reserved.
+ * This file is licensed to you under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License. You may obtain a copy
+ * of the License at http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software distributed under
+ * the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR REPRESENTATIONS
+ * OF ANY KIND, either express or implied. See the License for the specific language
+ * governing permissions and limitations under the License.
+ */
+
+import SchemaBuilder from '../base/schema.builder.js';
+import TrialUserActivity from './trial-user-activity.model.js';
+import TrialUserActivityCollection from './trial-user-activity.collection.js';
+import Entitlement from '../entitlement/entitlement.model.js';
+
+/*
+Schema Doc: https://electrodb.dev/en/modeling/schema/
+Attribute Doc: https://electrodb.dev/en/modeling/attributes/
+Indexes Doc: https://electrodb.dev/en/modeling/indexes/
+*/
+
+const schema = new SchemaBuilder(TrialUserActivity, TrialUserActivityCollection)
+  // Reference to OrganizationIdentityProvider (many-to-one relationship)
+  .addReference('belongs_to', 'TrialUser')
+  // Reference to Organization (many-to-one relationship)
+  .addReference('belongs_to', 'Entitlement')
+  // Reference to Site (many-to-one relationship)
+  .addReference('belongs_to', 'Site')
+  .addAttribute('type', {
+    type: Object.values(TrialUserActivity.TYPES),
+    required: true,
+  })
+  .addAttribute('details', {
+    type: 'any',
+  })
+  .addAttribute('productCode', {
+    type: Object.values(Entitlement.PRODUCT_CODES),
+    required: true,
+  })
+  .addIndex(
+    { composite: ['productCode'] },
+    { composite: ['createdAt'] },
+  );
+
+export default schema.build();

package/src/util/util.js

@@ -36,7 +36,14 @@ const referenceToBaseMethodName = (reference) => {
 
 const entityNameToAllPKValue = (entityName) => `ALL_${pluralize.plural(entityName.toUpperCase())}`;
 
-const idNameToEntityName = (idName) => capitalize(pluralize.singular(idName.replace('Id', '')));
+const idNameToEntityName = (idName) => {
+  let result = idName;
+  if (idName.endsWith('Id')) {
+    result = result.replace('Id', '');
+  }
+
+  return capitalize(pluralize.singular(result));
+};
 
 const isPositiveInteger = (value) => isInteger(value) && value > 0;