path: root/src/internal-server/app/Controllers/Http/ServiceController.js

const Service = use('App/Models/Service');
const { validateAll } = use('Validator');
const Env = use('Env');

const { v4: uuid } = require('uuid');
const path = require('path');
const fs = require('fs-extra');
const { LOCAL_HOSTNAME, DEFAULT_SERVICE_ORDER } = require('../../../../config');
const { API_VERSION } = require('../../../../environment');

const hostname = LOCAL_HOSTNAME;
const port = Env.get('PORT');

class ServiceController {
  // Create a new service for user
  async create({ request, response }) {
    // Validate user input
    const validation = await validateAll(request.all(), {
      name: 'required|string',
      recipeId: 'required',
    });
    if (validation.fails()) {
      return response.status(401).send({
        message: 'Invalid POST arguments',
        messages: validation.messages(),
        status: 401,
      });
    }

    const data = request.all();

    // Get new, unused uuid
    let serviceId;
    do {
      serviceId = uuid();
    } while (
      (await Service.query().where('serviceId', serviceId).fetch()).rows
        .length > 0
    );

    await Service.create({
      serviceId,
      name: data.name,
      recipeId: data.recipeId,
      settings: JSON.stringify(data),
    });

    return response.send({
      data: {
        userId: 1,
        id: serviceId,
        isEnabled: true,
        isNotificationEnabled: true,
        isBadgeEnabled: true,
        isMuted: false,
        isDarkModeEnabled: '', // TODO: This should ideally be a boolean (false). But, changing it caused the sidebar toggle to not work.
        spellcheckerLanguage: '',
        order: DEFAULT_SERVICE_ORDER,
        customRecipe: false,
        hasCustomIcon: false,
        workspaces: [],
        iconUrl: null,
        ...data,
      },
      status: ['created'],
    });
  }

  // List all services a user has created
  async list({ response }) {
    const services = (await Service.all()).rows;
    // Convert to array with all data Franz wants
    const servicesArray = services.map(service => {
      const settings =
        typeof service.settings === 'string'
          ? JSON.parse(service.settings)
          : service.settings;

      return {
        customRecipe: false,
        hasCustomIcon: false,
        isBadgeEnabled: true,
        isDarkModeEnabled: '', // TODO: This should ideally be a boolean (false). But, changing it caused the sidebar toggle to not work.
        isEnabled: true,
        isMuted: false,
        isNotificationEnabled: true,
        order: DEFAULT_SERVICE_ORDER,
        spellcheckerLanguage: '',
        workspaces: [],
        ...JSON.parse(service.settings),
        iconUrl: settings.iconId
          ? `http://${hostname}:${port}/${API_VERSION}/icon/${settings.iconId}`
          : null,
        id: service.serviceId,
        name: service.name,
        recipeId: service.recipeId,
        userId: 1,
      };
    });

    return response.send(servicesArray);
  }

  async edit({ request, response, params }) {
    if (request.file('icon')) {
      // Upload custom service icon
      await fs.ensureDir(path.join(Env.get('USER_PATH'), 'icons'));

      const icon = request.file('icon', {
        types: ['image'],
        size: '2mb',
      });
      const { id } = params;
      const service = (await Service.query().where('serviceId', id).fetch())
        .rows[0];
      const settings =
        typeof service.settings === 'string'
          ? JSON.parse(service.settings)
          : service.settings;

      // Generate new icon ID
      let iconId;
      do {
        iconId = uuid() + uuid();
      } while (fs.existsSync(path.join(Env.get('USER_PATH'), 'icons', iconId)));

      await icon.move(path.join(Env.get('USER_PATH'), 'icons'), {
        name: iconId,
        overwrite: true,
      });

      if (!icon.moved()) {
        return response.status(500).send(icon.error());
      }

      const newSettings = {
        ...settings,
        iconId,
        customIconVersion:
          settings && settings.customIconVersion
            ? settings.customIconVersion + 1
            : 1,
      };

      // Update data in database
      await Service.query()
        .where('serviceId', id)
        .update({
          name: service.name,
          settings: JSON.stringify(newSettings),
        });

      return response.send({
        data: {
          id,
          name: service.name,
          ...newSettings,
          iconUrl: `http://${hostname}:${port}/${API_VERSION}/icon/${newSettings.iconId}`,
          userId: 1,
        },
        status: ['updated'],
      });
    }
    // Update service info
    const data = request.all();
    const { id } = params;

    // Get current settings from db
    const serviceData = (await Service.query().where('serviceId', id).fetch())
      .rows[0];

    const settings = {
      ...(typeof serviceData.settings === 'string'
        ? JSON.parse(serviceData.settings)
        : serviceData.settings),
      ...data,
    };

    // Update data in database
    await Service.query()
      .where('serviceId', id)
      .update({
        name: data.name,
        settings: JSON.stringify(settings),
      });

    // Get updated row
    const service = (await Service.query().where('serviceId', id).fetch())
      .rows[0];

    return response.send({
      data: {
        id,
        name: service.name,
        ...settings,
        iconUrl: `${Env.get('APP_URL')}/${API_VERSION}/icon/${settings.iconId}`,
        userId: 1,
      },
      status: ['updated'],
    });
  }

  async icon({ params, response }) {
    const { id } = params;

    const iconPath = path.join(Env.get('USER_PATH'), 'icons', id);
    if (!fs.existsSync(iconPath)) {
      return response.status(404).send({
        status: "Icon doesn't exist",
      });
    }

    return response.download(iconPath);
  }

  async reorder({ request, response }) {
    const data = request.all();

    for (const service of Object.keys(data)) {
      // Get current settings from db
      const serviceData = (
        await Service.query() // eslint-disable-line no-await-in-loop
          .where('serviceId', service)
          .fetch()
      ).rows[0];

      const settings = {
        ...JSON.parse(serviceData.settings),
        order: data[service],
      };

      // Update data in database
      await Service.query() // eslint-disable-line no-await-in-loop
        .where('serviceId', service)
        .update({
          settings: JSON.stringify(settings),
        });
    }

    // Get new services
    const services = (await Service.all()).rows;
    // Convert to array with all data Franz wants
    const servicesArray = services.map(service => {
      const settings =
        typeof service.settings === 'string'
          ? JSON.parse(service.settings)
          : service.settings;

      return {
        customRecipe: false,
        hasCustomIcon: false,
        isBadgeEnabled: true,
        isDarkModeEnabled: '', // TODO: This should ideally be a boolean (false). But, changing it caused the sidebar toggle to not work.
        isEnabled: true,
        isMuted: false,
        isNotificationEnabled: true,
        order: DEFAULT_SERVICE_ORDER,
        spellcheckerLanguage: '',
        workspaces: [],
        ...JSON.parse(service.settings),
        iconUrl: settings.iconId
          ? `http://${hostname}:${port}/${API_VERSION}/icon/${settings.iconId}`
          : null,
        id: service.serviceId,
        name: service.name,
        recipeId: service.recipeId,
        userId: 1,
      };
    });

    return response.send(servicesArray);
  }

  update({ response }) {
    return response.send([]);
  }

  async delete({ params, response }) {
    // Update data in database
    await Service.query().where('serviceId', params.id).delete();

    return response.send({
      message: 'Sucessfully deleted service',
      status: 200,
    });
  }
}

module.exports = ServiceController;