search.go

package lib

import (
	"context"

	"github.com/qri-io/qri/base/params"
	qhttp "github.com/qri-io/qri/lib/http"
	"github.com/qri-io/qri/registry"
	"github.com/qri-io/qri/registry/regclient"
	"github.com/qri-io/qri/repo"
)

// SearchMethods groups together methods for search
type SearchMethods struct {
	d dispatcher
}

// Name returns the name of this method group
func (m SearchMethods) Name() string {
	return "search"
}

// Attributes defines attributes for each method
func (m SearchMethods) Attributes() map[string]AttributeSet {
	return map[string]AttributeSet{
		"search": {Endpoint: qhttp.AESearch, HTTPVerb: "POST"},
	}
}

// SearchParams defines paremeters for the search Method
type SearchParams struct {
	params.List
	Query string `json:"q"`
}

// SetNonZeroDefaults sets a default limit and offset
func (p *SearchParams) SetNonZeroDefaults() {
	if p.Offset < 0 {
		p.Offset = 0
	}
	if p.Limit <= 0 {
		p.Limit = params.DefaultListLimit
	}
}

// Search queries for items on qri related to given parameters
func (m SearchMethods) Search(ctx context.Context, p *SearchParams) ([]registry.SearchResult, error) {
	got, _, err := m.d.Dispatch(ctx, dispatchMethodName(m, "search"), p)
	if res, ok := got.([]registry.SearchResult); ok {
		return res, err
	}
	return nil, dispatchReturnError(got, err)
}

// Implementations for FSI methods follow

// searchImpl holds the method implementations for search
type searchImpl struct{}

// Search queries for items on qri related to given parameters
func (searchImpl) Search(scope scope, p *SearchParams) ([]registry.SearchResult, error) {
	client := scope.RegistryClient()
	if client == nil {
		return nil, repo.ErrNoRegistry
	}
	params := &regclient.SearchParams{
		Query:  p.Query,
		Limit:  p.Limit,
		Offset: p.Offset,
	}

	regResults, err := client.Search(scope.Context(), params)
	if err != nil {
		return nil, err
	}
	return regResults, nil
}