使用 AirCode 开发飞书机器人

const lark = require("@larksuiteoapi/node-sdk");

module.exports = new lark.Client({
  appId: process.env.app_id,
  appSecret: process.env.app_secret,
  appType: lark.AppType.SelfBuild,
});

const aircode = require("aircode");
const axios = require("axios");
const querystring = require("querystring");
const client = require("./client");

module.exports = async ({ url }) => {
  // 如果已经上传过了，从数据库中取出缓存
  const ImagesTable = aircode.db.table("images");

  // 查询数据库
  const result = await ImagesTable.where({ url }).find();

  if (result.length > 0) {
    return result[0].imageKey;
  }

  // 添加 param=200y200 的参数，使用小尺寸图片
  const splitUrl = url.split("?");
  const querys = querystring.parse(splitUrl[1] || "");
  const queryString = querystring.stringify({ ...querys, param: "200y200" });
  const newUrl = `${splitUrl[0]}?${queryString}`;

  // 上传图片到飞书中
  const response = await axios.get(newUrl, { responseType: "arraybuffer" });
  const { image_key } = await client.im.image.create({
    data: {
      image_type: "message",
      image: Buffer.from(response.data, "utf-8"),
    },
  });

  // 保存到数据库中
  ImagesTable.save({
    url: newUrl,
    imageKey: image_key,
  });

  return image_key;
};

const axios = require("axios");
const imageUpload = require("./image");

const request = axios.create({
  baseURL: process.env.music_api,
});

// 搜索歌曲
module.exports = async function (params) {
  const { keyword } = params;

  const result = await request.get(`/search?keywords=${keyword}&type=1018`);
  // 取前三首
  const songs = result.data.result.song.songs.slice(0, 3);

  const _list = await songs.map((data) => {
    const artistNames = data.ar.length
      ? [...data.ar]
          .reverse()
          .reduce((ac, a) => `${a.name} ${ac}`, "")
          .trim()
      : "";

    return {
      imgUrl: data.al.picUrl,
      title: `${data.name}`,
      desc: `${artistNames} · ${data.al.name}`,
      artistId: data.ar[0].id,
      albumId: data.al.id,
      artistName: artistNames,
      albumName: data.al.name,
      id: data.id,
    };
  });

  const imageKeys = await Promise.all(
    _list.map((data) => imageUpload({ url: data.imgUrl }))
  );
  const list = _list.map((data, index) => ({
    ...data,
    imageKey: imageKeys[index],
  }));

  return list;
};

// 生成消息卡片

const generatePath = (id) => {
  return `https://applink.feishu.cn/client/web_app/open?appId=${process.env.app_id}&mode=sidebar&id=${id}`;
};

const generateHeader = (keyword) => {
  return {
    template: "turquoise",
    title: {
      content: `🎵 ${keyword}`,
      tag: "plain_text",
    },
  };
};

const generateCard = (data) => {
  return {
    tag: "div",
    text: {
      tag: "lark_md",
      content: `**${data.title}**\n${data.desc}\n[打开播放器](${generatePath(
        data.id
      )})`,
    },
    extra: {
      tag: "img",
      img_key: data.imageKey,
      alt: {
        tag: "plain_text",
        content: data.title,
      },
    },
  };
};

module.exports = ({ keyword, list }) => {
  console.log("search keyword: ", keyword);
  console.log("search result: ", list);

  const header = generateHeader(keyword);

  const elements = list.reduce((result, item, index) => {
    const card = generateCard(item);
    if (index > 0) {
      result.push({ tag: "hr" });
    }
    return [...result, card];
  }, []);

  return {
    config: {
      wide_screen_mode: true,
    },
    elements,
    header,
  };
};

// @see https://docs.aircode.io/guide/functions/
const aircode = require("aircode");
const lark = require("@larksuiteoapi/node-sdk");

const client = require("./client");
const search = require("./search");
const reaction = require("./reaction");

const sendMessage = (chatId, content, type = "text") => {
  return client.im.message.create({
    params: {
      receive_id_type: "chat_id",
    },
    data: {
      receive_id: chatId,
      content: JSON.stringify(content),
      msg_type: type,
    },
  });
};

const eventDispatcher = new lark.EventDispatcher({}).register({
  "im.message.receive_v1": async (data) => {
    const open_chat_id = data.message.chat_id;
    const msg = JSON.parse(data.message.content).text;
    const keyword = msg.replace(/^@\w+/, "").trim();

    let res;

    if (!keyword) {
      res = await sendMessage(open_chat_id, {
        text: "请告诉我一些关键词，比如歌名，歌手名，歌单名，用户名等~",
      });
    } else {
      const list = await search({ keyword });
      const interactive = reaction({ keyword, list });
      res = await sendMessage(open_chat_id, interactive, "interactive");
    }

    return res;
  },
});

module.exports = async function (params, context) {
  console.log("Received params:", params);

  // url 验证
  if (params.type === "url_verification") {
    return { challenge: params.challenge };
  }

  // 根据 message_id 去重复消息
  const message_id = params.event.message.message_id;
  const LogsTable = aircode.db.table("logs");
  const findResult = await LogsTable.where({ message_id }).find();
  if (findResult.length > 0) return `重复消息 - ${message_id}`;
  LogsTable.save({ message_id });

  return await eventDispatcher.invoke(params);
};