ligh7hau5/commands/fediverse/media.js

const qs = require('qs');
const axios = require('axios');
const FormData = require('form-data');

const emojis = { public: '🌐', unlisted: '📝', private: '🔒️', direct: '✉️' };
exports.visibilityEmoji = v => emojis[v] || v;

const getFilename = header => {
  if(typeof header !== 'string') return null;
  try {
    const m = header.match(/inline; filename(?:=(.+)|\*=utf-8''(.+))/);
    return !m ? null : m[2] && decodeURIComponent(m[2]) || m[1];
  } catch(e) {
    return null;
  }
};

const mediaDownload = async (url, { whitelist, blacklist }) => {
  const media = await axios({ method: 'GET', url, responseType: 'arraybuffer' });
  if (media.statusText !== 'OK' || blacklist.includes(media.headers['content-type'])) throw media;
  if(whitelist.length && !whitelist.includes(media.headers['content-type'])) throw media;
  return {
    data: media.data,
    filename: getFilename(media.headers['content-disposition']),
    mimetype: media.headers['content-type']
  };
};

const mediaUpload = async ({ domain, token }, { data, filename, mimetype }) => {
  const form = new FormData();
  form.append('file', data, {
    filename: filename || 'upload',
    contentType: mimetype,
  });
  const upload = await axios({
    method: 'POST',
    url: `${domain}/api/v1/media`,
    headers: form.getHeaders({ Authorization: `Bearer ${token}` }),
    data: form,
  });
  if(upload.statusText !== 'OK') throw upload;
  return upload.data.id;
};

const run = async (matrixClient, { roomId }, content, replyId, mediaURL, subject, registrar) => {
  let mediaId = null;
  const fediverse = registrar.config.fediverse;
  if(mediaURL) {
    const media = await mediaDownload(mediaURL, registrar.config.matrix.mimetypes);
    mediaId = await mediaUpload(fediverse, media);
  }
  const response = await axios({
    method: 'POST',
    url: `${fediverse.domain}/api/v1/statuses`,
    headers: { Authorization: `Bearer ${fediverse.token}`, 'Content-Type': 'application/x-www-form-urlencoded' },
    data : qs.stringify({
      status: content,
      content_type: `text/markdown`,
      media_ids: mediaURL && [ mediaId ] || undefined,
      in_reply_to_id: replyId || undefined,
      spoiler_text: subject || undefined
    }, { arrayFormat: 'brackets' })
  });
  return matrixClient.sendHtmlNotice(roomId, '', `<a href="${response.data.url}">${response.data.id}</a>`);
}

exports.runQuery = async (client, room, userInput, registrar, { isReply, hasMedia, hasSubject }) => {
  try {
    const chunks = userInput.trim().split(' ');
    if(!chunks.length || chunks.length < !!isReply + !!hasMedia) throw '';
    let replyId = null;
    let mediaURL = null;
    const subject = hasSubject ? registrar.config.matrix.subject : null;
    if(isReply) {
      replyId = chunks[0];
      chunks.shift();
    }
    if(hasMedia) {
      let url = new URL(chunks[0]);
      chunks.shift();
      if(url.protocol === 'mxc:' && url.hostname && url.pathname)
        url = new URL(`${registrar.config.matrix.domain}/_matrix/media/r0/download/${url.hostname}${url.pathname}`);
      if(url.protocol !== 'https:') throw '';
      if(!registrar.config.matrix.domains.includes(url.hostname)) throw '';
      if(!/^\/_matrix\/media\/r0\/download\/[^/]+\/[^/]+\/?$/.test(url.pathname)) throw '';
      mediaURL = url.toString();
    }
    return await run(client, room, chunks.join(' '), replyId, mediaURL, subject, registrar);
  } catch(e) {
    return client.sendHtmlNotice(room.roomId, 'Sad!', `<strong>Sad!</strong>`).catch(()=>{});
  }
};