upstream-ory-keto/contrib/docs-code-samples/expand-api-display-access/99-cleanup/main.go

// Copyright © 2023 Ory Corp
// SPDX-License-Identifier: Apache-2.0

//go:build docscodesamples
// +build docscodesamples

package main

import (
	"context"

	"google.golang.org/grpc"
	"google.golang.org/grpc/credentials/insecure"
	"google.golang.org/protobuf/proto"

	rts "github.com/ory/keto/gen/go/ory/keto/relation_tuples/v1alpha2"
)

func purgeNamespace(nspace string) {
	rc, err := grpc.NewClient("127.0.0.1:4466", grpc.WithTransportCredentials(insecure.NewCredentials()))
	if err != nil {
		panic(err)
	}
	defer rc.Close()

	rClient := rts.NewReadServiceClient(rc)
	resp, err := rClient.ListRelationTuples(context.Background(), &rts.ListRelationTuplesRequest{
		Query: &rts.ListRelationTuplesRequest_Query{
			Namespace: nspace,
		},
	})
	if err != nil {
		panic(err)
	}

	wc, err := grpc.NewClient("127.0.0.1:4467", grpc.WithTransportCredentials(insecure.NewCredentials()))
	if err != nil {
		panic(err)
	}
	defer wc.Close()

	deltas := make([]*rts.RelationTupleDelta, len(resp.RelationTuples))
	for i, rt := range resp.RelationTuples {
		deltas[i] = &rts.RelationTupleDelta{
			Action:        rts.RelationTupleDelta_ACTION_DELETE,
			RelationTuple: proto.Clone(rt).(*rts.RelationTuple),
		}
	}

	wClient := rts.NewWriteServiceClient(wc)
	_, err = wClient.TransactRelationTuples(context.Background(), &rts.TransactRelationTuplesRequest{
		RelationTupleDeltas: deltas,
	})
	if err != nil {
		panic(err)
	}
}

func main() {
	purgeNamespace("directories")
	purgeNamespace("files")
}