mirror of
				https://github.com/zulip/zulip.git
				synced 2025-11-04 05:53:43 +00:00 
			
		
		
		
	And convert the corresponding function expressions to arrow style
while we’re here.
import * as babelParser from "recast/parsers/babel";
import * as recast from "recast";
import * as tsParser from "recast/parsers/typescript";
import { builders as b, namedTypes as n } from "ast-types";
import K from "ast-types/gen/kinds";
import fs from "fs";
import path from "path";
import process from "process";
const checkExpression = (node: n.Node): node is K.ExpressionKind =>
  n.Expression.check(node);
for (const file of process.argv.slice(2)) {
  console.log("Parsing", file);
  const ast = recast.parse(fs.readFileSync(file, { encoding: "utf8" }), {
    parser: path.extname(file) === ".ts" ? tsParser : babelParser,
  });
  let changed = false;
  recast.visit(ast, {
    visitCallExpression(path) {
      const { callee, arguments: args } = path.node;
      if (
        n.MemberExpression.check(callee) &&
        !callee.computed &&
        n.Identifier.check(callee.object) &&
        callee.object.name === "_" &&
        n.Identifier.check(callee.property) &&
        callee.property.name === "map" &&
        args.length === 2 &&
        checkExpression(args[0]) &&
        checkExpression(args[1])
      ) {
        const [arr, fn] = args;
        path.replace(
          b.callExpression(b.memberExpression(arr, b.identifier("map")), [
            n.FunctionExpression.check(fn) ||
            n.ArrowFunctionExpression.check(fn)
              ? b.arrowFunctionExpression(
                  fn.params,
                  n.BlockStatement.check(fn.body) &&
                    fn.body.body.length === 1 &&
                    n.ReturnStatement.check(fn.body.body[0])
                    ? fn.body.body[0].argument || b.identifier("undefined")
                    : fn.body
                )
              : fn,
          ])
        );
        changed = true;
      }
      this.traverse(path);
    },
  });
  if (changed) {
    console.log("Writing", file);
    fs.writeFileSync(file, recast.print(ast).code, { encoding: "utf8" });
  }
}
Signed-off-by: Anders Kaseorg <anders@zulipchat.com>
		
	
		
			
				
	
	
		
			92 lines
		
	
	
		
			2.7 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
			
		
		
	
	
			92 lines
		
	
	
		
			2.7 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
const IntDict = require('./int_dict').IntDict;
 | 
						|
 | 
						|
const bots = new IntDict();
 | 
						|
const bot_fields = ['api_key', 'avatar_url', 'default_all_public_streams',
 | 
						|
                    'default_events_register_stream', 'default_sending_stream',
 | 
						|
                    'email', 'full_name', 'is_active', 'owner', 'bot_type', 'user_id'];
 | 
						|
const services = new IntDict();
 | 
						|
const services_fields = ['base_url', 'interface',
 | 
						|
                         'config_data', 'service_name', 'token'];
 | 
						|
 | 
						|
const send_change_event = _.debounce(function () {
 | 
						|
    settings_bots.render_bots();
 | 
						|
}, 50);
 | 
						|
 | 
						|
const set_can_admin = function bot_data__set_can_admin(bot) {
 | 
						|
    if (page_params.is_admin) {
 | 
						|
        bot.can_admin = true;
 | 
						|
    } else if (bot.owner !== undefined && people.is_current_user(bot.owner)) {
 | 
						|
        bot.can_admin = true;
 | 
						|
    } else {
 | 
						|
        bot.can_admin = false;
 | 
						|
    }
 | 
						|
};
 | 
						|
 | 
						|
exports.add = function bot_data__add(bot) {
 | 
						|
    const clean_bot = _.pick(bot, bot_fields);
 | 
						|
    bots.set(bot.user_id, clean_bot);
 | 
						|
    set_can_admin(clean_bot);
 | 
						|
    const clean_services = bot.services.map(service => _.pick(service, services_fields));
 | 
						|
    services.set(bot.user_id, clean_services);
 | 
						|
 | 
						|
    send_change_event();
 | 
						|
};
 | 
						|
 | 
						|
exports.deactivate = function bot_data__deactivate(bot_id) {
 | 
						|
    bots.get(bot_id).is_active = false;
 | 
						|
    send_change_event();
 | 
						|
};
 | 
						|
 | 
						|
exports.del = function bot_data__del(bot_id) {
 | 
						|
    bots.delete(bot_id);
 | 
						|
    services.delete(bot_id);
 | 
						|
    send_change_event();
 | 
						|
};
 | 
						|
 | 
						|
exports.update = function bot_data__update(bot_id, bot_update) {
 | 
						|
    const bot = bots.get(bot_id);
 | 
						|
    _.extend(bot, _.pick(bot_update, bot_fields));
 | 
						|
    set_can_admin(bot);
 | 
						|
 | 
						|
    // We currently only support one service per bot.
 | 
						|
    const service = services.get(bot_id)[0];
 | 
						|
    if (typeof bot_update.services !== 'undefined' && bot_update.services.length > 0) {
 | 
						|
        _.extend(service, _.pick(bot_update.services[0], services_fields));
 | 
						|
    }
 | 
						|
    send_change_event();
 | 
						|
};
 | 
						|
 | 
						|
exports.get_all_bots_for_current_user = function bots_data__get_editable() {
 | 
						|
    return bots.filter_values(function (bot) {
 | 
						|
        return people.is_current_user(bot.owner);
 | 
						|
    });
 | 
						|
};
 | 
						|
 | 
						|
exports.get_editable = function bots_data__get_editable() {
 | 
						|
    return bots.filter_values(function (bot) {
 | 
						|
        return bot.is_active && people.is_current_user(bot.owner);
 | 
						|
    });
 | 
						|
};
 | 
						|
 | 
						|
exports.get = function bot_data__get(bot_id) {
 | 
						|
    return bots.get(bot_id);
 | 
						|
};
 | 
						|
 | 
						|
exports.get_bot_owner_email = function (bot_id) {
 | 
						|
    return bots.get(bot_id).owner;
 | 
						|
};
 | 
						|
 | 
						|
exports.get_services = function bot_data__get_services(bot_id) {
 | 
						|
    return services.get(bot_id);
 | 
						|
};
 | 
						|
 | 
						|
exports.initialize = function () {
 | 
						|
    for (const bot of page_params.realm_bots) {
 | 
						|
        exports.add(bot);
 | 
						|
    }
 | 
						|
 | 
						|
    delete page_params.realm_bots;
 | 
						|
};
 | 
						|
 | 
						|
window.bot_data = exports;
 |