使imgur链接在url末尾有.png或.jpg与nodeJS

const http = require('http');
const url = require('url');

const server = http.createServer((req, res) => {
  const query = url.parse(req.url, true).query;
  const imgurLink = query.imgurLink;

  // 检查imgur链接是否有效
  if (!imgurLink) {
    res.statusCode = 400;
    res.end('Invalid imgur link');
    return;
  }

  // 检查链接是否以.png或.jpg结尾
  if (!imgurLink.endsWith('.png') && !imgurLink.endsWith('.jpg')) {
    res.statusCode = 400;
    res.end('Invalid image format');
    return;
  }

  // 在url末尾添加.png或.jpg
  const modifiedLink = imgurLink + '.png'; // 或者使用.jpg

  res.statusCode = 200;
  res.setHeader('Content-Type', 'text/plain');
  res.end(modifiedLink);
});

server.listen(3000, 'localhost', () => {
  console.log('Server running at http://localhost:3000/');
});