omnivore/packages/content-handler/src/websites/zhihu-handler.ts

import { ContentHandler } from '../content-handler'

export class ZhihuHandler extends ContentHandler {
  constructor() {
    super()
    this.name = 'zhihu'
  }

  parseQuestion(element: Element) {
    const newQuestion = element.ownerDocument.createElement('div')
    const question = element.querySelector(`.QuestionHeader-main`)
    if (question) {
      const votes = element
        .querySelector(`div[itemprop='upvoteCount']`)
        ?.getAttribute('data-value')

      if (votes) {
        newQuestion.innerHTML = `<h2>问题: ${votes} vote${
          votes === '1' ? '' : 's'
        }</h2>${question.innerHTML}`
      }
    }
    return newQuestion
  }

  parseComments(element: Element) {
    const dom = element.ownerDocument
    const newComments = dom.createElement('div')

    // comments
    const commentsDiv = element.querySelector(`.comments`)
    if (commentsDiv) {
      const comments = commentsDiv.querySelectorAll(`.comment`)
      if (comments.length > 0) {
        newComments.innerHTML = `<h3>Comments</h3>`

        comments.forEach((comment) => {
          const author = comment.querySelector(`.comment-user`)
          const text = comment.querySelector(`.comment-copy`)?.textContent
          const authorHref = author?.getAttribute('href')
          const date = comment.querySelector(`.relativetime-clean`)?.textContent
          if (author && text && authorHref && date) {
            const newComment = dom.createElement('p')
            newComment.innerHTML = `<a href="${authorHref}"><b>${author.innerHTML}</b></a>: ${text} - ${date}`
            newComments.appendChild(newComment)
          }
        })
      }
    }

    return newComments
  }

  parseAuthors(element: Element) {
    const dom = element.ownerDocument
    const newAuthors = dom.createElement('div')

    const authors = element.querySelectorAll(`.post-signature`)
    authors.forEach((author) => {
      const isOwner = author.classList.contains('owner')
      const name = author.querySelector(`.user-details a`)?.textContent
      const link = author.querySelector(`.user-details a`)?.getAttribute('href')
      const reputation = author.querySelector(`.reputation-score`)?.textContent
      const badges = Array.from(
        author.querySelectorAll(`span[title*='badges']`)
      )
        .map((badge) => badge.getAttribute('title'))
        .join(', ')
      const date = author.querySelector(`.user-action-time`)?.textContent
      if (name && link && reputation && date) {
        const newAuthor = dom.createElement('p')
        newAuthor.innerHTML = `<a href="${link}"><b>${name}</b></a> - ${reputation} reputation - ${
          badges || 'no badge'
        } - ${date}`
        if (isOwner) {
          const author = dom.createElement('span')
          author.setAttribute('rel', 'author')
          author.innerHTML = name
          newAuthor.appendChild(author)
        }
        newAuthors.appendChild(newAuthor)
      }
    })

    return newAuthors
  }

  shouldPreParse(url: string, dom: Document): boolean {
    return new URL(url).hostname.endsWith('zhihu.com')
  }

  async preParse(url: string, dom: Document): Promise<Document> {
    const mainEntity = dom.querySelector(`div[itemprop='mainEntity']`)
    if (mainEntity) {
      const newMainEntity = dom.createElement('div')
      const question = mainEntity.querySelector('.QuestionHeader')
      if (question) {
        question.className = '_omnivore_zhihu_question'
        newMainEntity.appendChild(question)
      }

      const answers = mainEntity.querySelectorAll('.ContentItem.AnswerItem')
      answers.forEach((answer) => {
        answer
          .querySelector('.AuthorInfo')
          ?.setAttribute('class', '_omnivore_zhihu_author')

        answer.className = '_omnivore_zhihu_answer'
        newMainEntity.appendChild(answer)
      })

      dom.body.replaceChildren(newMainEntity)
    }

    return Promise.resolve(dom)
  }
}