googlephotos/lib/media_items/filters.js

Library to make working with the Google Photos API easy

'use strict';

const DateFilter = require('../common/date_filter');
const MediaTypeFilter = require('./media_type_filter');
const ContentFilter = require('./content_filter');

class Filters {
  constructor(includeArchivedMedia = false) {
    this.includeArchivedMedia = includeArchivedMedia;
  }

  setDateFilter(dateFilter) {
    this.dateFilter = dateFilter;
    return this;
  }

  setContentFilter(contentFilter) {
    this.contentFilter = contentFilter;
    return this;
  }

  setMediaTypeFilter(mediaTypeFilter) {
    this.mediaTypeFilter = mediaTypeFilter;
    return this;
  }

  setIncludeArchivedMedia(includeArchivedMedia) {
    this.includeArchivedMedia = includeArchivedMedia;
    return this;
  }

  toJSON() {
    return {
      dateFilter:
        this.dateFilter instanceof DateFilter ? this.dateFilter.toJSON() : this.dateFilter,
      mediaTypeFilter:
        this.mediaTypeFilter instanceof MediaTypeFilter
          ? this.mediaTypeFilter.toJSON()
          : this.mediaTypeFilter,
      contentFilter:
        this.contentFilter instanceof ContentFilter
          ? this.contentFilter.toJSON()
          : this.contentFilter,
      includeArchivedMedia: this.includeArchivedMedia,
    };
  }
}

module.exports = Filters;