server.js

const yapi = require('yapi.js');
const mongoose = require('mongoose');
const controller = require('./controller');
const InterfaceNotificationSender = require('./utils/interfaceNotificationSender');
const InterfaceModel = require('models/interface');
const Config = require('./utils/config');
const { SendLogViaDingDingSender } = require('./utils/logSender');

module.exports = function(options) {
  Config.instance = options;

  const originalSaveLog = this.commons.saveLog;

  this.commons.saveLog = function() {
    const args = Array.prototype.slice.call(arguments);
    originalSaveLog.apply(this, args);
    try {
      yapi.commons.log('yapi-plugin-dingding: 开始运行');
      const logData = args[0];
      if (!logData || logData.type != 'project') {
        yapi.commons.log('yapi-plugin-dingding: 日志不是 project 类型，跳过通知。');
        return;
      }
      (new SendLogViaDingDingSender(logData)).send().then().catch((err) => {
        yapi.commons.log(err, 'error');
      });
    } catch(err) {
      yapi.commons.log(err, 'error');
    }
  }

  yapi.connect.then(function() {
    let db = mongoose.connection.db.collection('dding_robots');
    db.createIndex({
      project_id: 1
    });
  });

  this.bindHook('add_router', function(router) {
    router({
      controller: controller,
      method: 'get',
      path: 'dding_robots/detail',
      action: 'show'
    });

    router({
      controller: controller,
      method: 'post',
      path: 'dding_robots/up',
      action: 'update'
    });

    router({
      controller: controller,
      method: 'post',
      path: 'dding_robots/test',
      action: 'test'
    });
  });
}