Golang bot-session.Session类(方法)实例源码

func process(s *session.Session, posts *http.Posts, i int, grp chan *group) {
	for i >= 0 {

		// I don't really have to parse here, forgot kosher implementation.
		// See http/json
		id := strings.Split(posts.Data[i].Id, "_")[1]
		if posts.Data[i].User_has_liked || s.CheckCache(id) || http.IsFollowing(s, id) {
			// Try to add to channel and stop if done
			grp <- &group{
				id: "continue",
			}
			i--
			continue
		}

		//
		user := http.GetUser(s, id)

		// Create perosn to get value
		person := session.Person{
			Followers: float64(user.Data.Counts.Follows),
			Following: float64(user.Data.Counts.Followed_by),
			Posts:     float64(user.Data.Counts.Media),
		}

		// Forget sigmoid for now
		grp <- &group{
			id:    posts.Data[i].Id,
			value: person.Followers / person.Following, // person.Sigmoid(session.GetTheta()) would be the ideal way
			user:  posts.Data[i].User.Username,
		}

		i--
	}
}

func BasicDecision(s *session.Session, follows int, likes int, intervals int, done chan bool) {
	// Round robin the hashtags. Allows for manual weighting eg: [#dog,#dog,#cute]
	posts := http.GetPosts(s, s.GetHashtag(intervals))

	// Go from end to reduce collision
	// Doesn't bother checking
	i := 19
	for (likes > 0 || follows > 0) && i >= 0 {

		// Process likes
		if likes > 0 {
			go http.LikePosts(s, posts.Data[i].Id)
			likes--

			// Doing this seperately reaches larger audience
			// Never exceeds 12/11 at a given time
		} else if follows > 0 {
			go http.FollowUser(s, posts.Data[i].Id)
			follows--
		}

		// Decrement
		i--
	}

	// Indicate doneness
	done <- true
}

func processTags(s *session.Session, weight float64, tags []string) {
	weight /= float64(len(tags))
	for _, next := range tags {
		h := s.Hashtag(next)
		h.Value += weight
		s.SaveHashtag(h)
	}
}

// Post actions
func LikePosts(s *session.Session, id string) {
	v := url.Values{}

	response, err := s.Post("https://api.instagram.com/v1/media/"+id+"/likes", v)
	if err != nil {
		panic(err)
	}
	s.Log(string(response.StatusCode))
}

func FollowUser(s *session.Session, id string) {
	v := url.Values{}
	v.Set("action", "follow")

	response, err := s.Post("https://api.instagram.com/v1/users/"+strings.Split(id, "_")[1]+"/relationship", v)
	if err != nil {
		panic(err)
	}
	s.Log(string(response.StatusCode))
}

// Authenticate
func Authenticate(s *session.Session, code string) {
	decoder := s.Auth(code)

	//Decode request
	var auth Auth
	err := decoder.Decode(&auth)
	if err != nil {
		panic(err)
	}

	s.SetAuth(auth.Access_token, auth.User.Id)

}

func IntelligentDecision(s *session.Session, follows int, likes int, intervals int, done chan bool) {

	// Still do round robin, but this time the hashtags are smart
	// and our choice is educated
	posts := http.GetPosts(s, s.GetHashtag(intervals))

	// Set up channels for async download/ processing from instagram
	next := make(chan *http.Posts)
	grp := make(chan *group)
	count := 0
	calls := 0
	go sort(s, grp, follows, likes, &calls, &count, done)
	go listen(s, grp, next, &calls, &count)
	next <- &posts
}

func IsFollowing(s *session.Session, id string) bool {
	response, err := s.Get("https://api.instagram.com/v1/users/" + id + "/relationship")
	if err != nil {
		panic(err)
	}

	var status Status
	decoder := json.NewDecoder(response.Body)
	err = decoder.Decode(&status)
	if err != nil {
		panic(err)
	}

	return status.Data.Outgoing_status == "follows"
}

func GetMedia(s *session.Session, id string) Posts {
	params := map[string]string{"MIN_TIMESTAMP": utils.SixHoursAgo(), "COUNT": "3"}
	response, err := s.GetParamed("https://api.instagram.com/v1/users/"+id+"/media/recent/", params)
	if err != nil {
		panic(err)
	}

	//Decode request
	var posts Posts
	decoder := json.NewDecoder(response.Body)
	err = decoder.Decode(&posts)
	if err != nil {
		panic(err)
	}

	return posts
}

func GetUser(s *session.Session, id string) User {
	response, err := s.Get("https://api.instagram.com/v1/users/" + id)
	if err != nil {
		panic(err)
	}

	//Decode request
	var user User
	decoder := json.NewDecoder(response.Body)
	err = decoder.Decode(&user)
	if err != nil {
		panic(err)
	}

	return user

}

func getPeople(s *session.Session, url string) (users Users) {
	response, err := s.Get(url)
	if err != nil {
		panic(err)
	}

	data, err := ioutil.ReadAll(response.Body)
	if err == nil && data != nil {
		err = json.Unmarshal(data, &users)
	}
	if err != nil {
		s.Log(string(data[:]))
		panic(err)
	}

	return
}

func GetPosts(s *session.Session, hashtag string) Posts {

	response, err := s.Get("https://api.instagram.com/v1/tags/" + hashtag + "/media/recent")
	if err != nil {
		panic(err)
	}

	//Decode request
	var posts Posts
	decoder := json.NewDecoder(response.Body)
	err = decoder.Decode(&posts)
	if err != nil {
		panic(err)
	}

	return posts
}

func processHandle(w http.ResponseWriter, r *http.Request, s *session.Session) {

	// Grab intervals since day start
	intervals := utils.Intervals()

	// Had some fancy math for peroidictiy. But
	// we could just brute force 100 per hour
	likes := int(utils.LIKES / utils.CALLS)
	utils.Limit(&likes, intervals, utils.LIKES)

	if !s.Usable() {
		fmt.Fprint(w, "Please set hashtags and authorize")
		return
	}

	// Follow ratio function where target is the desired
	// amount of followers.
	// e^(x*ln(magic)/target)
	// I wish could say there's some science behind why
	// we're doing this, but ultimately we just need a
	// decreasing function and some percentage of your
	// target feels right
	count := action.GetStatus(s)
	follows := int(utils.FollowerDecay(count.Followed_by, count.Follows, s.GetMagic(), s.GetTarget()))
	utils.Limit(&follows, intervals, utils.FOLLOWS)
	if follows < 0 {
		follows = 0
	}

	// Hang on channel otherwise jobs cut out
	done := make(chan bool)

	// Save status at midnight
	if intervals == 0 {
		go s.SetRecords(count.Followed_by, count.Follows)
	}
	if s.GetLearnt() {
		IntelligentDecision(s, follows, likes, intervals, done)
	} else {
		BasicDecision(s, follows, likes, intervals, done)
	}
	// Wait for finish. Defeats the purpose of aysnc, but only way to run in prod
	<-done
	fmt.Fprint(w, "Processing")
}

func GetTag(s *session.Session, hashtag string) Tag {

	response, err := s.Get("https://api.instagram.com/v1/tags/" + hashtag)
	if err != nil {
		panic(err)
	}

	//Decode request
	var tag Tag
	decoder := json.NewDecoder(response.Body)
	err = decoder.Decode(&tag)
	if err != nil {
		panic(err)
	}

	return tag

}

// Math looks good, but haven't rigorously tested it
func LogisticRegression(s *session.Session) {

	// Grab all people because of many iterations
	people := s.GetPeople()
	objective := Objective{
		People: people,
		Lambda: s.GetLambda(),
		Alpha:  s.GetAlpha(),
		Size:   float64(len(people)),
	}

	start := []float64{1, 1, 1, 0} // Bias, Following, Followers, Posts
	minimum := Minimize(objective, start)
	log.Println(minimum)

	s.SetTheta(minimum)
	s.StopProcessing()
}

// Get actions
func GetStatus(s *session.Session) (count Counts) {

	response, err := s.Get("https://api.instagram.com/v1/users/" + s.GetId())
	if err != nil {
		panic(err)
	}

	//Decode request
	var status Status
	decoder := json.NewDecoder(response.Body)
	err = decoder.Decode(&status)
	if err != nil {
		panic(err)
	}

	count = status.Data.Counts
	return
}

func updateHandle(w http.ResponseWriter, r *http.Request, s *session.Session) {
	// Probs implement TOTP, potentially vulnerable to MTM
	if s.VerifiedUpdate(r.URL.Query()["hash"][0]) {
		s.SetHashtags(strings.Split(r.URL.Query()["hashtags"][0], " "))
		s.ParseTheta(strings.Split(r.URL.Query()["theta"][0], " "))
		s.SetLearnt()
		fmt.Fprint(w, "Updated")
	} else {
		fmt.Fprint(w, "Not Verified")
	}
}

func GetNextPost(s *session.Session, url string) Posts {
	response, err := s.RawGet(url)
	if err != nil {
		panic(err)
	}

	//Decode request
	var bunch Posts
	data, err := ioutil.ReadAll(response.Body)
	if err == nil && data != nil {
		err = json.Unmarshal(data, &bunch)
	}
	if err != nil {
		s.Log(string(data[:]))
		panic(err)
	}

	return bunch
}

func process(s *session.Session, users *http.Users, i int, follows float64) {
	for i >= 0 {

		id := users.Data[i].Id
		user := http.GetUser(s, id)

		log.Println(user)

		if user.Data.Counts.Followed_by+user.Data.Counts.Follows > 0 {
			//check follower records, if following and guy in other records, don't do anythin
			person := session.Person{
				Followers: float64(user.Data.Counts.Follows),
				Following: float64(user.Data.Counts.Followed_by),
				Posts:     float64(user.Data.Counts.Media),
				Follows:   !s.CheckCache(id),
			}

			// Because unset properties won't change, this should be fine
			if int(follows) == utils.SCORN {
				person.Followed = true
				person.Follows = !person.Follows
			}

			// Add to variable and to Keys
			s.PutPerson(person, id)
		}

		// Decrement
		i--
	}

	// Catches up and thus done
	if s.FinishedCount() {
		s.SavePeople()
		s.StopProcessing()
	}
}

// Build a little status bar for debug and ./toolbelt -l purposes
func StatusBar(s *session.Session, title string) (bar string) {
	bar = "    " + title + ":"
	BARSIZE := 100 - len(title)
	i := int(BARSIZE * s.GetState() / s.GetLimit())
	j := BARSIZE - i

	for i+j > 0 {
		if i > 0 {
			i--
			bar += "*"
		} else {
			j--
			bar += "-"
		}
	}
	bar += utils.IntToString(s.GetState()) + "/" + utils.IntToString(s.GetLimit())
	return
}