eventsFactory.js

import safe from 'safe-regex';
import { createProjectEventsByIdLoader } from '../dataLoaders';
import RedisHelper from '../redisHelper';
import ChartDataService from '../services/chartDataService';
import { getMidnightWithTimezoneOffset, getUTCMidnight } from '../utils/dates';

const Factory = require('./modelFactory');
const mongo = require('../mongo');
const Event = require('../models/event');
const { ObjectId } = require('mongodb');
const { composeEventPayloadByRepetition } = require('../utils/merge');

const MAX_DB_READ_BATCH_SIZE = Number(process.env.MAX_DB_READ_BATCH_SIZE);

/**
 * Chart series labels
 */
const ChartType = {
  Accepted: 'accepted',
  RateLimited: 'rate-limited',
};

/**
 * @typedef {import('mongodb').UpdateWriteOpResult} UpdateWriteOpResult
 */

/**
 * @typedef {Object} EventRepetitionSchema
 * @property {String} _id — repetition's identifier
 * @property {String} groupHash - event's hash. Generates according to the rule described in EventSchema
 * @property {EventPayload} payload - repetition's payload
 * @property {Number} timestamp - repetition's Unix timestamp
 * @property {Number} originalTimestamp - UNIX timestmap of the original event
 * @property {String} originalEventId - id of the original event
 * @property {String} projectId - id of the project, which repetition it is
 */

/**
 * @typedef {Object} EventRepetitionsPortionSchema
 * @property {EventRepetitionSchema[]} repetitions - list of repetitions
 * @property {String | null} nextCursor - pointer to the first repetition of the next portion, null if there are no repetitions left
 */

/**
 * @typedef {Object} DailyEventSchema
 * @property {String} _id - id of the dailyEvent
 * @property {String} groupHash - group hash of the dailyEvent
 * @property {Number} groupingTimestamp - UNIX timestamp that represents the day of dailyEvent
 * @property {Number} affectedUsers - number of users affected this day
 * @property {Number} count - number of events this day
 * @property {String} lastRepetitionId - id of the last repetition this day
 * @property {Number} lastRepetitionTime - UNIX timestamp that represent time of the last repetition this day
 * @property {Event} event - one certain event that represents all of the repetitions this day
 */

/**
 * @typedef {Object} DailyEventsCursor
 * @property {Number} groupingTimestampBoundary - boundary value of groupingTimestamp field of the last event in the portion
 * @property {Number} sortValueBoundary - boundary value of the field by which events are sorted (count/affectedUsers/lastRepetitionTime) of the last event in the portion
 * @property {String} idBoundary - boundary value of _id field of the last event in the portion
 */

/**
 * @typedef {Object} DaylyEventsPortionSchema
 * @property {DailyEventSchema[]} dailyEvents - original event of the daily one
 * @property {DailyEventsCursor | null} nextCursor - object with boundary values of the first event in the next portion
 */

/**
 * @typedef {Object} EventsFilters
 * @property {boolean} [starred] - if true, events with 'starred' mark should be included to the output
 * @property {boolean} [resolved] - if true, events with 'resolved' should be included to the output
 * @property {boolean} [ignored] - if true, events with 'ignored' mark should be included to the output
 */

/**
 * EventsFactory
 *
 * Factory Class for Event's Model
 */
class EventsFactory extends Factory {
  /**
   * Event types with collections where they stored
   * @return {{EVENTS: string, DAILY_EVENTS: string, REPETITIONS: string, RELEASES: string}}
   * @constructor
   */
  get TYPES() {
    return {
      EVENTS: 'events',
      REPETITIONS: 'repetitions',
      DAILY_EVENTS: 'dailyEvents',
      RELEASES: 'releases',
    };
  }

  /**
   * Creates Event instance
   * @param {ObjectId} projectId
   */
  constructor(projectId) {
    super();

    /**
     * Redis helper instance (singleton)
     */
    this.redis = RedisHelper.getInstance();

    /**
     * Chart data service for fetching data from Redis TimeSeries
     */
    this.chartDataService = new ChartDataService(this.redis);

    if (!projectId) {
      throw new Error('Can not construct Event model, because projectId is not provided');
    }

    this.projectId = projectId;
    this.eventsDataLoader = createProjectEventsByIdLoader(mongo.databases.events, this.projectId);
  }

  /**
   * Returns pointer to the collection
   *
   * @param {String} type - each events in order to optimization holds data in different collections.
   *                      This argument defines which collection need to be used.
   *
   * @returns {String}
   */
  getCollection(type) {
    return mongo.databases.events.collection(
      type + ':' + this.projectId
    );
  }

  /**
   * Is collection of events exists
   *
   * @param {String} type - type of collection to check
   *
   * @return {Promise<boolean>}
   */
  isCollectionExists(type) {
    return mongo.databases.events.listCollections({ name: type + ':' + this.projectId }).hasNext();
  }

  /**
   * Finds events by passed query
   *
   * @param {object} [query={}] - query
   * @param {Number} [limit=10] - query limit
   * @param {Number} [skip=0] - query skip
   * @returns {Event[]} - events matching query
   */
  async find(query = {}, limit = 10, skip = 0) {
    limit = this.validateLimit(limit);
    skip = this.validateSkip(skip);

    const cursor = this.getCollection(this.TYPES.EVENTS)
      .find(query)
      .sort([ ['_id', -1] ])
      .limit(limit)
      .skip(skip);

    const result = await cursor.toArray();

    return result.map(eventSchema => {
      return new Event({
        ...eventSchema,
        projectId: this.projectId,
      });
    });
  }

  /**
   * Find event by id
   *
   * @param {string|ObjectId} id - event's id
   * @returns {Event|null}
   */
  async findById(id) {
    const searchResult = await this.eventsDataLoader.load(id);

    const event = searchResult ? new Event(searchResult) : null;

    return event;
  }

  /**
   * Find an event by any custom query
   *
   * @param {object} query - any custom mongo query
   * @return {Event}
   */
  async findOneByQuery(query) {
    const searchResult = await this.getCollection(this.TYPES.EVENTS)
      .findOne(query);

    return new Event(searchResult);
  }

  /**
   * Returns events that grouped by day
   *
   * @param {Number} limit - events count limitations
   * @param {DailyEventsCursor} paginationCursor - object that contains boundary values of the last event in the previous portion
   * @param {'BY_DATE' | 'BY_COUNT' | 'BY_AFFECTED_USERS'} sort - events sort order
   * @param {EventsFilters} filters - marks by which events should be filtered (resolved, starred, ignored only; assignee is separate)
   * @param {String} search - Search query
   * @param {String|undefined} release - release name
   * @param {String|undefined} assignee - user id or __filter_unassigned__ / __filter_any_assignee__
   *
   * @return {DaylyEventsPortionSchema}
   */
  async findDailyEventsPortion(
    limit = 10,
    paginationCursor = null,
    sort = 'BY_DATE',
    filters = {},
    search = '',
    release,
    assignee
  ) {
    if (typeof search !== 'string') {
      throw new Error('Search parameter must be a string');
    }

    /**
     * Check if pattern is safe RegExp
     */
    if (!safe(search)) {
      throw new Error('Invalid regular expression pattern');
    }

    switch (sort) {
      case 'BY_COUNT':
        sort = 'count';
        break;
      case 'BY_DATE':
        sort = 'lastRepetitionTime';
        break;
      case 'BY_AFFECTED_USERS':
        sort = 'affectedUsers';
        break;
      default:
        sort = 'lastRepetitionTime';
        break;
    }

    const escapedSearch = search.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');

    limit = this.validateLimit(limit);

    const pipeline = [
      {
        $match: paginationCursor ? {
          /**
           * This condition is used for cursor-based pagination
           * We sort result by groupingTimestamp desc, [sort] desc, _id desc
           * So we need to fetch documents that are less than the last document of the previous portion (based on all three conditions)
           */
          $or: [
            {
              /**
               * If groupingTimestamp is less than the cursors one
               * - daily events of the next day
               */
              groupingTimestamp: { $lt: paginationCursor.groupingTimestampBoundary },
            },
            {
              /**
               * If groupingTimestamp equals to the cursor one, but [sort] is less than the cursors one
               * - daily events of the same day, but with less count/affectedUsers/lastRepetitionTime
               */
              $and: [
                { groupingTimestamp: paginationCursor.groupingTimestampBoundary },
                { [sort]: { $lt: paginationCursor.sortValueBoundary } },
              ],
            },
            {
              /**
               * If groupingTimestamp and [sort] equals to the cursors ones, but _id is less or equal to the cursors one
               * - daily events of the same day with the same count/affectedUsers/lastRepetitionTime, but that were created earlier
               */
              $and: [
                { groupingTimestamp: paginationCursor.groupingTimestampBoundary },
                { [sort]: paginationCursor.sortValueBoundary },
                { _id: { $lte: new ObjectId(paginationCursor.idBoundary) } },
              ],
            },
          ],
        } : {},
      },
      {
        $sort: {
          groupingTimestamp: -1,
          [sort]: -1,
          _id: -1,
        },
      },
    ];

    const searchFilter = search.trim().length > 0
      ? {
        $or: [
          {
            'repetition.delta': {
              $regex: escapedSearch,
              $options: 'i',
            },
          },
          {
            'event.payload.title': {
              $regex: escapedSearch,
              $options: 'i',
            },
          },
          {
            'event.payload.backtrace.file': {
              $regex: escapedSearch,
              $options: 'i',
            },
          },
          {
            'event.payload.context': {
              $regex: escapedSearch,
              $options: 'i',
            },
          },
          {
            'event.payload.addons': {
              $regex: escapedSearch,
              $options: 'i',
            },
          },
        ],
      }
      : {};

    const markFilters = ['resolved', 'starred', 'ignored'];
    const matchFilter = filters
      ? Object.fromEntries(
        Object
          .entries(filters)
          .filter(([ mark ]) => markFilters.includes(mark))
          .map(([mark, exists]) => [`event.marks.${mark}`, { $exists: exists } ])
      )
      : {};

    // Filter by release if provided (coerce event payload release to string)
    const releaseFilter = release
      ? {
        $expr: {
          $eq: [
            {
              $convert: {
                input: '$event.payload.release',
                to: 'string',
                onError: '',
                onNull: '',
              },
            },
            String(release),
          ],
        },
      }
      : {};

    /**
     * Sentinel values from garage assignee filter (not user ids)
     */
    const FILTER_UNASSIGNED = '__filter_unassigned__';
    const FILTER_ANY_ASSIGNEE = '__filter_any_assignee__';

    const assigneeFilter = (() => {
      if (!assignee) {
        return {};
      }
      if (assignee === FILTER_UNASSIGNED) {
        /**
         * Use $and so this does not collide with searchFilter’s top-level $or in $match spread
         */
        return {
          $and: [
            {
              $or: [
                { 'event.assignee': { $exists: false } },
                { 'event.assignee': null },
                { 'event.assignee': '' },
              ],
            },
          ],
        };
      }
      if (assignee === FILTER_ANY_ASSIGNEE) {
        return {
          'event.assignee': {
            $exists: true,
            $nin: [null, ''],
          },
        };
      }

      return { 'event.assignee': String(assignee) };
    })();

    /**
     * These filters match joined event.* fields, so their $match must run
     * after the lookups. With none set, $limit can move before the lookups
     * and skip joining rows we'd drop anyway (~8.8s).
     * Trim search to match searchFilter's own condition.
     */
    const hasContentFilters =
      search.trim().length > 0 ||
      Boolean(release) ||
      Boolean(assignee) ||
      Object.keys(matchFilter).length > 0;

    if (!hasContentFilters) {
      pipeline.push({ $limit: limit + 1 });
    }

    pipeline.push(
      /**
       * Left outer join original event on groupHash field
       */
      {
        $lookup: {
          from: 'events:' + this.projectId,
          localField: 'groupHash',
          foreignField: 'groupHash',
          as: 'event',
        },
      },
      {
        $lookup: {
          from: 'repetitions:' + this.projectId,
          localField: 'lastRepetitionId',
          foreignField: '_id',
          as: 'repetition',
        },
      },
      /**
       * Desctruct event and repetition arrays since there are only one document in both arrays
       */
      {
        $unwind: '$event',
      },
      {
        $unwind: {
          path: '$repetition',
          preserveNullAndEmptyArrays: true,
        },
      }
    );

    if (hasContentFilters) {
      pipeline.push(
        {
          $match: {
            ...matchFilter,
            ...searchFilter,
            ...releaseFilter,
            ...assigneeFilter,
          },
        },
        { $limit: limit + 1 }
      );
    }

    pipeline.push({ $unset: 'groupHash' });

    const cursor = this.getCollection(this.TYPES.DAILY_EVENTS).aggregate(pipeline);
    const result = await cursor.toArray();

    let nextCursor;

    if (result.length === limit + 1) {
      const nextCursorEvent = result.pop();

      nextCursor = {
        groupingTimestampBoundary: nextCursorEvent.groupingTimestamp,
        sortValueBoundary: nextCursorEvent[sort],
        idBoundary: nextCursorEvent._id.toString(),
      };
    }

    const composedResult = result.map(dailyEvent => {
      const repetition = dailyEvent.repetition;
      const event = dailyEvent.event;

      dailyEvent.event = this._composeEventWithRepetition(event, repetition);
      dailyEvent.id = dailyEvent._id.toString();

      delete dailyEvent.repetition;
      delete dailyEvent._id;

      return dailyEvent;
    });

    return {
      nextCursor: nextCursor,
      dailyEvents: composedResult,
    };
  }

  /**
   * Get project chart data from Redis or fallback to MongoDB
   *
   * @param {string} projectId - project ID
   * @param {string} startDate - start date (ISO string)
   * @param {string} endDate - end date (ISO string)
   * @param {number} groupBy - grouping interval in minutes (1=minute, 60=hour, 1440=day)
   * @param {number} timezoneOffset - user's local timezone offset in minutes
   * @returns {Promise<Array>}
   */
  async getProjectChartData(projectId, startDate, endDate, groupBy = 60, timezoneOffset = 0) {
    // Calculate days for MongoDB fallback
    const start = new Date(startDate).getTime();
    const end = new Date(endDate).getTime();
    const days = Math.ceil((end - start) / (24 * 60 * 60 * 1000));

    try {
      const [acceptedSeries, rateLimitedSeries] = await Promise.all([
        this.chartDataService.getProjectChartData(
          projectId,
          startDate,
          endDate,
          groupBy,
          timezoneOffset,
          'events-accepted'
        ),
        this.chartDataService.getProjectChartData(
          projectId,
          startDate,
          endDate,
          groupBy,
          timezoneOffset,
          'events-rate-limited'
        ),
      ]);

      return [
        {
          label: ChartType.Accepted,
          data: acceptedSeries,
        },
        {
          label: ChartType.RateLimited,
          data: rateLimitedSeries,
        },
      ];
    } catch (err) {
      console.error('[EventsFactory] getProjectChartData error:', err);

      const fallbackAccepted = await this.findChartData(days, timezoneOffset, '');

      return [
        {
          label: ChartType.Accepted,
          data: fallbackAccepted,
        },
        {
          label: ChartType.RateLimited,
          data: this._composeZeroSeries(fallbackAccepted),
        },
      ];
    }
  }

  /**
   * Get event daily chart data from MongoDB only
   *
   * @param {string} groupHash - event's group hash
   * @param {number} days - how many days to fetch
   * @param {number} timezoneOffset - user's local timezone offset in minutes
   * @returns {Promise<Array>}
   */
  async getEventDailyChart(groupHash, days, timezoneOffset = 0) {
    const data = await this.findChartData(days, timezoneOffset, groupHash);

    return [
      {
        label: ChartType.Accepted,
        data,
      },
    ];
  }

  /**
   * Fetch timestamps and total count of errors (or target error) for each day since
   *
   * @param {number} days - how many days we need to fetch for displaying in a chart
   * @param {number} timezoneOffset - user's local timezone offset in minutes
   * @param {string} groupHash - event's group hash for showing only target event
   * @return {ProjectChartItem[]}
   */
  async findChartData(days, timezoneOffset = 0, groupHash = '') {
    const today = new Date();
    const since = today.setDate(today.getDate() - days) / 1000;

    /**
     * Compose options for find method
     * @type {{groupingTimestamp: {$gt: number}}}
     */
    const options = {
      groupingTimestamp: {
        $gt: since,
      },
    };

    /**
     * Add eq check if groupHash was passed
     */
    if (groupHash) {
      options.groupHash = {
        $eq: groupHash,
      };
    }

    const dailyEventsCursor = await this.getCollection(this.TYPES.DAILY_EVENTS)
      .find(options, {
        projection: {
          lastRepetitionTime: 1,
          groupingTimestamp: 1,
          count: 1,
        },
      })
      .batchSize(MAX_DB_READ_BATCH_SIZE);

    const groupedCounts = {};

    for await (const item of dailyEventsCursor) {
      const groupingTimestamp = getMidnightWithTimezoneOffset(
        item.lastRepetitionTime,
        item.groupingTimestamp,
        timezoneOffset
      );

      const key = `groupingTimestamp:${groupingTimestamp}`;
      const current = groupedCounts[key] || 0;

      if (item.count === undefined || item.count === null) {
        console.warn(`Missing 'count' field for daily event with key ${key}. Defaulting to 0.`);
        groupedCounts[key] = current;
      } else {
        groupedCounts[key] = current + item.count;
      }
    }

    /**
     * Now fill all requested days
     */
    let result = [];

    for (let i = 0; i < days; i++) {
      const now = new Date();
      const day = new Date(now.setDate(now.getDate() - i));
      const dayMidnight = getUTCMidnight(day) / 1000;

      let groupedCount = groupedCounts[`groupingTimestamp:${dayMidnight}`];

      if (!groupedCount) {
        groupedCount = 0;
      }

      result.push({
        timestamp: dayMidnight,
        count: groupedCount,
      });
    }

    /**
     * Order by time ascendance
     */
    result = result.sort((a, b) => a.timestamp - b.timestamp);

    return result;
  }

  /**
   * Compose zero-filled chart series using timestamps from the provided template
   *
   * @param {Array<{timestamp: number, count: number}>} template - reference series for timestamps
   * @returns {Array<{timestamp: number, count: number}>}
   */
  _composeZeroSeries(template = []) {
    if (!Array.isArray(template) || template.length === 0) {
      return [];
    }

    return template.map((point) => ({
      timestamp: point.timestamp,
      count: 0,
    }));
  }

  /**
   * Returns number of documents that occurred after the last visit time
   *
   * @param {Number} lastVisit - user's last visit time on project
   *
   * @return {Promise<Number>}
   *
   * @todo move to Project model
   */
  async getUnreadCount(lastVisit) {
    const query = {
      timestamp: {
        $gt: lastVisit,
      },
    };

    return this.getCollection(this.TYPES.EVENTS)
      .countDocuments(query);
  }

  /**
   * Returns Event repetitions
   *
   * @param {string|ObjectId} originalEventId - id of the original event
   * @param {Number} limit - count limitations
   * @param {Number} cursor - pointer to the next repetition
   *
   * @return {EventRepetitionsPortionSchema}
   */
  async getEventRepetitions(originalEventId, limit = 10, cursor = null) {
    limit = this.validateLimit(limit);

    cursor = cursor ? new ObjectId(cursor) : null;

    const result = {
      repetitions: [],
      nextCursor: null,
    };

    /**
     * Get original event
     * @type {Event}
     */
    const eventOriginal = await this.findById(originalEventId);

    if (!eventOriginal) {
      throw new Error(`Original event not found for ${originalEventId}`);
    }

    /**
     * Get portion based on cursor if cursor is not null
     */
    const query = cursor ? {
      groupHash: eventOriginal.groupHash,
      _id: { $lte: cursor },
    } : {
      groupHash: eventOriginal.groupHash,
    };

    /**
     * Collect repetitions
     * @type {EventRepetitionSchema[]}
     */
    const repetitions = await this.getCollection(this.TYPES.REPETITIONS)
      .find(query)
      .sort({ _id: -1 })
      .limit(limit + 1)
      .toArray();

    if (repetitions.length === limit + 1) {
      result.nextCursor = repetitions.pop()._id;
    }

    for (const repetition of repetitions) {
      const event = this._composeEventWithRepetition(eventOriginal, repetition);

      result.repetitions.push({
        ...event,
        projectId: this.projectId,
      });
    }

    const isLastPortion = result.nextCursor === null;

    /**
     * For last portion:
     * add original event to the end of repetitions list
     */
    if (isLastPortion) {
      /**
       * Get only 'repetitions' fields from event to fit Repetition scheme
       * @type {EventRepetitionSchema}
       */
      const firstRepetition = {
        ...eventOriginal,
        originalTimestamp: eventOriginal.timestamp,
        originalEventId: eventOriginal._id,
        projectId: this.projectId,
      };

      result.repetitions.push(firstRepetition);
    }

    return result;
  }

  /**
   * Returns certain repetition of the original event
   *
   * @param {String} repetitionId - id of Repetition to find
   * @param {String} originalEventId - id of the original event
   * @return {EventRepetitionSchema|null}
   */
  async getEventRepetition(repetitionId, originalEventId) {
    /**
     * If originalEventId equals repetitionId than user wants to get first repetition which is original event
     */
    if (repetitionId.toString() === originalEventId.toString()) {
      const originalEvent = await this.eventsDataLoader.load(originalEventId);

      /**
       * All events have same type with originalEvent id
       */
      originalEvent.originalEventId = originalEventId;
      originalEvent.originalTimestamp = originalEvent.timestamp;
      originalEvent.projectId = this.projectId;

      return originalEvent || null;
    }

    /**
     * Otherwise we need to get original event and repetition and merge them
     */
    const repetition = await this.getCollection(this.TYPES.REPETITIONS)
      .findOne({
        _id: new ObjectId(repetitionId),
      });

    const originalEvent = await this.eventsDataLoader.load(originalEventId);

    /**
     * If one of the ids are invalid (originalEvent or repetition not found) return null
     */
    if (!originalEvent || !repetition) {
      throw new Error(`Cant find event repetition for repetitionId: ${repetitionId} and originalEventId: ${originalEventId}`);
    }

    return this._composeEventWithRepetition(originalEvent, repetition);
  }

  /**
   * Return last occurrence of event
   * @param {string} eventId - id of event to find repetition
   * @return {EventRepetitionSchema|null}
   */
  async getEventLastRepetition(eventId) {
    const repetitions = await this.getEventRepetitions(eventId, 1);

    if (repetitions.length === 0) {
      return null;
    }

    return repetitions.shift();
  }

  /**
   * Get a release from corresponding to this event
   *
   * @param {string} eventId - id of event to get the release
   * @returns {Release|null}
   */
  async getEventRelease(eventId) {
    const eventOriginal = await this.findById(eventId);

    if (!eventOriginal || !eventOriginal.payload.release) {
      return null;
    }

    const release = await mongo.databases.events.collection(this.TYPES.RELEASES).findOne({
      release: eventOriginal.payload.release,
      projectId: this.projectId.toString(),
    });

    return release;
  }

  /**
   * Mark event as visited for passed user
   *
   * @param {string|ObjectId} eventId - id of the original event
   * @param {string|ObjectId} userId - id of the user who is visiting the event
   *
   * @return {Promise<UpdateWriteOpResult>}
   */
  async visitEvent(eventId, userId) {
    const result = await this.getCollection(this.TYPES.EVENTS)
      .updateOne(
        { _id: new ObjectId(eventId) },
        { $addToSet: { visitedBy: new ObjectId(userId) } }
      );

    if (result.matchedCount === 0) {
      throw new Error(`Event not found for eventId: ${eventId}`);
    }

    return result;
  }

  /**
   * Mark many original events as visited for passed user
   *
   * @param {string[]} eventIds - original event ids
   * @param {string|ObjectId} userId - id of the user who is visiting events
   * @returns {Promise<UpdateWriteOpResult>}
   */
  async bulkVisitEvents(eventIds, userId) {
    const uniqueEventIds = [ ...new Set((eventIds || []).map(id => String(id))) ];
    const collection = this.getCollection(this.TYPES.EVENTS);
    const userObjectId = new ObjectId(userId);

    return collection.updateMany(
      {
        _id: { $in: uniqueEventIds.map(id => new ObjectId(id)) },
        visitedBy: { $ne: userObjectId },
      },
      { $addToSet: { visitedBy: userObjectId } }
    );
  }

  /**
   * Mark or unmark event as Resolved, Ignored or Starred
   *
   * @param {string|ObjectId} eventId - id of the original event to mark
   * @param {string} mark - mark label
   *
   * @return {Promise<UpdateWriteOpResult>}
   */
  async toggleEventMark(eventId, mark) {
    const collection = this.getCollection(this.TYPES.EVENTS);

    const event = await this.eventsDataLoader.load(eventId);

    if (!event) {
      throw new Error(`Event not found for eventId: ${eventId}`);
    }

    const query = { _id: new ObjectId(event._id) };

    const markKey = `marks.${mark}`;

    let update;

    if (event.marks && event.marks[mark]) {
      update = {
        $unset: { [markKey]: '' },
      };
    } else {
      update = {
        $set: { [markKey]: Math.floor(Date.now() / 1000) },
      };
    }

    return collection.updateOne(query, update);
  }

  /**
   * Bulk set or clear mark for original events.
   *
   * @param {string[]} eventIds - original event ids
   * @param {string} mark - 'resolved' | 'ignored' | 'starred'
   * @param {boolean} enabled - true to set mark, false to clear
   * @returns {Promise<UpdateWriteOpResult>}
   */
  async bulkSetEventMarks(eventIds, mark, enabled) {
    const uniqueEventIds = [ ...new Set((eventIds || []).map(id => String(id))) ];
    const objectIds = uniqueEventIds.map(id => new ObjectId(id));
    const collection = this.getCollection(this.TYPES.EVENTS);
    const nowSec = Math.floor(Date.now() / 1000);
    const markKey = `marks.${mark}`;

    if (!enabled) {
      return collection.updateMany(
        {
          _id: { $in: objectIds },
          [markKey]: { $exists: true },
        },
        { $unset: { [markKey]: '' } }
      );
    }

    return collection.updateMany(
      {
        _id: { $in: objectIds },
        [markKey]: { $exists: false },
      },
      { $set: { [markKey]: nowSec } }
    );
  }

  /**
   * Bulk set/clear assignee for many original events.
   *
   * @param {string[]} eventIds - original event ids
   * @param {string|null|undefined} assignee - target assignee id, null/undefined to clear
   * @returns {Promise<UpdateWriteOpResult>}
   */
  async bulkUpdateAssignee(eventIds, assignee) {
    const uniqueEventIds = [ ...new Set((eventIds || []).map(id => String(id))) ];
    const collection = this.getCollection(this.TYPES.EVENTS);
    const normalizedAssignee = assignee ? String(assignee) : '';

    return collection.updateMany(
      {
        _id: { $in: uniqueEventIds.map(id => new ObjectId(id)) },
        assignee: { $ne: normalizedAssignee },
      },
      { $set: { assignee: normalizedAssignee } }
    );
  }

  /**
   * Remove a single event and all related data (repetitions, daily events)
   *
   * @param {string|ObjectId} eventId - id of the original event to remove
   * @return {Promise<boolean>}
   */
  async removeEvent(eventId) {
    const eventsCollection = this.getCollection(this.TYPES.EVENTS);

    const event = await eventsCollection.findOne({ _id: new ObjectId(eventId) });

    // If event is not found, throw error
    if (!event) {
      throw new Error(`Event not found for eventId: ${eventId}`);
    }

    const { groupHash } = event;

    // Delete original event
    const result = await eventsCollection.deleteOne({ _id: new ObjectId(eventId) });

    // Delete all repetitions with same groupHash
    if (await this.isCollectionExists(this.TYPES.REPETITIONS)) {
      await this.getCollection(this.TYPES.REPETITIONS).deleteMany({ groupHash });
    }

    // Delete all daily event records with same groupHash
    if (await this.isCollectionExists(this.TYPES.DAILY_EVENTS)) {
      await this.getCollection(this.TYPES.DAILY_EVENTS).deleteMany({ groupHash });
    }

    return result.acknowledged && result.deletedCount > 0;
  }

  /**
   * Remove all project events
   *
   * @return {Promise<void>}
   */
  async remove() {
    /**
     * Check if collection is existing
     * Drop collection only when it's existing
     */
    if (await this.isCollectionExists(this.TYPES.EVENTS)) {
      await this.getCollection(this.TYPES.EVENTS).drop();
    }

    if (await this.isCollectionExists(this.TYPES.DAILY_EVENTS)) {
      await this.getCollection(this.TYPES.DAILY_EVENTS).drop();
    }

    if (await this.isCollectionExists(this.TYPES.REPETITIONS)) {
      await this.getCollection(this.TYPES.REPETITIONS).drop();
    }
  }

  /**
   * Update assignee to selected event
   *
   * @param {string} eventId - id of the original event to update
   * @param {string} assignee - assignee id for this event
   * @return {Promise<void>}
   */
  async updateAssignee(eventId, assignee) {
    const collection = this.getCollection(this.TYPES.EVENTS);

    const query = { _id: new ObjectId(eventId) };

    const update = {
      $set: { assignee: assignee },
    };

    const result = await collection.updateOne(query, update);

    if (result.updatedCount === 0) {
      throw new Error(`Event not found for eventId: ${eventId}`);
    }

    return result;
  }

  /**
   * Compose event with repetition
   *
   * @param {Event} event - event
   * @param {Repetition|null} repetition - repetition null
   * @returns {Event} event merged with repetition
   */
  _composeEventWithRepetition(event, repetition) {
    if (!repetition) {
      return {
        ...event,
        originalTimestamp: event.timestamp,
        originalEventId: event._id,
        projectId: this.projectId,
      };
    }

    return {
      ...event,
      _id: repetition._id,
      originalTimestamp: event.timestamp,
      originalEventId: event._id,
      timestamp: repetition.timestamp,
      payload: composeEventPayloadByRepetition(event.payload, repetition),
      projectId: this.projectId,
    };
  }
}

module.exports = EventsFactory;