search.go

package onedrive

import (
	"context"
)

type EntityType string

// DriveItem declares to search drive items
const DriveItem EntityType = "driveItem"

// SearchService allows searching almost any resource
type SearchService interface {
	Search(ctx context.Context, query SearchRequest) (*SearchResponse, error)
}

type searchService struct {
	client *OneDrive
}

type queryPayload struct {
	Query string `json:"queryString"`
}

type SearchRequests struct {
	Requests []SearchRequest `json:"requests"`
}

type SearchRequest struct {
	EntityTypes []EntityType  `json:"entityTypes"`
	Query       SearchOptions `json:"query"`
	From        int           `json:"from"`
	Size        int           `json:"size"`
}

type ResponseValue struct {
	RecordContainers []HitRecord `json:"hitsContainers"`
}

type HitRecord struct {
	Records []Resource `json:"hits"`
}

type Resource struct {
	Detail ResourceDetail `json:"resource"`
}

type ResourceDetail struct {
	Name                 string          `json:"name"`
	LastModifiedDateTime string          `json:"lastModifiedDateTime"`
	ParentReference      ParentReference `json:"parentReference"`
}

type ParentReference struct {
	SiteID      string      `json:"siteId"`
	DriveID     string      `json:"driveId"`
	Identifiers Identifiers `json:"sharepointIds"`
}

type Identifiers struct {
	ListID     string `json:"listId"`
	ListItemID string `json:"listItemUniqueId"`
}

type SearchResponse struct {
	Entries []ResponseValue `json:"value"`
}

func (s searchService) Search(ctx context.Context, query SearchRequest) (*SearchResponse, error) {
	body := SearchRequests{Requests: []SearchRequest{query}}
	req, err := s.client.newRequest("POST", "/search/query", nil, body)
	if err != nil {
		return nil, err
	}

	searchResp := new(SearchResponse)
	_, err = s.client.do(req.WithContext(ctx), searchResp)
	if err != nil {
		return nil, err
	}

	return searchResp, nil
}