Пример #1
0
func auth(w http.ResponseWriter, r *http.Request) {
	username, password, ok := r.BasicAuth()
	if !ok {
		response.Forbidden(w)
		return
	}

	instapaperClient := instapaper.New(username, password)
	ok, err := instapaperClient.Authenticate()
	if err != nil {
		response.InternalServerError(w)
		return
	}

	if ok {
		response.OK(w)
		return
	}

	response.Forbidden(w)
}
Пример #2
0
func importStories(w http.ResponseWriter, r *http.Request) {
	if r.Method != "POST" {
		log.Printf("%v method not allowed for /import", r.Method)
		response.MethodNotAllowed(w)
		return
	}

	username, password, ok := r.BasicAuth()
	if !ok {
		response.Forbidden(w)
		return
	}
	limit := parseLimit(r)
	log.Printf("importing %v stories for %v", limit, username)

	hnClient := hn.New()
	stories, err := hnClient.TopStories()
	if err != nil {
		response.BadGateway(w, err)
		return
	}

	var errs []error
	var addedStories []hn.Item
	instapaperClient := instapaper.New(username, password)
	s := semaphore.New(10)
	var wg sync.WaitGroup
	for i, id := range stories {
		if i >= limit {
			break
		}

		wg.Add(1)
		s.Acquire(1)

		go func(id int) {
			defer wg.Done()
			defer s.Release(1)

			story, err := hnClient.GetPost(id)
			if err != nil {
				errs = append(errs, err)
				return
			}
			if story.URL == nil {
				log.Printf("skipping story %v with title %v", id, story.Title)
				return
			}

			_, err = instapaperClient.Add(instapaper.AddParams{
				URL:   *story.URL,
				Title: story.Title,
			})
			if err != nil {
				errs = append(errs, err)
				log.Println("error adding", story, err)
				return
			}
			addedStories = append(addedStories, story)
		}(id)
	}
	wg.Wait()

	if len(errs) != 0 {
		response.BadGateway(w, errs, addedStories)
		return
	}

	log.Printf("imported %v stories for %v", len(addedStories), username)
	response.JSON(w, addedStories)
}