Added branding handlers (#86)

* Added branding handlers

* Changed getColorPalette connection to replica

* Moved json validator

* Revert "Moved json validator"

This reverts commit 805c60d.

* Added branding migration with branding not null

* Code review fixes

---------

Co-authored-by: Stanislav Kiselev <“[email protected]”>

src/components/validation-schema-compiler.ts

@@ -5,6 +5,7 @@ import {
     MAX_META_OBJECT_SYMBOLS,
     MAX_UNVERSIONED_DATA_OBJECT_SYMBOLS,
     MAX_PRESET_DATA_OBJECT_SYMBOLS,
+    MAX_BRANDING_OBJECT_SYMBOLS,
 } from '../const';
 import US_ERRORS from '../const/us-error-constants';
 import {AppError} from '@gravity-ui/nodekit';
@@ -115,6 +116,30 @@ ajv.addKeyword('restrictMetaSize', {
     errors: true,
 });
 
+ajv.addKeyword('restrictBrandingSize', {
+    validate: function validate(schema: object | string | boolean, data: any) {
+        if (data === null) {
+            return true;
+        } else {
+            const dataStringified = JSON.stringify(data);
+
+            // @ts-ignore
+            validate.errors = [
+                {
+                    format: 'string',
+                    message: `branding jsonb object can contain not greater ${MAX_BRANDING_OBJECT_SYMBOLS} symbols`,
+                    params: {
+                        keyword: 'restrictBrandingSize',
+                    },
+                },
+            ];
+
+            return dataStringified.length <= MAX_BRANDING_OBJECT_SYMBOLS;
+        }
+    },
+    errors: true,
+});
+
 ajv.addKeyword('restrictUnversionedDataSize', {
     validate: function validate(schema: object | string | boolean, data: any) {
         if (data === null) {

src/const/common.ts

@@ -146,6 +146,7 @@ export const CODING_BASE = '0123456789abcdefghijklmnopqrstuvwxyz'.split('');
 export const MAX_META_OBJECT_SYMBOLS = 2000;
 export const MAX_UNVERSIONED_DATA_OBJECT_SYMBOLS = 5000;
 export const MAX_PRESET_DATA_OBJECT_SYMBOLS = 15000;
+export const MAX_BRANDING_OBJECT_SYMBOLS = 15000;
 
 export const AJV_PATTERN_KEYS_NOT_OBJECT = {
     '.*': {

src/db/migrations/20240306213719_change_branding_column_in_tenants_to_not_null.ts

@@ -0,0 +1,13 @@
+import type {Knex} from 'knex';
+
+exports.up = function (knex: Knex): Promise<unknown> {
+    return knex.raw(`
+        ALTER TABLE tenants ALTER COLUMN branding SET NOT NULL;
+    `);
+};
+
+exports.down = function (knex: Knex): Promise<unknown> {
+    return knex.raw(`
+        ALTER TABLE tenants ALTER COLUMN branding DROP NOT NULL;
+    `);
+};

src/db/models/new/entry/index.ts

@@ -27,6 +27,7 @@ export const EntryColumn = {
     UnversionedData: 'unversionedData',
     WorkbookId: 'workbookId',
     Mirrored: 'mirrored',
+    Branding: 'branding',
 } as const;
 
 export class Entry extends Model {

src/db/models/new/tenant/index.ts

@@ -15,6 +15,7 @@ export const TenantColumn = {
     BillingAccountId: 'billingAccountId',
     BillingInstanceServiceId: 'billingInstanceServiceId',
     BillingStartedAt: 'billingStartedAt',
+    Branding: 'branding',
 } as const;
 
 export class Tenant extends Model {
@@ -39,6 +40,7 @@ export class Tenant extends Model {
     [TenantColumn.BillingAccountId]!: Nullable<string>;
     [TenantColumn.BillingInstanceServiceId]!: Nullable<string>;
     [TenantColumn.BillingStartedAt]!: Nullable<string>;
+    [TenantColumn.Branding]!: Record<string, unknown>;
 }
 
 export {BillingRate};

src/db/presentations/joined-migration-tenant/index.ts

@@ -17,6 +17,7 @@ const selectedTenantColumns = [
     TenantColumn.BillingAccountId,
     TenantColumn.BillingInstanceServiceId,
     TenantColumn.BillingStartedAt,
+    TenantColumn.Branding,
 ] as const;
 
 const selectedMigrationTenantColumns = [
@@ -44,6 +45,7 @@ export const JoinedMigrationTenantColumn = {
     ToId: MigrationTenantColumn.ToId,
     Migrating: MigrationTenantColumn.Migrating,
     MigrationMeta: MigrationTenantColumn.MigrationMeta,
+    Branding: TenantColumn.Branding,
 } as const;
 
 export const selectedJoinedMigrationTenantColumns = [

src/services/color-palettes/get-color-palette.ts

@@ -14,7 +14,7 @@ export const getColorPalette = async (
         colorPaletteId,
     });
 
-    const colorPalette = await ColorPaletteModel.query(ColorPaletteModel.primary)
+    const colorPalette = await ColorPaletteModel.query(ColorPaletteModel.replica)
         .select()
         .where({
             [ColorPaletteModelColumn.ColorPaletteId]: colorPaletteId,