// hexo-wx-sync.js 自动将hexo博客中的原创文章同步到微信公众号

const axios = require('axios');
const fs = require('fs');
const path = require('path');
const cheerio = require('cheerio');
var req = require('request-promise');
const { marked } = require('marked')
const juice = require('juice')


//1. 获取文章html 将内容中带有'本文由作者原创'的文章过滤出来 排除wx_pushed.txt中已经同步的文章
//2. 替换文章中的图片
//3. 替换内链样式
//4. 增加封面图
//5. 推送文章到公众号
//6. 记录已经推送的文章名称到wx_pushed.txt中

const PLUGIN_NAME = 'hexo-wx-sync';
const config = hexo.config.wx_sync
const parentDir = path.dirname(__dirname);//当前文件的父目录 也就是hexo
const PUSHED_ARTICLES_FILE = `${parentDir}/${PLUGIN_NAME}/wx_pushed.txt`;

let accessToken = ""

let pushedArticles = [];
// 加载已发布的文章列表
function loadPushedArticles() {
  if (fs.existsSync(PUSHED_ARTICLES_FILE)) {
    pushedArticles = fs.readFileSync(PUSHED_ARTICLES_FILE, 'utf8').split('\n').filter(Boolean);
  }
}

// 保存已发布的文章列表
function savePushedArticles() {
  fs.writeFileSync(PUSHED_ARTICLES_FILE, pushedArticles.join('\n'));
}





function filterPost(post) {
  loadPushedArticles()
  // 获取文章路径
  const postPath = parentDir + "/source/" + post.source;
  // 读取markdown文章内容
  const mdContent = fs.readFileSync(postPath, 'utf8');
  // 选择还未上传的原创文章
  if (!pushedArticles.includes(post.source) && mdContent.includes("本文为作者原创 转载时请注明出处")) {
    //将markdown转成html格式
    return marked(removeMetaData(mdContent,post))
  } else {
    return null
  }
}

function removeMetaData(content,post) {
  // 删除标题
  content = content.replace(/^title:.*$/m, "");

  // 删除日期
  content = content.replace(/^date:.*$/m,`原文地址: ${getPostSourceUrl(post)}<br>如果本文有相关附件下载 请将上面地址复制到浏览器打开` );

  // 删除标签
  content = content.replace(/^tags:.*$/m, '');

  // 删除空行
  content = content.replace(/^\s*$/gm, '');
  


  return content;
}

function publishToWechat(parsedContent, post, coverImageId) {
  // 发布文章: 'https://api.weixin.qq.com/cgi-bin/material/add_news'

  axios.post(`https://api.weixin.qq.com/cgi-bin/draft/add?access_token=${accessToken}`, {
    articles: [{
      title: `${post.title} `,
      thumb_media_id: coverImageId, // 刚才取到的封面图素材 id
      author: hexo.config.author,
      digest: getDigest(post.content),//文章摘要
      content: parsedContent, // 刚才处理好的文章
      content_source_url: getPostSourceUrl(post), // 非必填，这里我写的是我博客这篇文章的 URL
      need_open_comment: 1, // 是否打开留言功能
      show_cover_pic: 0, // 是否把封面图添加到文章开头
    }],
  }).then(response => {

    //推送成功会返回一个media_id 
    da = response.data
    if ('media_id' in da) {
      console.log(`${post.title} successfully pushed to WeChat.`);
      // 记录已推送的文章名称到 wx_pushed.txt 文件中
      // fs.appendFileSync(PUSHED_ARTICLES_FILE, `${post.source}\n`);
      pushedArticles.push(post.source);
    } else {
      console.log(da)
    }

  }).catch(error => {
    console.error(`Failed to push ${post.title} to WeChat: ${error}`);
  });;
}
function getDigest(content) {
  // 获取文章的摘要信息
  const $ = cheerio.load(content);
  const summary = $('p').first().text();
  return summary
}
function getPostSourceUrl(post) {
  // 这里我们使用网站的地址+文件名.html
  const domain = hexo.config.url
  const urlRegex = /(\d+)[.]md/g;
  const match = urlRegex.exec(post.source);
  const suffix = match ? match[1] : ''
  return domain + "/" + suffix + ".html"
}
// 替换文章中的图片
async function replaceImages(content) {
  console.log("开始替换图片")
  const $ = cheerio.load(content);
  // 获取所有的img元素
  const imgs = $('img');
  // 使用for循环遍历所有的img元素
  for (let i = 0; i < imgs.length; i++) {
    const newUrl = await uploadImg($(imgs[i]).attr('src'));
    // 修改img标签的src属性
    $(imgs[i]).attr('src', newUrl);
  }
  console.log("图片替换完成")
  return $.html();
}


async function request(option) {
  return await req(option).then((response) => {
    console.log("请求成功: " + response);
    return response
  }).catch( async (err) => {
    console.log("请求错误: " + err);
    //如果请求失败 重新请求
    return  await request(option)
    return '{"media_id":"fTLNXU-IBCWKkfWOlCRS17gXwNG5_75-7aBJQalLz1BBBE5UJ43cj3JfxCyYiTN-","url":""}'
  });
}



// 上传图片到素材库并获取新的图片地址
async function uploadImg(url) {
  console.log("需要上传的图片地址:" + url)
  const imageStream = (await axios.get(url, { responseType: 'stream' })).data;
  const formData = { media: imageStream };
  const res = await request({
    url: `https://api.weixin.qq.com/cgi-bin/media/uploadimg?access_token=${accessToken}`,
    method: 'POST',
    formData,
  });
  return JSON.parse(res).url;
}

// 将内联样式改为行内样式 否则微信平台不识别
function replaceInternalLinks(content) {
  //实现内链样式替换逻辑
  const themeStr = fs.readFileSync(`${parentDir}/${PLUGIN_NAME}/html-component/theme6.css`, 'utf-8')
  let parsedContent = juice(`
  <!--插入主题样式-->
  <style>
  ${themeStr}
  </style>
  <!--插入marked转换后的html-->
  <div class="markdown-body">
  ${content}
  `)
  //后面这个div可以不用加 否则会多出一个

  return removeWhitespaceExceptCodeBlocks(parsedContent);
}
//移除代码块以外的回车换行符
function removeWhitespaceExceptCodeBlocks(parsedContent) {
  const $ = cheerio.load(parsedContent);
    //预先存储未替换空格的pre标签
    const origin = $('pre')
    // 删除回车换行
    parsedContent = parsedContent.replace(/[\r\n]+/g, '')
    const $2 = cheerio.load(parsedContent);
    const after = $2('pre')
    for (let i = 0; i < after.length; i++) {
        // 进行标签替换
         $(after[i]).replaceWith($(origin[i]));
    }
    return $2.html();
}

function addCoverImage() {
  const data = fs.readFileSync(`${parentDir}/${PLUGIN_NAME}/coverJson.json`, 'utf8');
  const jsonData = JSON.parse(data);
  coverList = jsonData["images"]
  //随机获取封面图media_id
  const randomIndex = Math.floor(Math.random() * coverList.length);
  const randomMediaId = coverList[randomIndex]["media_id"];
  return randomMediaId
}
async function getAccessToken() {
  const res = await request({
    url: `https://api.weixin.qq.com/cgi-bin/token?grant_type=client_credential&appid=${config.appid}&secret=${config.secret}`,
    method: 'GET',
  })
  return JSON.parse(res).access_token
}



// 在执行 hexo d 时触发
hexo.extend.generator.register('wechat', async function (locals) {
  if (!config.enable) {
    console.log('关闭hexo-wx-sync插件');
    return;
  }
  //获取微信公众号accesToken
  accessToken = await getAccessToken()
  console.log("accessToken" + accessToken)
  // 遍历所有文章
  locals.posts.forEach(async post => {//post.content 获取的是html格式内容
    //过滤出未上传且为原创的文章
    const content = filterPost(post)
    if (content == null) {
      return
    }
    // 替换文章中的图片
    // const replacedContent = await replaceContentImage(content);
    const replacedContent = await replaceImages(content);
    //将样式注入到标签中
    const finalContent = replaceInternalLinks(replacedContent);
    // 增加封面图
    const coverImageId = addCoverImage();
    console.log("finalContent" + finalContent)
    // 调用微信接口发布文章
    publishToWechat(finalContent, post, coverImageId);
  });
});

module.exports = (redisConfig) =>async (ctx, next) => {
  const client = redis.createClient(redisConfig.port, redisConfig.host);
  client.auth(redisConfig.password);
  ctx.redis = {
    get: (key) => new Promise((resolve, reject) => {
      client.get(key, (err, result) => {
        if (err) {
          reject(err);
        } else {
          resolve(result);
        }
      });
    }),
    set: (key, value) => new Promise((resolve, reject) => {
      client.set(key, value, (err, result) => {
        if (err) {
          reject(err);
        } else {
          resolve(result);
        }
      });
    }),
    expire: (key, expire) => new Promise((resolve, reject) => {
      client.expire(key, expire, (err, result) => {
        if (err) {
          reject(err);
        } else {
          resolve(result);
        }
      });
    }),
    del: (key) => new Promise((resolve, reject) => {
      client.del(key, (err, result) => {
        if (err) {
          reject(err);
        } else {
          resolve(result);
        }
      });
    }),
  };
  await next();
};

module.exports = async (ctx) => {
  try {
    let accessToken = await ctx.redis.get('wechatAccessToken');
    if (!accessToken) {
      const res = await axios.get(`https://api.weixin.qq.com/cgi-bin/token?grant_type=client_credential&appid=${wechatConfig.appid}&secret=${wechatConfig.secret}`);
      if (res?.status !== 200) {
        throw new Error('get access token error');
      }
      accessToken = res?.data?.access_token;
      const expiresIn = res?.data?.expires_in;
      if (!accessToken) {
        throw new Error('get access token error');
      } else {
        await ctx.redis.set('wechatAccessToken', accessToken);
        await ctx.redis.expire('wechatAccessToken', expiresIn);
      }
    }
    return Promise.resolve(accessToken);
  } catch (err) {
    return Promise.reject(err);
  }
};

const images = parsedContent.match(/<img.*?(?:>|\/>)/gi);
if (images) {
  for (const image of images) {
    const src = image.match(/src=['"]?([^'"]*)['"]?/i);
    const url = src?.[1];
    // 对取出的 URL 做处理
  }
}

if (url) {
  const imageStream = (await axios.get(url, { responseType: 'stream' })).data;
  const formData = { media: imageStream };
  const res = await request({
    url: `https://api.weixin.qq.com/cgi-bin/media/uploadimg?access_token=${accessToken}`,
    method: 'POST',
    formData,
  });
  parsedContent = parsedContent.replace(url, JSON.parse(res).url);
}

const accessToken = await getWechatAccessToken(ctx);

const coverImageStream = (await axios.get(coverImage, { responseType: 'stream' })).data;
const coverImageRes = await request({
  url: `https://api.weixin.qq.com/cgi-bin/material/add_material?access_token=${accessToken}&type=image`,
  method: 'POST',
  formData: { media: coverImageStream },
});
const coverImageId = JSON.parse(coverImageRes).media_id;

parsedContent = juice(`
  <style>
    ${cssString}
  </style>
  <div>
    ${parsedContent}
  </div>
`);

await axios.post(`https://api.weixin.qq.com/cgi-bin/draft/add?access_token=${accessToken}`, {
  articles: [{
    title: '你的文章标题',
    thumb_media_id: coverImageId, // 刚才取到的封面图素材 id
    author: '文章作者',
    digest: '文章摘要',
    content: parsedContent, // 刚才处理好的文章
    content_source_url: '原文链接', // 非必填，这里我写的是我博客这篇文章的 URL
    need_open_comment: 1, // 是否打开留言功能
    show_cover_pic: 0, // 是否把封面图添加到文章开头
  }],
});

'use strict';

const fs = require('fs');
const path = require('path');
const cheerio = require('cheerio');
const crypto = require('crypto');
const { log } = require('console');

const DEFAULT_EXCLUDE_DOMAINS = ['example.com']; // 默认排除的域名列表

hexo.extend.filter.register('before_generate', function () {
    const config = hexo.config.external_links || {};
    const excludeDomains = config.exclude_domains || DEFAULT_EXCLUDE_DOMAINS;
    if (!config.enable) {
        console.log('关闭external_links插件');
        return;
      }
    hexo.extend.generator.register('external_links', function (locals) {
        const posts = locals.posts;
        const author = hexo.config.author;
        const outputDir = hexo.config.external_links_output || 'external_links';
        const linkMap = new Map(); // 存储链接和其对应的哈希值的映射

        posts.forEach(post => {
            const $ = cheerio.load(post.content);
            $('a').each(function () {
                const href = filterUrl($(this).attr('href'));
                const hostname = getHostname(href);

                if (hostname!=""&&!excludeDomains.includes(hostname)) {
                    const hash = generateHash(href);
                    linkMap.set(href, hash); // 存储链接和哈希值的映射关系

                    //替换文章中的原始链接
                    const newHref = `/${outputDir}/${hash}.html`;
                    $(this).attr('href', newHref);
                    $(this).attr('target', "blank");//在新标签页中打开超链接
                }
            });

            post.content = $.html();
        });

        const outputPath = path.join(hexo.public_dir, outputDir);
        if (!fs.existsSync(outputPath)) {
            fs.mkdirSync(outputPath);
        }

        // 生成每个链接对应的页面
        linkMap.forEach((hash, href) => {
            const htmlContent = generateLinkPage(href, hash,author);
            const outputFile = path.join(outputPath, `${hash}.html`);
            fs.writeFileSync(outputFile, htmlContent);
        });

   
    });
});

function generateLinkPage(href, hash,author) {
    const html = `
        <!DOCTYPE html>
        <html lang="en">
        <head>
            <meta charset="UTF-8">
            <meta name="viewport" content="width=device-width, initial-scale=1.0">
            <title>访问外部网站-{author}</title>
            <style>
                body {
                    margin: 20px;
                    padding-top: 100px;
                    color: #222;
                    font-size: 13px;
                    font-family: 'Helvetica Neue', Helvetica, Arial, sans-serif;
                    line-height: 1.5;
                    -webkit-tap-highlight-color: rgba(0, 0, 0, 0);
                }
                .wrapper {
                    margin: auto;
                    padding-left: 30px;
                    padding-right: 30px;
                    max-width: 540px;
 									  padding-top: 25px;
                    padding-bottom: 25px;
                    background-color: #f7f7f7;
                    border: 1px solid #babbbc;
                    border-radius: 5px;
                }
             
              
        
                .button {
                    padding: 0;
                    font-family: inherit;
                    background: none;
                    border: none;
                    outline: none;
                    cursor: pointer;
                }
                .button:hover {
                    background-color: #0070cd;
                }
                a{
                    text-decoration: none
                }
        
                .button:active {
                    background-color: #0077d9;
                }
        
          
                .link {
                    margin-bottom: 10px;
                }
              
                .button {
                    display: inline-block;
                    padding: 10px 16px;
                    color: #fff;
                    font-size: 14px;
                    line-height: 1;
                    background-color: #0077d9;
                    border-radius: 3px;
                }
                .actions {
                    margin-top: 15px;
                    padding-top: 30px;
                    text-align: right;
                    border-top: 1px solid #d8d8d8;
                }
            </style>
        </head>
        <body>
        
            <div class="wrapper">
                <div class="content">
                    <h1>即将离开${author}</h1>
                    <p class="info">您即将离开${author}，前往外部网站。</p>
                    <p class="link">${href}</p>
                </div>
                <div class="actions">
                    <a class="button" href="${href}" one-link-mark="yes">继续访问</a>
                </div>
            </div>
        </body>
        </html>
    `;
    return html;
}

function getHostname(url) {
    try {
        return new URL(url).hostname;
    } catch (error) {
        return '';
    }
}
function filterUrl(url) {
    const urlRegex = /http[s]?:\/\/[\u4e00-\u9fa5\w.-\/:]+[\u4e00-\u9fa5\w.?&\/=-]+/g;
    const match = urlRegex.exec(url);
    return match ? match[0] : ''
}

function generateHash(data) {
    return crypto.createHash('md5').update(data).digest('hex');
}

package.json{
  "name": "hexo-my-plugin",
  "version": "0.0.1",
  "main": "index"
}