// ==UserScript==
// @name         A Day With ChatGPT
// @namespace    https://github.com/mefengl
// @version      0.2.3
// @description  A wonderful day spent with ChatGPT
// @icon         https://www.google.com/s2/favicons?sz=64&domain=openai.com
// @author       mefengl
// @match        https://book.douban.com/*
// @match        https://www.zhihu.com/*
// @match        https://github.com/*
// @match        https://chat.openai.com/chat
// @require      https://cdn.staticfile.org/jquery/3.6.1/jquery.min.js
// @grant        GM_openInTab
// @grant        GM_registerMenuCommand
// @grant        GM_unregisterMenuCommand
// @grant        GM_getValue
// @grant        GM_setValue
// @grant        GM_addValueChangeListener
// @license MIT
// @downloadURL none
// ==/UserScript==

(function () {
  'use strict';

  const default_menu_all = {
  };
  const menu_all = GM_getValue("menu_all", default_menu_all);
  // 菜单更新的逻辑
  // 使用 douban 开启
  $(() => location.href.includes("book.douban") && GM_setValue("douban-book", true) && console.log("开启 douban读书 菜单"));
  if (GM_getValue("douban-book") == true) {
    default_menu_all.douban_book = false;
  }
  // 使用 zhihu 开启
  $(() => location.href.includes("zhihu") && GM_setValue("zhihu", true) && console.log("开启 zhihu 菜单"));
  if (GM_getValue("zhihu") == true) {
    default_menu_all.zhihu = false;
  }
  // 使用 github 开启
  $(() => location.href.includes("github") && GM_setValue("github", true) && console.log("开启 github 菜单"));
  if (GM_getValue("github") == true) {
    default_menu_all.github = false;
  }
  // 检查是否有新增菜单
  for (let name in default_menu_all) {
    console.log(name);
    if (!(name in menu_all)) {
      menu_all[name] = default_menu_all[name];
    }
  }
  const menu_id = GM_getValue("menu_id", {});
  function update_menu() {
    for (let name in menu_all) {
      const value = menu_all[name];
      // 卸载原来的
      if (menu_id[name]) {
        GM_unregisterMenuCommand(menu_id[name]);
      }
      switch (name) {
        case "douban_book":
          // 添加新的
          menu_id[name] = GM_registerMenuCommand(
            " douban读书：" + (value ? "✅" : "❌"),
            () => {
              menu_all[name] = !menu_all[name];
              GM_setValue("menu_all", menu_all);
              // 调用时触发，刷新菜单
              update_menu();
              // 该设置需刷新生效
              location.reload();
            }
          );
          break;
        case "zhihu":
          // 添加新的
          menu_id[name] = GM_registerMenuCommand(
            " zhihu：" + (value ? "✅" : "❌"),
            () => {
              menu_all[name] = !menu_all[name];
              GM_setValue("menu_all", menu_all);
              // 调用时触发，刷新菜单
              update_menu();
              // 该设置需刷新生效
              location.reload();
            }
          );
          break;
        case "github":
          // 添加新的
          menu_id[name] = GM_registerMenuCommand(
            " github：" + (value ? "✅" : "❌"),
            () => {
              menu_all[name] = !menu_all[name];
              GM_setValue("menu_all", menu_all);
              // 调用时触发，刷新菜单
              update_menu();
              // 该设置需刷新生效
              location.reload();
            }
          );
          break;
        default:
          break;
      }
    };
    GM_setValue("menu_id", menu_id);
  }
  update_menu();

  // douban_book
  const douban_book_prompts = [
    (title, author) => `${author}的《${title}》简要介绍会是：`,
    (title, author) => `《${title}》的作者${author}简要介绍会是：`,
    (title, author) => `${author}的《${title}》的主要观点列成表格会是：`,
    (title, author) => `${author}的《${title}》的类似书籍或文章和它们的区别会是：`,
    (title, author) => `${author}的《${title}》的观点相反的书籍或文章和对应的观点会是：`,
    (title, author) => `${author}的《${title}》的正面评论会是：`,
    (title, author) => `${author}的《${title}》的负面评论会是：`,
    (title, author) => `${author}的《${title}》在哪些方面是有争议的会是：`,
    (title, author) => `${author}的《${title}》当人们生活在不同的时代时，会有什么不同的观点会是：`,
    (title, author) => `${author}的《${title}》当人们生活在火星时，会有什么不同的观点会是：`,
    (title, author) => `${author}的《${title}》综合Goodreads评分和豆瓣等各种评分和评价会是：`,
    (title, author) => `${author}的《${title}》比较重要的章节会是：`,
    (title, author) => `${author}的《${title}》可以和这些书籍或文章进行比较阅读：`,
    (title, author) => `${author}的《${title}》可以和这些书籍或文章进行主题阅读：`,
  ]
  // a cache for lastest 10 title-author
  menu_all.douban_book && $(() => {
    if (location.href.includes("book.douban.com/subject")) {
      const title = $("meta[property='og:title']").attr("content");
      const author = $("meta[property='book:author']").attr("content");
      const douban_book_cache = GM_getValue("douban_book_cache", []);
      // if not in cache, add it, or return
      // if (!douban_book_cache.some(item => item.title == title && item.author == author)) {
      if (true) {
        douban_book_cache.push({ title, author });
        if (douban_book_cache.length > 10) {
          douban_book_cache.shift();
        }
        GM_setValue("douban_book_cache", douban_book_cache);

        // trigger ChatGPT
        const prompt_texts = douban_book_prompts.map(prompt => prompt(title, author));
        GM_setValue("prompt_texts", prompt_texts);
      } else { return; }
    }
  });
  
  // zhihu
  const zhihu_prompts = [
    (question) => `问题：${question}，这么问可能会更好：`,
    (question) => `问题：${question}，暗含的观点是：`,
    (question) => `问题：${question}，提问者和提问者的目的是：`,
    (question) => `问题：${question}，问题的相关背景是：`,
    (question) => `问题：${question}，问题的相关历史是：`,
    (question) => `问题：${question}，问题的相关法律是：`,
    (question) => `问题：${question}，问题的相关政策是：`,
    (question) => `问题：${question}，问题的相关经济是：`,
    (question) => `问题：${question}，问题的相关科学是：`,
    (question) => `问题：${question}，问题的相关文化是：`,
    (question) => `问题：${question}，问题的责任方是：`,
    (question) => `问题：${question}，不同的职业对这个问题的看法会是：`,
    (question) => `问题：${question}，不同的年龄段对这个问题的看法会是：`,
    (question) => `问题：${question}，不同的性别对这个问题的看法会是：`,
    (question) => `问题：${question}，不同的国家对这个问题的看法会是：`,
    (question) => `问题：${question}，火星人对这个问题的看法会是：`,
    (question) => `问题：${question}，在火星上，这个问题：`,
    (question) => `问题：${question}，类似问题和它们的区别会是：`,
    (question) => `问题：${question}，观点相反的问题和对应的观点会是：`,
    (question) => `问题：${question}，正面回答会是：`,
    (question) => `问题：${question}，反面回答会是：`,
    (question) => `问题：${question}，有趣的回答会是：`,
    (question) => `问题：${question}，主要观点列成表格会是：`,
    (question) => `问题：${question}，相关书籍、文章、视频或网站会是：`,
    (question) => `从这个问题：${question}，可以引申出这些问题：`,
    (question) => `问题：${question}，可以和这些问题进行比较阅读：`,
    (question) => `问题：${question}，应该去反思：`,
    (question) => `问题：${question}，想要改进或解决它，可以从这些方面入手：`,
  ]

  menu_all.zhihu && $(() => {
    if (location.href.includes("zhihu.com/question")) {
      const question = $('meta[itemprop="name"]').attr('content');
      const zhihu_cache = GM_getValue("zhihu_cache", []);
      // if not in cache, add it, or return
      // if (!zhihu_cache.some(item => item.question == question)) {
      if (true) {
        zhihu_cache.push({ question });
        if (zhihu_cache.length > 10) {
          zhihu_cache.shift();
        }
        GM_setValue("zhihu_cache", zhihu_cache);
      
        // trigger ChatGPT
        const prompt_texts = zhihu_prompts.map(prompt => prompt(question));
        GM_setValue("prompt_texts", prompt_texts);
      } else { return; }
    }
  });

  // github
  const github_prompts = [
    (website) => `${website}的简要介绍是：`,
    (website) => `${website}的主要功能列成表格是：`,
    (website) => `${website}的维护情况是：`,
    (website) => `${website}的安装方法和最小使用样例是：`,
    (website) => `${website}的最佳实践是：`,
    (website) => `${website}的类似项目是：`,
    (website) => `${website}的不同之处是：`,
    (website) => `${website}的正面评论是：`,
    (website) => `${website}的负面评论是：`,
    (website) => `${website}的相关书籍、文章、视频或网站是：`,
    (website) => `${website}的上一代项目是：`,
    (website) => `${website}的下一代项目是：`,
    (website) => `${website}的应用场景是：`,
    (website) => `${website}的不适用场景是：`,
    (website) => `${website}可以解决的问题是：`,
    (website) => `我应该用这个项目，因为：`,
    (website) => `我不应该用这个项目，因为：`,
    (website) => `如果我想改进${website}，我可以从这些项目入手：`,
  ]
  
  menu_all.github && $(() => {
    // github repo?
    if (!location.href.includes("github.com")) { return; }
    const parts = location.href.split("/");
    if (parts.length >= 5 && parts[parts.length - 2] && parts[parts.length - 1]) {
      const github_cache = GM_getValue("github_cache", []);
      // if (!github_cache.some(item => item.website == location.href)) {
      if (true) {
        github_cache.push({ website: location.href });
        if (github_cache.length > 10) {
          github_cache.shift();
        }
        GM_setValue("github_cache", github_cache);
      
        // trigger ChatGPT
        const prompt_texts = github_prompts.map(prompt => prompt(location.href));
        GM_setValue("prompt_texts", prompt_texts);
      } else { return; }
    } 
  });

  /* ************************************************************************* */
  // ChatGPT response to prompt_texts
  const get_submit_button = () => {
    const form = document.querySelector('form');
    const buttons = form.querySelectorAll('button');
    const result = buttons[buttons.length - 1]; // by textContent maybe better
    return result;
  };
  const get_textarea = () => {
    const form = document.querySelector('form');
    const textareas = form.querySelectorAll('textarea');
    const result = textareas[0];
    return result;
  };
  const get_regenerate_button = () => {
    const form = document.querySelector('form');
    const buttons = form.querySelectorAll('button');
    for (let i = 0; i < buttons.length; i++) {
      const buttonText = buttons[i].textContent.trim().toLowerCase();
      if (buttonText.includes('regenerate')) {
        return buttons[i];
      }
    }
  };

  let last_trigger_time = +new Date();
  $(() => {
    if (location.href.includes("chat.openai")) {
      console.log("ChatGPT");
      GM_addValueChangeListener("prompt_texts", (name, old_value, new_value) => {
        if (+new Date() - last_trigger_time < 500) {
          return;
        }
        last_trigger_time = new Date();
        setTimeout(async () => {
          console.log("ChatGPT页面响应prompt_texts");
          const prompt_texts = new_value;
          console.log(prompt_texts);
          if (prompt_texts.length > 0) {
            console.log("进入处理");
            // get prompt_texts from local
            let firstTime = true;
            while (prompt_texts.length > 0) {
              if (!firstTime) {
                await new Promise(resolve => setTimeout(resolve, 2000));
              }
              if (!firstTime && get_regenerate_button() == undefined) {
                continue;
              }
              firstTime = false;
              const prompt_text = prompt_texts.shift();
              console.log(prompt_text);
              // write the prompt_text
              get_textarea().value = prompt_text;
              // submit
              get_submit_button().click();
            }
          }
        }, 0);
        GM_setValue("prompt_texts", []);
      });
    }
  });
})();