example.go

package main

import (
	"fmt"
	"io/ioutil"

	"github.com/DeanThompson/zhihu-go"
)

var (
	logger = zhihu.Logger{true}
)

func main() {
	zhihu.Init("./config.json")

	// 黄继新，和知乎在一起
	user := zhihu.NewUser("https://www.zhihu.com/people/jixin", "")
	showUser(user)

	logger.Success("========== split ==========")

	// Python 编程，应该养成哪些好的习惯？
	questionUrl := "https://www.zhihu.com/question/28966220"
	question := zhihu.NewQuestion(questionUrl, "")
	showQuestion(question)

	logger.Success("========== split ==========")

	// 龙有九个儿子，是跟谁生的？为什么「龙生九子，各不成龙」？豆子 的答案
	answer := zhihu.NewAnswer("https://www.zhihu.com/question/23759686/answer/41997389", nil, nil)
	showAnswer(answer)

	logger.Success("========== split ==========")

	// 程序员为了期权加入创业公司，值得吗？ 匿名用户的答案
	answer2 := zhihu.NewAnswer("https://www.zhihu.com/question/28023819/answer/49723406", nil, nil)
	showAnswer(answer2)

	logger.Success("========== split ==========")

	// 黄继新 A4U
	collection := zhihu.NewCollection("https://www.zhihu.com/collection/19677733", "", nil)
	showCollection(collection)

	// Python
	topic := zhihu.NewTopic("https://www.zhihu.com/topic/19552832", "")
	showTopic(topic)
}

func showQuestion(question *zhihu.Question) {
	logger.Info("Question fields:")
	logger.Info("	url: %s", question.Link)
	logger.Info("	title: %s", question.GetTitle())
	logger.Info("	detail: %s", question.GetDetail())
	logger.Info("	answers num: %d", question.GetAnswersNum())
	logger.Info("	followers num: %d", question.GetFollowersNum())
	logger.Info("	comments num: %d", question.GetCommentsNum())

	for i, topic := range question.GetTopics() {
		logger.Info("	topic-%d: %s", i+1, topic.String())
	}

	for i, follower := range question.GetFollowersN(5) {
		logger.Info("	top follower-%d: %s", i+1, follower.String())
	}

	for i, follower := range question.GetFollowers() {
		logger.Info("	follower-%d: %s", i+1, follower.String())
		if i >= 10 {
			logger.Info("	%d followers not shown.", question.GetFollowersNum()-i-1)
			break
		}
	}

	allAnswers := question.GetAllAnswers()
	for i, answer := range allAnswers {
		logger.Info("	answer-%d: %s", i+1, answer.String())
		filename := fmt.Sprintf("/tmp/%s-%s的回答.html", question.GetTitle(), answer.GetAuthor().GetUserID())
		dumpAnswerHTML(filename, answer)
		if i >= 10 {
			logger.Info("	%d answers not shown.", len(allAnswers)-i-1)
			break
		}
	}

	topXAnswers := question.GetTopXAnswers(25)
	for i, answer := range topXAnswers {
		logger.Info("	top-%d answer: %s", i+1, answer.String())
	}

	logger.Info("	top-1 answer: %s", question.GetTopAnswer().String())
	logger.Info("	visit times: %d", question.GetVisitTimes())
}

func showAnswer(answer *zhihu.Answer) {
	logger.Info("Answer fields:")
	logger.Info("	url: %s", answer.Link)

	question := answer.GetQuestion()
	logger.Info("	question url: %s", question.Link)
	logger.Info("	question title: %s", question.GetTitle())

	logger.Info("	author: %s", answer.GetAuthor().String())
	logger.Info("	upvote num: %d", answer.GetUpvote())
	logger.Info("	comments num: %d", answer.GetCommentsNum())
	logger.Info("	collected num: %d", answer.GetCollectedNum())
	logger.Info("	data ID: %d", answer.GetID())

	// dump content
	filename := fmt.Sprintf("/tmp/answer_%d.html", answer.GetID())
	dumpAnswerHTML(filename, answer)

	voters := answer.GetVoters()
	for i, voter := range voters {
		logger.Info("	voter-%d: %s", i+1, voter.String())
		if i >= 10 {
			remain := len(voters) - i - 1
			logger.Info("	%d votes not shown.", remain)
			break
		}
	}
}

func showCollection(collection *zhihu.Collection) {
	logger.Info("Collection fields:")
	logger.Info("	url: %s", collection.Link)
	logger.Info("	name: %s", collection.GetName())
	logger.Info("	creator: %s", collection.GetCreator().String())
	logger.Info("	followers num: %d", collection.GetFollowersNum())
	logger.Info("	comments num: %d", collection.GetCommentsNum())
	logger.Info("	questions num: %d", collection.GetQuestionsNum())
	logger.Info("	answers num: %d", collection.GetAnswersNum())

	for i, follower := range collection.GetFollowersN(5) {
		logger.Info("	top follower-%d: %s", i+1, follower.String())
	}

	for i, follower := range collection.GetFollowers() {
		logger.Info("	follower-%d: %s", i+1, follower.String())
	}

	for i, question := range collection.GetQuestionsN(5) {
		logger.Info("	top question-%d: %s", i+1, question.String())
	}

	for i, question := range collection.GetQuestions() {
		logger.Info("	question-%d: %s", i+1, question.String())
	}

	for i, answer := range collection.GetAnswersN(5) {
		logger.Info("	top answer-%d: %s", i+1, answer.String())
	}

	for i, answer := range collection.GetAnswers() {
		logger.Info("	answer-%d: %s", i+1, answer.String())
	}
}

func showUser(user *zhihu.User) {
	logger.Info("User fields:")
	logger.Info("	is anonymous: %v", user.IsAnonymous())
	logger.Info("	userId: %s", user.GetUserID())
	logger.Info("	dataId: %s", user.GetDataID())
	logger.Info("	avatar: %s", user.GetAvatar())
	logger.Info("	avatar with size hd: %s", user.GetAvatarWithSize("hd"))
	logger.Info("	bio: %s", user.GetBio())
	logger.Info("	location: %s", user.GetLocation())
	logger.Info("	business: %s", user.GetBusiness())
	logger.Info("	education: %s", user.GetEducation())
	logger.Info("	gender: %s", user.GetGender())
	logger.Info("	weibo: %s", user.GetWeiboURL())
	logger.Info("	followers num: %d", user.GetFollowersNum())
	logger.Info("	followees num: %d", user.GetFolloweesNum())
	logger.Info("	followed columns num: %d", user.GetFollowedColumnsNum())
	logger.Info("	followed topics num: %d", user.GetFollowedTopicsNum())
	logger.Info("	agree num: %d", user.GetAgreeNum())
	logger.Info("	thanks num: %d", user.GetThanksNum())
	logger.Info("	asks num: %d", user.GetAsksNum())
	logger.Info("	answers num: %d", user.GetAnswersNum())
	logger.Info("	posts num: %d", user.GetPostsNum())
	logger.Info("	collections num: %d", user.GetCollectionsNum())
	logger.Info("	logs num: %d", user.GetLogsNum())

	for i, topic := range user.GetFollowedTopicsN(5) {
		logger.Info("	top followed topic-%d: %s", i+1, topic.String())
	}

	//	for i, topic := range user.GetFollowedTopics() {
	//		logger.Info("	followed topic-%d: %s", i+1, topic.String())
	//	}

	for i, follower := range user.GetFollowersN(5) {
		logger.Info("	top follower-%d: %s", i+1, follower.String())
	}

	//	for i, follower := range user.GetFollowers() {
	//		logger.Info("	follower-%d: %s", i+1, follower.String())
	//	}

	for i, followee := range user.GetFolloweesN(5) {
		logger.Info("	top followee-%d: %s", i+1, followee.String())
	}

	//	for i, followee := range user.GetFollowees() {
	//		logger.Info("	followee-%d: %s", i+1, followee.String())
	//	}

	for i, ask := range user.GetAsksN(5) {
		logger.Info("	top ask-%d: %s", i+1, ask.String())
	}

	//	for i, ask := range user.GetAsks() {
	//		logger.Info("	ask-%d: %s", i+1, ask.String())
	//	}

	for i, answer := range user.GetAnswersN(5) {
		logger.Info("	top answer-%d: %s", i+1, answer.String())
	}

	//	for i, answer := range user.GetAnswers() {
	//		logger.Info("	answer-%d: %s", i+1, answer.String())
	//	}

	for i, collection := range user.GetCollectionsN(5) {
		logger.Info("	top collection-%d: %s", i+1, collection.String())
	}

	//	for i, collection := range user.GetCollections() {
	//		logger.Info("	collection-%d: %s", i+1, collection.String())
	//	}

	for i, like := range user.GetLikes() {
		logger.Info("	like-%d: %s", i+1, like.String())
	}
}

func showTopic(topic *zhihu.Topic) {
	logger.Info("Topic fields:")
	logger.Info("	url: %s", topic.Link)
	logger.Info("	name: %s", topic.GetName())
	logger.Info("	description: %s", topic.GetDescription())
	logger.Info("	followers num: %d", topic.GetFollowersNum())

	for i, author := range topic.GetTopAuthors() {
		logger.Info("	top-%d author: %s", i+1, author.String())
	}
}

func dumpAnswerHTML(filename string, answer *zhihu.Answer) error {
	err := ioutil.WriteFile(filename, []byte(answer.GetContent()), 0666)
	if err == nil {
		logger.Info("	content dumped to %s", filename)
	}
	return err
}