perkeep/pkg/search/handler.go

1240 lines
31 KiB
Go

/*
Copyright 2011 Google Inc.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package search
import (
"bytes"
"errors"
"fmt"
"log"
"net/http"
"net/url"
"os"
"sort"
"strconv"
"strings"
"sync"
"time"
"camlistore.org/pkg/blobref"
"camlistore.org/pkg/blobserver"
"camlistore.org/pkg/httputil"
"camlistore.org/pkg/jsonconfig"
)
const buffered = 32 // arbitrary channel buffer size
const maxPermanodes = 50 // arbitrary limit on the number of permanodes fetched
func init() {
blobserver.RegisterHandlerConstructor("search", newHandlerFromConfig)
}
type Handler struct {
index Index
owner *blobref.BlobRef
}
// IGetRecentPermanodes is the interface encapsulating the GetRecentPermanodes query.
type IGetRecentPermanodes interface {
// GetRecentPermanodes returns recently-modified permanodes.
// This is a higher-level query returning more metadata than the index.GetRecentPermanodes,
// which only scans the blobrefs but doesn't return anything about the permanodes.
// TODO: rename this one?
GetRecentPermanodes(*RecentRequest) (*RecentResponse, error)
}
var (
_ IGetRecentPermanodes = (*Handler)(nil)
)
func NewHandler(index Index, owner *blobref.BlobRef) *Handler {
return &Handler{index: index, owner: owner}
}
func newHandlerFromConfig(ld blobserver.Loader, conf jsonconfig.Obj) (http.Handler, error) {
indexPrefix := conf.RequiredString("index") // TODO: add optional help tips here?
ownerBlobStr := conf.RequiredString("owner")
devBlockStartupPrefix := conf.OptionalString("devBlockStartupOn", "")
if err := conf.Validate(); err != nil {
return nil, err
}
if devBlockStartupPrefix != "" {
_, err := ld.GetHandler(devBlockStartupPrefix)
if err != nil {
return nil, fmt.Errorf("search handler references bogus devBlockStartupOn handler %s: %v", devBlockStartupPrefix, err)
}
}
indexHandler, err := ld.GetHandler(indexPrefix)
if err != nil {
return nil, fmt.Errorf("search config references unknown handler %q", indexPrefix)
}
indexer, ok := indexHandler.(Index)
if !ok {
return nil, fmt.Errorf("search config references invalid indexer %q (actually a %T)", indexPrefix, indexHandler)
}
ownerBlobRef := blobref.Parse(ownerBlobStr)
if ownerBlobRef == nil {
return nil, fmt.Errorf("search 'owner' has malformed blobref %q; expecting e.g. sha1-xxxxxxxxxxxx",
ownerBlobStr)
}
return &Handler{
index: indexer,
owner: ownerBlobRef,
}, nil
}
// TODO: figure out a plan for an owner having multiple active public keys, or public
// key rotation
func (h *Handler) Owner() *blobref.BlobRef {
return h.owner
}
func (h *Handler) Index() Index {
return h.index
}
func jsonMap() map[string]interface{} {
return make(map[string]interface{})
}
func jsonMapList() []map[string]interface{} {
return make([]map[string]interface{}, 0)
}
func (sh *Handler) ServeHTTP(rw http.ResponseWriter, req *http.Request) {
ret := jsonMap()
_ = req.Header.Get("X-PrefixHandler-PathBase")
suffix := req.Header.Get("X-PrefixHandler-PathSuffix")
if req.Method == "GET" {
switch suffix {
case "camli/search/recent":
sh.serveRecentPermanodes(rw, req)
return
case "camli/search/permanodeattr":
sh.servePermanodesWithAttr(rw, req)
return
case "camli/search/describe":
sh.serveDescribe(rw, req)
return
case "camli/search/claims":
sh.serveClaims(rw, req)
return
case "camli/search/files":
sh.serveFiles(rw, req)
return
case "camli/search/signerattrvalue":
sh.serveSignerAttrValue(rw, req)
return
case "camli/search/signerpaths":
sh.serveSignerPaths(rw, req)
return
case "camli/search/edgesto":
sh.serveEdgesTo(rw, req)
return
}
}
// TODO: discovery for the endpoints & better error message with link to discovery info
ret["error"] = "Unsupported search path or method"
ret["errorType"] = "input"
httputil.ReturnJSON(rw, ret)
}
// RecentRequest is a request to get a RecentResponse.
type RecentRequest struct {
N int // if zero, default number of results
Before time.Time // if zero, now
ThumbnailSize int // if zero, no thumbnails
}
func (r *RecentRequest) URLSuffix() string {
// TODO: Before
return fmt.Sprintf("camli/search/recent?n=%d&thumbnails=%d", r.n(), r.thumbnailSize())
}
func (r *RecentRequest) FromHTTP(req *http.Request) error {
r.N, _ = strconv.Atoi(req.FormValue("n"))
r.ThumbnailSize = thumbnailSize(req)
// TODO: populate Before
return nil
}
// n returns the sanitized number of search results.
func (r *RecentRequest) n() int {
if r.N <= 0 || r.N > 1000 {
return 50
}
return r.N
}
func (r *RecentRequest) thumbnailSize() int {
v := r.ThumbnailSize
if v == 0 {
return 0
}
if v < minThumbSize || v > maxThumbSize {
return defThumbSize
}
return v
}
// WithAttrRequest is a request to get a WithAttrResponse.
type WithAttrRequest struct {
N int // max number of results
Signer *blobref.BlobRef // if nil, search will return index.ErrNotFound
// Requested attribute. If blank, all attributes are searched (for Value)
// as fulltext.
Attr string
// Value of the requested attribute. If blank, permanodes which have
// request.Attribute as an attribute are searched.
Value string
Fuzzy bool // fulltext search (if supported).
}
func (r *WithAttrRequest) FromHTTP(req *http.Request) error {
v := req.FormValue("signer")
if v == "" {
return errors.New("missing required parameter: \"signer\"")
}
r.Signer = blobref.Parse(v)
if r.Signer == nil {
return fmt.Errorf("failed to parse signer blobref: %v", v)
}
r.Value = req.FormValue("value")
fuzzy := req.FormValue("fuzzy") // exact match if empty
fuzzyMatch := false
if fuzzy != "" {
lowered := strings.ToLower(fuzzy)
if lowered == "true" || lowered == "t" {
fuzzyMatch = true
}
}
r.Attr = req.FormValue("attr") // all attributes if empty
if r.Attr == "" { // and force fuzzy in that case.
fuzzyMatch = true
}
r.Fuzzy = fuzzyMatch
maxResults := maxPermanodes
max := req.FormValue("max")
if max != "" {
maxR, err := strconv.Atoi(max)
if err != nil {
return fmt.Errorf("Invalid specified max results 'max': " + err.Error())
}
if maxR < maxResults {
maxResults = maxR
}
}
r.N = maxResults
return nil
}
// A MetaMap is a map from blobref to a DescribedBlob.
type MetaMap map[string]*DescribedBlob
func (m MetaMap) Get(br *blobref.BlobRef) *DescribedBlob {
if br == nil {
return nil
}
return m[br.String()]
}
// RecentResponse is the JSON response from $searchRoot/camli/search/recent.
type RecentResponse struct {
Recent []*RecentItem `json:"recent"`
Meta MetaMap `json:"meta"`
Error string `json:"error,omitempty"`
ErrorType string `json:"errorType,omitempty"`
}
func (r *RecentResponse) Err() error {
if r.Error != "" || r.ErrorType != "" {
if r.ErrorType != "" {
return fmt.Errorf("%s: %s", r.ErrorType, r.Error)
}
return errors.New(r.Error)
}
return nil
}
// DescribeResponse is the JSON response from $searchRoot/camli/search/describe.
type DescribeResponse struct {
Meta MetaMap `json:"meta"`
}
// WithAttrResponse is the JSON response from $searchRoot/camli/search/permanodeattr.
type WithAttrResponse struct {
WithAttr []*WithAttrItem `json:"withAttr"`
Meta MetaMap `json:"meta"`
}
// SignerPathsResponse is the JSON response from $searchRoot/camli/search/signerpaths.
type SignerPathsResponse struct {
Paths []*SignerPathsItem `json:"paths"`
Meta MetaMap `json:"meta"`
}
// A RecentItem is an item returned from $searchRoot/camli/search/recent in the "recent" list.
type RecentItem struct {
BlobRef *blobref.BlobRef `json:"blobref"`
// TODO: add a type with concrete type of time.Time that implements
// encode/decodeJSON hooks to RFC3339 strings
ModTime string `json:"modtime"` // RFC3339
Owner *blobref.BlobRef `json:"owner"`
}
// A WithAttrItem is an item returned from $searchRoot/camli/search/permanodeattr.
type WithAttrItem struct {
Permanode *blobref.BlobRef `json:"permanode"`
}
// A SignerPathsItem is an item returned from $searchRoot/camli/search/signerpaths.
type SignerPathsItem struct {
ClaimRef *blobref.BlobRef `json:"claimRef"`
BaseRef *blobref.BlobRef `json:"baseRef"`
Suffix string `json:"suffix"`
}
func thumbnailSize(r *http.Request) int {
return thumbnailSizeStr(r.FormValue("thumbnails"))
}
const (
minThumbSize = 25
defThumbSize = 50
maxThumbSize = 800
)
func thumbnailSizeStr(s string) int {
if s == "" {
return 0
}
if i, _ := strconv.Atoi(s); i >= minThumbSize && i <= maxThumbSize {
return i
}
return defThumbSize
}
func (sh *Handler) GetRecentPermanodes(req *RecentRequest) (*RecentResponse, error) {
ch := make(chan *Result)
errch := make(chan error)
go func() {
errch <- sh.index.GetRecentPermanodes(ch, sh.owner, req.n())
}()
dr := sh.NewDescribeRequest()
var recent []*RecentItem
for res := range ch {
dr.Describe(res.BlobRef, 2)
recent = append(recent, &RecentItem{
BlobRef: res.BlobRef,
Owner: res.Signer,
ModTime: time.Unix(res.LastModTime, 0).UTC().Format(time.RFC3339),
})
}
if err := <-errch; err != nil {
return nil, err
}
metaMap, err := dr.metaMap(req.thumbnailSize())
if err != nil {
return nil, err
}
res := &RecentResponse{
Recent: recent,
Meta: metaMap,
}
return res, nil
}
func serveJSONError(rw http.ResponseWriter, err error) {
ret := jsonMap()
ret["error"] = err.Error()
httputil.ReturnJSON(rw, ret)
}
func (sh *Handler) serveRecentPermanodes(rw http.ResponseWriter, req *http.Request) {
var rr RecentRequest
if err := rr.FromHTTP(req); err != nil {
serveJSONError(rw, err)
return
}
res, err := sh.GetRecentPermanodes(&rr)
if err != nil {
serveJSONError(rw, err)
return
}
httputil.ReturnJSON(rw, res)
}
func (sh *Handler) GetPermanodesWithAttr(req *WithAttrRequest) (*WithAttrResponse, error) {
ch := make(chan *blobref.BlobRef, buffered)
errch := make(chan error)
go func() {
errch <- sh.index.SearchPermanodesWithAttr(ch,
&PermanodeByAttrRequest{Attribute: req.Attr,
Query: req.Value,
Signer: req.Signer,
FuzzyMatch: req.Fuzzy,
MaxResults: req.N})
}()
dr := sh.NewDescribeRequest()
var withAttr []*WithAttrItem
for res := range ch {
dr.Describe(res, 2)
withAttr = append(withAttr, &WithAttrItem{
Permanode: res,
})
}
metaMap, err := dr.metaMap(0)
if err != nil {
return nil, err
}
res := &WithAttrResponse{
WithAttr: withAttr,
Meta: metaMap,
}
return res, nil
}
// servePermanodesWithAttr uses the indexer to search for the permanodes matching
// the request.
// The valid values for the "attr" key in the request (i.e the only attributes
// for a permanode which are actually indexed as such) are "tag" and "title".
func (sh *Handler) servePermanodesWithAttr(rw http.ResponseWriter, req *http.Request) {
var wr WithAttrRequest
if err := wr.FromHTTP(req); err != nil {
serveJSONError(rw, err)
return
}
res, err := sh.GetPermanodesWithAttr(&wr)
if err != nil {
serveJSONError(rw, err)
return
}
httputil.ReturnJSON(rw, res)
}
func (sh *Handler) serveClaims(rw http.ResponseWriter, req *http.Request) {
ret := jsonMap()
pn := blobref.Parse(req.FormValue("permanode"))
if pn == nil {
http.Error(rw, "Missing or invalid 'permanode' param", 400)
return
}
// TODO: rename GetOwnerClaims to GetClaims?
claims, err := sh.index.GetOwnerClaims(pn, sh.owner)
if err != nil {
log.Printf("Error getting claims of %s: %v", pn.String(), err)
} else {
sort.Sort(claims)
jclaims := jsonMapList()
for _, claim := range claims {
jclaim := jsonMap()
jclaim["blobref"] = claim.BlobRef.String()
jclaim["signer"] = claim.Signer.String()
jclaim["permanode"] = claim.Permanode.String()
jclaim["date"] = claim.Date.Format(time.RFC3339)
jclaim["type"] = claim.Type
if claim.Attr != "" {
jclaim["attr"] = claim.Attr
}
if claim.Value != "" {
jclaim["value"] = claim.Value
}
jclaims = append(jclaims, jclaim)
}
ret["claims"] = jclaims
}
httputil.ReturnJSON(rw, ret)
}
type DescribeRequest struct {
sh *Handler
mu sync.Mutex // protects following:
m MetaMap
done map[string]bool // blobref -> described
errs map[string]error // blobref -> error
wg *sync.WaitGroup // for load requests
}
type DescribedBlob struct {
Request *DescribeRequest `json:"-"`
BlobRef *blobref.BlobRef `json:"blobRef"`
MimeType string `json:"mimeType"`
CamliType string `json:"camliType"`
Size int64 `json:"size,"`
// if camliType "permanode"
Permanode *DescribedPermanode `json:"permanode,omitempty"`
// if camliType "file"
File *FileInfo `json:"file,omitempty"`
// if camliType "directory"
Dir *FileInfo `json:"dir,omitempty"`
Thumbnail string `json:"thumbnailSrc,omitempty"`
ThumbnailWidth int `json:"thumbnailWidth,omitempty"`
ThumbnailHeight int `json:"thumbnailHeight,omitempty"`
// Stub is set if this is not loaded, but referenced.
Stub bool `json:"-"`
}
// PermanodeFile returns the blobref path from this permanode to its
// File camliContent, else (nil, false)
func (b *DescribedBlob) PermanodeFile() (path []*blobref.BlobRef, fi *FileInfo, ok bool) {
if b == nil || b.Permanode == nil {
return
}
if contentRef := b.Permanode.Attr.Get("camliContent"); contentRef != "" {
if cdes := b.Request.DescribedBlobStr(contentRef); cdes != nil && cdes.File != nil {
return []*blobref.BlobRef{b.BlobRef, cdes.BlobRef}, cdes.File, true
}
}
return
}
func (b *DescribedBlob) DomID() string {
if b == nil {
return ""
}
return b.BlobRef.DomID()
}
func (b *DescribedBlob) Title() string {
if b == nil {
return ""
}
if b.Permanode != nil {
if t := b.Permanode.Attr.Get("title"); t != "" {
return t
}
if contentRef := b.Permanode.Attr.Get("camliContent"); contentRef != "" {
return b.Request.DescribedBlobStr(contentRef).Title()
}
}
if b.File != nil {
return b.File.FileName
}
if b.Dir != nil {
return b.Dir.FileName
}
return ""
}
func (b *DescribedBlob) Description() string {
if b == nil {
return ""
}
if b.Permanode != nil {
return b.Permanode.Attr.Get("description")
}
return ""
}
func (b *DescribedBlob) Members() []*DescribedBlob {
if b == nil {
return nil
}
m := make([]*DescribedBlob, 0)
if b.Permanode != nil {
for _, bstr := range b.Permanode.Attr["camliMember"] {
if br := blobref.Parse(bstr); br != nil {
m = append(m, b.PeerBlob(br))
}
}
}
return m
}
func (b *DescribedBlob) ContentRef() (br *blobref.BlobRef, ok bool) {
if b != nil && b.Permanode != nil {
if cref := b.Permanode.Attr.Get("camliContent"); cref != "" {
br = blobref.Parse(cref)
return br, br != nil
}
}
return
}
// Given a blobref string returns a Description or nil.
// dr may be nil itself.
func (dr *DescribeRequest) DescribedBlobStr(blobstr string) *DescribedBlob {
if dr == nil {
return nil
}
dr.mu.Lock()
defer dr.mu.Unlock()
return dr.m[blobstr]
}
// PeerBlob returns a DescribedBlob for the provided blobref.
//
// Unlike DescribedBlobStr, the returned DescribedBlob is never nil.
//
// If the blob was never loaded along with the the receiver (or if the
// receiver is nil), a stub DescribedBlob is returned with its Stub
// field set true.
func (b *DescribedBlob) PeerBlob(br *blobref.BlobRef) *DescribedBlob {
if b.Request == nil {
return &DescribedBlob{BlobRef: br, Stub: true}
}
b.Request.mu.Lock()
defer b.Request.mu.Unlock()
return b.peerBlob(br)
}
// version of PeerBlob when b.Request.mu is already held.
func (b *DescribedBlob) peerBlob(br *blobref.BlobRef) *DescribedBlob {
if peer, ok := b.Request.m[br.String()]; ok {
return peer
}
return &DescribedBlob{Request: b.Request, BlobRef: br, Stub: true}
}
// HasSecureLinkTo returns true if there's a valid link from this blob
// to the other blob. This is used in access control (hence the
// somewhat redundant "Secure" in the name) and should be paranoid
// against e.g. random user/attacker-control attributes making links
// to other blobs.
//
// TODO: don't linear scan here. rewrite this in terms of ResolvePrefixHop,
// passing down some policy perhaps? or maybe that's enough.
func (b *DescribedBlob) HasSecureLinkTo(other *blobref.BlobRef) bool {
if b == nil || other == nil {
return false
}
ostr := other.String()
if b.Permanode != nil {
if b.Permanode.Attr.Get("camliContent") == ostr {
return true
}
for _, mstr := range b.Permanode.Attr["camliMember"] {
if mstr == ostr {
return true
}
}
}
return false
}
func (b *DescribedBlob) isPermanode() bool {
return b.Permanode != nil
}
// returns a path relative to the UI handler.
//
// Locking: requires that DescribedRequest is done loading or that
// Request.mu is held (as it is from metaMap)
func (b *DescribedBlob) thumbnail(thumbSize int) (path string, width, height int, ok bool) {
if thumbSize <= 0 || !b.isPermanode() {
return
}
if b.Stub {
return "node.png", thumbSize, thumbSize, true
}
pnAttr := b.Permanode.Attr
if members := pnAttr["camliMember"]; len(members) > 0 {
return "folder.png", thumbSize, thumbSize, true
}
if content, ok := b.ContentRef(); ok {
peer := b.peerBlob(content)
if peer.File != nil {
if peer.File.IsImage() {
image := fmt.Sprintf("thumbnail/%s/%s?mw=%d&mh=%d", peer.BlobRef,
url.QueryEscape(peer.File.FileName), thumbSize, thumbSize)
// TODO: return the correct thumbSizes here,
// once we know from the indexer the
// dimensions (after correction for exif
// orientation). For now the thumbnails will
// all be stretched fat squares.
return image, thumbSize, thumbSize, true
}
// TODO: different thumbnails based on peer.File.MimeType.
return "file.png", thumbSize, thumbSize, true
}
if peer.Dir != nil {
return "folder.png", thumbSize, thumbSize, true
}
}
return "node.png", thumbSize, thumbSize, true
}
// TODO: delete this function
func (b *DescribedBlob) jsonMap() map[string]interface{} {
m := jsonMap()
m["blobRef"] = b.BlobRef.String()
if b.MimeType != "" {
m["mimeType"] = b.MimeType
}
if b.CamliType != "" {
m["camliType"] = b.CamliType
}
m["size"] = b.Size
if b.Permanode != nil {
m["permanode"] = b.Permanode.jsonMap()
}
if b.File != nil {
m["file"] = b.File
}
if b.Dir != nil {
m["dir"] = b.Dir
}
return m
}
type DescribedPermanode struct {
Attr url.Values `json:"attr"` // a map[string][]string
}
func (dp *DescribedPermanode) jsonMap() map[string]interface{} {
m := jsonMap()
am := jsonMap()
m["attr"] = am
for k, vv := range dp.Attr {
if len(vv) > 0 {
vl := make([]string, len(vv))
copy(vl[:], vv[:])
am[k] = vl
}
}
return m
}
// NewDescribeRequest returns a new DescribeRequest holding the state
// of blobs and their summarized descriptions. Use DescribeBlob
// one or more times before calling Result.
func (sh *Handler) NewDescribeRequest() *DescribeRequest {
return &DescribeRequest{
sh: sh,
m: make(MetaMap),
errs: make(map[string]error),
wg: new(sync.WaitGroup),
}
}
// Given a blobref and a few hex characters of the digest of the next hop, return the complete
// blobref of the prefix, if that's a valid next hop.
func (sh *Handler) ResolvePrefixHop(parent *blobref.BlobRef, prefix string) (child *blobref.BlobRef, err error) {
// TODO: this is a linear scan right now. this should be
// optimized to use a new database table of members so this is
// a quick lookup. in the meantime it should be in memcached
// at least.
if len(prefix) < 8 {
return nil, fmt.Errorf("Member prefix %q too small", prefix)
}
dr := sh.NewDescribeRequest()
dr.Describe(parent, 1)
res, err := dr.Result()
if err != nil {
return
}
des, ok := res[parent.String()]
if !ok {
return nil, fmt.Errorf("Failed to describe member %q in parent %q", prefix, parent)
}
if des.Permanode != nil {
if cr, ok := des.ContentRef(); ok && strings.HasPrefix(cr.Digest(), prefix) {
return cr, nil
}
for _, member := range des.Members() {
if strings.HasPrefix(member.BlobRef.Digest(), prefix) {
return member.BlobRef, nil
}
}
}
return nil, fmt.Errorf("Member prefix %q not found in %q", prefix, parent)
}
type DescribeError map[string]error
func (de DescribeError) Error() string {
var buf bytes.Buffer
for b, err := range de {
fmt.Fprintf(&buf, "%s: %v; ", b, err)
}
return fmt.Sprintf("Errors (%d) describing blobs: %s", len(de), buf.String())
}
// Result waits for all outstanding lookups to complete and
// returns the map of blobref (strings) to their described
// results. The returned error is non-nil if any errors
// occured, and will be of type DescribeError.
func (dr *DescribeRequest) Result() (desmap map[string]*DescribedBlob, err error) {
dr.wg.Wait()
// TODO: set "done" / locked flag, so no more DescribeBlob can
// be called.
if len(dr.errs) > 0 {
return dr.m, DescribeError(dr.errs)
}
return dr.m, nil
}
func (dr *DescribeRequest) metaMap(thumbSize int) (map[string]*DescribedBlob, error) {
// thumbSize of zero means to not include the thumbnails.
dr.wg.Wait()
dr.mu.Lock()
defer dr.mu.Unlock()
for k, err := range dr.errs {
// TODO: include all?
return nil, fmt.Errorf("error populating %s: %v", k, err)
}
m := make(map[string]*DescribedBlob)
for k, desb := range dr.m {
m[k] = desb
if src, w, h, ok := desb.thumbnail(thumbSize); ok {
desb.Thumbnail = src
desb.ThumbnailWidth = w
desb.ThumbnailHeight = h
}
}
return m, nil
}
func (dr *DescribeRequest) describedBlob(b *blobref.BlobRef) *DescribedBlob {
dr.mu.Lock()
defer dr.mu.Unlock()
bs := b.String()
if des, ok := dr.m[bs]; ok {
return des
}
des := &DescribedBlob{Request: dr, BlobRef: b}
dr.m[bs] = des
return des
}
func (dr *DescribeRequest) DescribeSync(br *blobref.BlobRef) (*DescribedBlob, error) {
dr.Describe(br, 1)
res, err := dr.Result()
if err != nil {
return nil, err
}
return res[br.String()], nil
}
func (dr *DescribeRequest) Describe(br *blobref.BlobRef, depth int) {
if depth <= 0 {
return
}
dr.mu.Lock()
defer dr.mu.Unlock()
if dr.done == nil {
dr.done = make(map[string]bool)
}
if dr.done[br.String()] {
return
}
dr.done[br.String()] = true
dr.wg.Add(1)
go func() {
defer dr.wg.Done()
dr.describeReally(br, depth)
}()
}
func (dr *DescribeRequest) addError(br *blobref.BlobRef, err error) {
if err == nil {
return
}
dr.mu.Lock()
defer dr.mu.Unlock()
// TODO: append? meh.
dr.errs[br.String()] = err
}
func (dr *DescribeRequest) describeReally(br *blobref.BlobRef, depth int) {
mime, size, err := dr.sh.index.GetBlobMimeType(br)
if err == os.ErrNotExist {
return
}
if err != nil {
dr.addError(br, err)
return
}
// TODO: convert all this in terms of
// DescribedBlob/DescribedPermanode/DescribedFile, not json
// maps. Then add JSON marhsallers to those types. Add tests.
des := dr.describedBlob(br)
des.setMimeType(mime)
des.Size = size
switch des.CamliType {
case "permanode":
des.Permanode = new(DescribedPermanode)
dr.populatePermanodeFields(des.Permanode, br, dr.sh.owner, depth)
case "file":
var err error
des.File, err = dr.sh.index.GetFileInfo(br)
if err != nil {
if os.IsNotExist(err) {
log.Printf("index.GetFileInfo(file %s) failed; index stale?", br)
} else {
dr.addError(br, err)
}
}
case "directory":
var err error
des.Dir, err = dr.sh.index.GetFileInfo(br)
if err != nil {
if os.IsNotExist(err) {
log.Printf("index.GetFileInfo(directory %s) failed; index stale?", br)
} else {
dr.addError(br, err)
}
}
}
}
func (sh *Handler) serveDescribe(rw http.ResponseWriter, req *http.Request) {
br := blobref.Parse(req.FormValue("blobref"))
if br == nil {
// TODO: make this be a 400 Bad Request HTTP type.
serveJSONError(rw, errors.New("input: missing or invalid 'blobref' param"))
return
}
dr := sh.NewDescribeRequest()
dr.Describe(br, 4)
metaMap, err := dr.metaMap(thumbnailSize(req))
if err != nil {
serveJSONError(rw, err)
return
}
httputil.ReturnJSON(rw, &DescribeResponse{metaMap})
}
func (sh *Handler) serveFiles(rw http.ResponseWriter, req *http.Request) {
ret := jsonMap()
defer httputil.ReturnJSON(rw, ret)
br := blobref.Parse(req.FormValue("wholedigest"))
if br == nil {
ret["error"] = "Missing or invalid 'wholedigest' param"
ret["errorType"] = "input"
return
}
files, err := sh.index.ExistingFileSchemas(br)
if err != nil {
ret["error"] = err.Error()
ret["errorType"] = "server"
return
}
strList := []string{}
for _, br := range files {
strList = append(strList, br.String())
}
ret["files"] = strList
return
}
func (dr *DescribeRequest) populatePermanodeFields(pi *DescribedPermanode, pn, signer *blobref.BlobRef, depth int) {
pi.Attr = make(url.Values)
attr := pi.Attr
claims, err := dr.sh.index.GetOwnerClaims(pn, signer)
if err != nil {
log.Printf("Error getting claims of %s: %v", pn.String(), err)
dr.addError(pn, fmt.Errorf("Error getting claims of %s: %v", pn.String(), err))
return
}
sort.Sort(claims)
claimLoop:
for _, cl := range claims {
switch cl.Type {
case "del-attribute":
if cl.Value == "" {
delete(attr, cl.Attr)
} else {
sl := attr[cl.Attr]
filtered := make([]string, 0, len(sl))
for _, val := range sl {
if val != cl.Value {
filtered = append(filtered, val)
}
}
attr[cl.Attr] = filtered
}
case "set-attribute":
delete(attr, cl.Attr)
fallthrough
case "add-attribute":
if cl.Value == "" {
continue
}
sl, ok := attr[cl.Attr]
if ok {
for _, exist := range sl {
if exist == cl.Value {
continue claimLoop
}
}
} else {
sl = make([]string, 0, 1)
attr[cl.Attr] = sl
}
attr[cl.Attr] = append(sl, cl.Value)
}
}
// If the content permanode is now known, look up its type
if content, ok := attr["camliContent"]; ok && len(content) > 0 {
cbr := blobref.Parse(content[len(content)-1])
dr.Describe(cbr, depth-1)
}
// Resolve children
if members, ok := attr["camliMember"]; ok {
for _, member := range members {
membr := blobref.Parse(member)
if membr != nil {
dr.Describe(membr, depth-1)
}
}
}
}
// TODO(mpl): we probably want to remove most of the calls on that one,
// or make it an error instead of a panic, when req is an unfiltered user input.
func mustGet(req *http.Request, param string) string {
v := req.FormValue(param)
if v == "" {
panic(fmt.Sprintf("missing required parameter %q", param))
}
return v
}
// TODO: convert this to have two *string pointers to error and errorType, and
// then add error and errType (embedded? if that works in Go 1.0) in all the other
// JSON response message struct types.
func setPanicError(m map[string]interface{}) {
p := recover()
if p == nil {
return
}
m["error"] = p.(string)
m["errorType"] = "input"
}
// SignerAttrValueResponse is the JSON response to $search/camli/search/signerattrvalue
type SignerAttrValueResponse struct {
Permanode *blobref.BlobRef `json:"permanode"`
Meta MetaMap `json:"meta"`
}
func (sh *Handler) serveSignerAttrValue(rw http.ResponseWriter, req *http.Request) {
done := false // get to success at the end?
ret := jsonMap()
defer func() {
if !done {
httputil.ReturnJSON(rw, ret)
}
}()
defer setPanicError(ret)
signer := blobref.MustParse(mustGet(req, "signer"))
attr := mustGet(req, "attr")
value := mustGet(req, "value")
pn, err := sh.index.PermanodeOfSignerAttrValue(signer, attr, value)
if err != nil {
ret["error"] = err.Error()
return
}
dr := sh.NewDescribeRequest()
dr.Describe(pn, 2)
metaMap, err := dr.metaMap(0)
if err != nil {
ret["error"] = err.Error()
return
}
done = true
httputil.ReturnJSON(rw, &SignerAttrValueResponse{
Permanode: pn,
Meta: metaMap,
})
}
// Unlike the index interface's EdgesTo method, the "edgesto" Handler
// here additionally filters out since-deleted permanode edges.
func (sh *Handler) serveEdgesTo(rw http.ResponseWriter, req *http.Request) {
ret := jsonMap()
defer httputil.ReturnJSON(rw, ret)
defer setPanicError(ret)
toRef := blobref.MustParse(mustGet(req, "blobref"))
toRefStr := toRef.String()
blobInfo := jsonMap()
ret[toRefStr] = blobInfo
jsonEdges := jsonMapList()
edges, err := sh.index.EdgesTo(toRef, nil)
if err != nil {
panic(err)
}
type mapOrError struct {
m map[string]interface{} // nil if no map
err error
}
resc := make(chan mapOrError)
verify := func(edge *Edge) {
fromStr := edge.From.String()
db, err := sh.NewDescribeRequest().DescribeSync(edge.From)
if err != nil {
resc <- mapOrError{err: err}
return
}
found := false
if db.Permanode != nil {
for attr, vv := range db.Permanode.Attr {
if IsBlobReferenceAttribute(attr) {
for _, v := range vv {
if v == toRefStr {
found = true
}
}
}
}
}
var em map[string]interface{}
if found {
em = jsonMap()
em["from"] = fromStr
em["fromType"] = "permanode"
}
resc <- mapOrError{m: em}
}
verifying := 0
for _, edge := range edges {
if edge.FromType == "permanode" {
verifying++
go verify(edge)
continue
}
em := jsonMap()
em["from"] = edge.From.String()
em["fromType"] = edge.FromType
jsonEdges = append(jsonEdges, em)
}
for i := 0; i < verifying; i++ {
res := <-resc
if res.err != nil {
panic(res.err) // caught and put in JSON response
}
if res.m != nil {
jsonEdges = append(jsonEdges, res.m)
}
}
blobInfo["edgesTo"] = jsonEdges
}
func (sh *Handler) serveSignerPaths(rw http.ResponseWriter, req *http.Request) {
// TODO: finish breaking this function up into a pure query half
// and an HTTP half which uses the pure querying half.
vs := req.FormValue("signer")
if vs == "" {
serveJSONError(rw, errors.New("missing required parameter: \"signer\""))
return
}
signer := blobref.Parse(vs)
if signer == nil {
serveJSONError(rw, fmt.Errorf("failed to parse signer blobref: %v", vs))
return
}
vt := req.FormValue("target")
if vt == "" {
serveJSONError(rw, errors.New("missing required parameter: \"target\""))
return
}
target := blobref.Parse(vt)
if target == nil {
serveJSONError(rw, fmt.Errorf("failed to parse target blobref: %v", vt))
return
}
paths, err := sh.index.PathsOfSignerTarget(signer, target)
if err != nil {
serveJSONError(rw, err)
return
}
var jpaths []*SignerPathsItem
for _, path := range paths {
jpaths = append(jpaths, &SignerPathsItem{
ClaimRef: path.Claim,
BaseRef: path.Base,
Suffix: path.Suffix,
})
}
dr := sh.NewDescribeRequest()
for _, path := range paths {
dr.Describe(path.Base, 2)
}
metaMap, err := dr.metaMap(0)
if err != nil {
serveJSONError(rw, err)
return
}
res := &SignerPathsResponse{
Paths: jpaths,
Meta: metaMap,
}
httputil.ReturnJSON(rw, res)
}
const camliTypePrefix = "application/json; camliType="
func (d *DescribedBlob) setMimeType(mime string) {
d.MimeType = mime
if strings.HasPrefix(mime, camliTypePrefix) {
d.CamliType = mime[len(camliTypePrefix):]
}
}