feat(cdk): expose cache to components

.gitignore

@@ -32,4 +32,3 @@ override.tf.json
 *_override.tf
 *_override.tf.json
 *.terraform.lock.hcl
-

Makefile

@@ -148,7 +148,7 @@ fmt-check: ## Lists formatting issues
 
 .PHONY: imports-check
 imports-check: ## Lists imports issues
-	@test -z $(shell goimports -l $(shell go list -f {{.Dir}} ./... | grep -v proto))
+	@test -z $(shell goimports -l $(shell find . -type f -name '*.go' -not -path './vendor/*' -not -path './cli/cdk/go/proto/*'))
 
 .PHONY: run-api-example
 run-api-example: ## Run an API example like 'make run-api-example example=api/_examples/active-containers/main.go'
@@ -186,6 +186,20 @@ test-resources: ## *CI ONLY* Prepares CI test containers
 go-component-from := integration/test_resources/cdk/go-component/bin/go-component
 go-component-to := ~/.config/lacework/components/go-component/go-component
 
+.PHONY: cdk-go-component-ci
+cdk-go-component-ci: ## Creates a go-component for development
+	scripts/prepare_test_resources.sh go_component
+	mkdir -p $(shell dirname $(go-component-to))
+	echo '{"name":"go-component","description":"(dev-mode) A go-component for development","type":"CLI_COMMAND","artifacts":[],"breadcrumbs":{},"version":"0.0.0-dev"}' \
+		> $(shell dirname $(go-component-to))/.dev
+ifeq (x86_64, $(shell uname -m))
+	cp $(go-component-from)-$(shell uname -s | tr '[:upper:]' '[:lower:]')-amd64 $(go-component-to)
+else ifeq (arm64, $(shell uname -m))
+	cp $(go-component-from)-$(shell uname -s | tr '[:upper:]' '[:lower:]')-arm64 $(go-component-to)
+else
+	cp $(go-component-from)-$(shell uname -s | tr '[:upper:]' '[:lower:]')-386 $(go-component-to)
+endif
+
 .PHONY: cdk-go-component
 cdk-go-component: install-cli ## Creates a go-component for development
 	scripts/prepare_test_resources.sh go_component

cli/cmd/cache.go

@@ -204,20 +204,27 @@ type cliAsset struct {
 	ExpiresAt time.Time   `json:"expires_at"`
 }
 
-// WriteAssetToCache stores an asset with an expiration time
+// writeAssetToCache stores an asset with an expiration time and returns errors
 //
 // Simple Example: Having a struct named vulnReport
 //
 // ```go
 // cli.WriteAssetToCache("my-report", time.Now().Add(time.Hour * 1), vulnReport{Foo: "bar"})
 // ```
-func (c *cliState) WriteAssetToCache(key string, expiresAt time.Time, data interface{}) {
+// writeAssetToCache stores an asset with an expiration time
+//
+// Simple Example: Having a struct named vulnReport
+//
+// ```go
+// cli.WriteAssetToCache("my-report", time.Now().Add(time.Hour * 1), vulnReport{Foo: "bar"})
+// ```
+func (c *cliState) writeAssetToCache(key string, expiresAt time.Time, data interface{}) error {
 	if c.noCache {
-		return
+		return nil
 	}
 
 	if expiresAt.Before(time.Now()) {
-		return // avoid writing assets that are already expired
+		return nil // avoid writing assets that are already expired
 	}
 
 	c.Log.Debugw("saving asset",
@@ -226,7 +233,12 @@ func (c *cliState) WriteAssetToCache(key string, expiresAt time.Time, data inter
 		"data", data,
 		"expires_at", expiresAt,
 	)
-	err := c.Cache.Write(key, structToString(cliAsset{data, expiresAt}))
+	return c.Cache.Write(key, structToString(cliAsset{data, expiresAt}))
+}
+
+// WriteAssetToCache wraps WriteAssetToCacheErroring and squashes errors
+func (c *cliState) WriteAssetToCache(key string, expiresAt time.Time, data interface{}) {
+	err := c.writeAssetToCache(key, expiresAt, data)
 	if err != nil {
 		c.Log.Warnw("unable to write asset in cache",
 			"feature", "cache",

cli/cmd/cdk.go

@@ -62,6 +62,34 @@ func (c *cliState) GrpcTarget() string {
 	return fmt.Sprintf("localhost:%v", c.cdkServerPort)
 }
 
+func (c *cliState) ReadCache(ctx context.Context, in *cdk.ReadCacheRequest) (*cdk.ReadCacheResponse, error) {
+	if in.Key == "" {
+		return nil, errors.New("cache key must be supplied")
+	}
+
+	var data []byte
+	if !c.ReadCachedAsset(in.Key, &data) { // not expired
+		return &cdk.ReadCacheResponse{Hit: true, Data: data}, nil
+	}
+	return &cdk.ReadCacheResponse{
+		Hit:  false,
+		Data: nil,
+	}, nil
+}
+
+func (c *cliState) WriteCache(ctx context.Context, in *cdk.WriteCacheRequest) (*cdk.WriteCacheResult, error) {
+	if in.Key == "" {
+		return nil, errors.New("cache key must be supplied")
+	}
+
+	err := c.writeAssetToCache(in.Key, in.Expires.AsTime(), in.Data)
+	if err != nil {
+		msg := err.Error()
+		return &cdk.WriteCacheResult{Error: true, Message: msg}, nil
+	}
+	return &cdk.WriteCacheResult{Error: false, Message: ""}, nil
+}
+
 // Ping implements CDK.Ping
 func (c *cliState) Ping(ctx context.Context, in *cdk.PingRequest) (*cdk.PongReply, error) {
 	c.Log.Debugw("message", "from", "CDK/Ping", "component_name", in.GetComponentName())