src-electron/rest/ide-api-handler.js

/**
 *
 *    Copyright (c) 2020 Silicon Labs
 *
 *    Licensed 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 CONDITIONS OF ANY KIND, either express or implied.
 *    See the License for the specific language governing permissions and
 *    limitations under the License.
 */

/**
 * This module provides the REST API to the IDE component handling.
 *
 * @module REST API: generation functions
 */

const env = require('../util/env')
const studio = require('../ide-integration/studio-rest-api')
const restApi = require('../../src-shared/rest-api.js')
const querySession = require('../db/query-session.js')
const { StatusCodes } = require('http-status-codes')

/**
 * Get component tree.
 *
 * @param {*} db
 * @returns Response data
 */
function httpGetComponentTree(db) {
  return async (req, res) => {
    try {
      let r = await studio.getProjectInfo(db, req.zapSessionId)
      res.status(StatusCodes.OK).send(r.data)
    } catch (err) {
      if (err.response) {
        res.send(err.response.data)
      } else {
        res.send(err.message)
      }
    }
  }
}

/**
 * Update component.
 *
 * @param {*} db
 * @param {*} request
 * @param {*} response
 * @param {*} add
 */
async function httpPostComponentUpdateHandler(db, request, response, add) {
  let { clusterId, side, componentIds } = request.body
  try {
    let res = await studio.updateComponentByClusterIdAndComponentId(
      db,
      request.zapSessionId,
      componentIds,
      clusterId,
      add,
      side
    )
    await studio.sendComponentUpdateStatus(db, request.zapSessionId, {
      data: res,
      added: add
    })
    response.send(res)
  } catch (err) {
    response.send(err)
  }
}

/**
 *  Enable components by 'componentId' or corresponding components specified, via 'defaults', by 'clusterId' / 'roles'
 *
 * @param {*} db
 */
function httpPostComponentAdd(db) {
  return (request, response) =>
    httpPostComponentUpdateHandler(db, request, response, true)
}

/**
 * Remove component.
 *
 * @param {*} db
 * @returns Promise of component removed
 */
function httpPostComponentRemove(db) {
  return (request, response) =>
    httpPostComponentUpdateHandler(db, request, response, false)
}

exports.get = [
  {
    uri: restApi.uc.componentTree,
    callback: httpGetComponentTree
  }
]

exports.post = [
  {
    uri: restApi.uc.componentAdd,
    callback: httpPostComponentAdd
  },
  {
    uri: restApi.uc.componentRemove,
    callback: httpPostComponentRemove
  }
]