feat(referrers): shared index API (#441)

Signed-off-by: Miguel Martinez Trivino <[email protected]>

Author: migmartri

app/cli/cmd/referrer_discover.go

@@ -24,12 +24,21 @@ import (
 
 func newReferrerDiscoverCmd() *cobra.Command {
 	var digest, kind string
+	var fromPublicIndex bool
 
 	cmd := &cobra.Command{
 		Use:   "discover",
 		Short: "(Preview) inspect pieces of evidence or artifacts stored through Chainloop",
 		RunE: func(cmd *cobra.Command, args []string) error {
-			res, err := action.NewReferrerDiscover(actionOpts).Run(context.Background(), digest, kind)
+			var res *action.ReferrerItem
+			var err error
+
+			if fromPublicIndex {
+				res, err = action.NewReferrerDiscoverPublicIndex(actionOpts).Run(context.Background(), digest, kind)
+			} else {
+				res, err = action.NewReferrerDiscoverPrivate(actionOpts).Run(context.Background(), digest, kind)
+			}
+
 			if err != nil {
 				return err
 			}
@@ -43,6 +52,8 @@ func newReferrerDiscoverCmd() *cobra.Command {
 	err := cmd.MarkFlagRequired("digest")
 	cobra.CheckErr(err)
 	cmd.Flags().StringVarP(&kind, "kind", "k", "", "optional kind of the referrer, used to disambiguate between multiple referrers with the same digest")
+	cobra.CheckErr(err)
+	cmd.Flags().BoolVar(&fromPublicIndex, "public", false, "discover from public shared index instead of your organizations'")
 
 	return cmd
 }

app/cli/internal/action/referrer_discover.go

@@ -25,6 +25,9 @@ import (
 type ReferrerDiscover struct {
 	cfg *ActionsOpts
 }
+type ReferrerDiscoverPublic struct {
+	cfg *ActionsOpts
+}
 
 type ReferrerItem struct {
 	Digest       string          `json:"digest"`
@@ -35,13 +38,29 @@ type ReferrerItem struct {
 	References   []*ReferrerItem `json:"references"`
 }
 
-func NewReferrerDiscover(cfg *ActionsOpts) *ReferrerDiscover {
+func NewReferrerDiscoverPrivate(cfg *ActionsOpts) *ReferrerDiscover {
 	return &ReferrerDiscover{cfg}
 }
 
 func (action *ReferrerDiscover) Run(ctx context.Context, digest, kind string) (*ReferrerItem, error) {
 	client := pb.NewReferrerServiceClient(action.cfg.CPConnection)
-	resp, err := client.Discover(ctx, &pb.ReferrerServiceDiscoverRequest{
+	resp, err := client.DiscoverPrivate(ctx, &pb.ReferrerServiceDiscoverPrivateRequest{
+		Digest: digest, Kind: kind,
+	})
+	if err != nil {
+		return nil, err
+	}
+
+	return pbReferrerItemToAction(resp.Result), nil
+}
+
+func NewReferrerDiscoverPublicIndex(cfg *ActionsOpts) *ReferrerDiscoverPublic {
+	return &ReferrerDiscoverPublic{cfg}
+}
+
+func (action *ReferrerDiscoverPublic) Run(ctx context.Context, digest, kind string) (*ReferrerItem, error) {
+	client := pb.NewReferrerServiceClient(action.cfg.CPConnection)
+	resp, err := client.DiscoverPublicShared(ctx, &pb.DiscoverPublicSharedRequest{
 		Digest: digest, Kind: kind,
 	})
 	if err != nil {