用自动脚本整理 Derpibooru 图集

(async () => {
  const galleryId = 14196; // 图集ID
  const perSet = 6; // 每个图组的图片数量
  const perPage = 50; // 获取图集内所有图片时，每页返回多少张（最高50）
  const filterId = 56027; // 图片过滤器编号（这里使用的是 Everything）

  const listImagesApi = `/api/v1/json/search/images?q=gallery_id:${galleryId}&sf=gallery_id:${galleryId}&sd=asc&filter_id=${filterId}&per_page=${perPage}&page=`;
  const orderImagesApi = `/galleries/${galleryId}/order`;

  // main procedure
  var current = await getImgList();
  console.log("original data: ", current);
  var sorted = sortImgs(current);
  console.log("sorted data: ", sorted);
  await submitOrder([...sorted].reverse());

  /**
   * 获取图集内图片
   * @returns {number[]} 图片ID列表
   */
  async function getImgList() {
    console.log("GET IMAGE LIST...");
    /** @type {number[]} */ var result = [];
    let i = 1;
    do {
      var data = JSON.parse(await (await fetch(listImagesApi + i)).text());
      result.push(...data["images"].map((i) => i["id"]));
      console.log(`✓ page ${i} of ${Math.ceil(data["total"] / perPage)}`);
    } while (i++ * perPage < data["total"]);
    console.log(`FETCH COMPLETE!`);
    return result;
  }

  /**
   * 排序图片：每个图组内图片按ID排序，图组按首张图片ID排序
   * @param {number[]} current 当前图片ID列表
   * @returns {number[]} 排序后的图片ID列表
   */
  function sortImgs(current) {
    /** @type {number[][]} */ var sets = [];
    for (let i = 0; i < current.length; i += perSet) {
      let set = current.slice(i, i + perSet);
      set.sort((a, b) => a - b);
      sets.push(set);
    }
    sets.sort((a, b) => a[0] - b[0]);
    return sets.flat();
  }

  /**
   * 提交新的图片顺序到服务器
   * @param {number[]} sorted
   */
  function submitOrder(sorted) {
    console.log("SUBMITTING DATA...");
    var xhr = new XMLHttpRequest();
    xhr.open("PATCH", orderImagesApi);
    xhr.setRequestHeader("Content-Type", "application/json");
    xhr.setRequestHeader("x-csrf-token", window.booru.csrfToken);
    xhr.send(
      JSON.stringify({
        image_ids: [...sorted],
        _method: "PATCH"
      })
    );
    return new Promise((resolve, reject) => {
      xhr.onload = () => {
        console.log("DONE!");
        resolve();
      };
      xhr.onerror = () => {
        reject("REQUEST FAILED!");
      };
    });
  }
})().catch((err) => console.error(err));

/**
 * @typedef {{
 *   comment_count: number,
 *   created_at: string,
 *   downvotes: number,
 *   faves: number,
 *   first_seen_at: string,
 *   id: number,
 *   score: number,
 *   size: number,
 *   tag_count: number,
 *   updated_at: string,
 *   upvotes: number,
 *   wilson_score: number,
 * }} ImageResponse
 */

const perPage = 50 // 获取图集内所有图片时，每页返回多少张（最高50）
const filterId = 56027 // 图片过滤器编号（Everything）

/**
 * 获取图集内图片
 * @param {number} galleryId 图集ID
 * @param {number} perSet 每个图组的图片数量
 */
async function getImgList(galleryId, perSet = 6) {
  const listImagesApi = `/api/v1/json/search/images?q=gallery_id:${galleryId}&sf=gallery_id:${galleryId}&sd=asc&filter_id=${filterId}&per_page=${perPage}&page=`

  console.log("GET IMAGE LIST...")
  /** @type {ImageResponse[]} */ var result = []
  let i = 1

  do {
    var data = JSON.parse(await (await fetch(listImagesApi + i)).text())
    result.push(...data["images"])
    console.log(`✓ page ${i} of ${Math.ceil(data["total"] / perPage)}`)
  } while (i++ * perPage < data["total"])

  console.log(`FETCH COMPLETE!`)

  /** @type {ImageResponse[][]} */ var sets = []
  for (let i = 0; i < result.length; i += perSet) {
    let set = result.slice(i, i + perSet)
    sets.push(set)
  }
  return sets
}

/**
 * @param {ImageResponse[][]} sets
 */
function toIds(sets) {
  return sets.flat().map(img => img.id)
}

/**
 * 提交新的图片顺序到服务器
 * @param {number} galleryId 图集ID
 * @param {number[]} sorted
 */
function submitOrder(galleryId, sorted) {
  const orderImagesApi = `/galleries/${galleryId}/order`

  console.log("SUBMITTING DATA...")
  var xhr = new XMLHttpRequest()
  xhr.open("PATCH", orderImagesApi)
  xhr.setRequestHeader("Content-Type", "application/json")
  xhr.setRequestHeader("x-csrf-token", window.booru.csrfToken)
  xhr.send(
    JSON.stringify({
      image_ids: [...sorted],
      _method: "PATCH",
    })
  )

  return new Promise((resolve, reject) => {
    xhr.onload = () => {
      console.log("DONE!")
      resolve()
    }
    xhr.onerror = () => {
      reject("REQUEST FAILED!")
    }
  })
}

// 整理主图集

var galleryId = 14196
var current = await getImgList(galleryId)
console.log("original data: ", current)

var coverIndex = -1
current.forEach((set, i) => {
  set.sort((l, r) => l.id - r.id)
  set._wilsonSum = set.reduce((s, c) => s + c.wilson_score, 0)
  if (set.findIndex(img => img.id === 2461114) !== -1) coverIndex = i
})
if (coverIndex === -1) throw "cover set not found"
var coverSet = current[coverIndex]

var wilsonSorted = current.slice()
wilsonSorted.splice(coverIndex, 1)
wilsonSorted.sort((l, r) => r._wilsonSum - l._wilsonSum)

var result = toIds([
  ...wilsonSorted.slice(4).sort((l, r) => l[5].id - r[5].id),
  ...wilsonSorted.slice(0, 4).reverse(),
  coverSet,
])
console.log("sorted data: ", result)

await submitOrder(galleryId, result.slice().reverse())

// 整理其他图集

var galleryId = 18473
var current = await getImgList(galleryId)
console.log("original data: ", current)

current.forEach((set, i) => {
  set.sort((l, r) => l.id - r.id)
})

var result = toIds(current.sort((l, r) => l[5].id - r[5].id))
console.log("sorted data: ", result)

await submitOrder(galleryId, result.slice().reverse())