hashtag/go/codesearchsource/codesearchsource.go

package codesearchsource

import (
	"context"

	"github.com/spf13/viper"
	"go.skia.org/infra/go/codesearch"
	"go.skia.org/infra/go/httputils"
	"go.skia.org/infra/go/sklog"
	"go.skia.org/infra/hashtag/go/source"
)

// codesearchSource implements source.Source.
type codesearchSource struct {
	cs *codesearch.CodeSearch

	// prefix is added to each query.
	prefix string
}

// New returns a new Source.
func New() (source.Source, error) {
	c := httputils.DefaultClientConfig().With2xxOnly().Client()
	cs := codesearch.New(c)
	return &codesearchSource{
		cs:     cs,
		prefix: viper.GetString("sources.codesearch.prefix"),
	}, nil
}

// See source.Source.
func (cs *codesearchSource) ByHashtag(hashtag string) <-chan source.Artifact {
	ret := make(chan source.Artifact)
	go func() {
		defer close(ret)
		results, err := cs.cs.Query(context.Background(), cs.prefix+" "+hashtag, nil)
		if err != nil {
			sklog.Errorf("Failed to build code search: %s", err)
			return
		}
		for _, r := range results.SearchResult {
			ret <- source.Artifact{
				Title: r.TopFile.File.Name,
				URL:   cs.cs.URL(r),
			}
		}
	}()

	return ret

}

// See source.Source.
func (cs *codesearchSource) ByUser(hashtag string) <-chan source.Artifact {
	ret := make(chan source.Artifact)
	close(ret)
	return ret
}