15168027867

Committed 21 May 2025 04:53PM UTC coverage: 64.286% (-8.9%) from 73.233%

Build # 15168027867

Build Type

Pull #37

github

Committed by

web-flow

Commit Message

Merge a92a4ea91 into 18355eef5

Pull Request Pull Request #37: init package attempt

Run Details

0 of 65 new or added lines in 1 file covered. (0.0%)

342 of 532 relevant lines covered (64.29%)

0.69 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

0.0

/cortex/table_cortex_package.go

package cortex

import (
        "context"
        "fmt"

        "github.com/imroc/req/v3"
        "github.com/turbot/steampipe-plugin-sdk/v5/grpc/proto"
        "github.com/turbot/steampipe-plugin-sdk/v5/plugin"
)

type Package struct {
        DateCreated string `yaml:"dateCreated"`
        Id          string `yaml:"id"`
        Name        string `yaml:"name"`
        PackageType string `yaml:"packageType"`
        Version     string `yaml:"version"`
}

type CortexPackageRow struct {
        PackageTag  string
        PackageType string
        Name        string
        Version     string
}

func tableCortexPackage() *plugin.Table {
        return &plugin.Table{
                Name:        "cortex_packages",
                Description: "Cortex list package api.",
                List: &plugin.ListConfig{
                        Hydrate: listPackagesHydrator,
                        KeyColumns: []*plugin.KeyColumn{
                                {Name: "package_tag", Require: plugin.Required},
                        },
                },
                Columns: []*plugin.Column{
                        {Name: "package_tag", Type: proto.ColumnType_STRING, Description: "Package Tag."},
                        {Name: "package_type", Type: proto.ColumnType_STRING, Description: "Package Type."},
                        {Name: "name", Type: proto.ColumnType_STRING, Description: "Package Name."},
                        {Name: "version", Type: proto.ColumnType_STRING, Description: "Version."},
                },
        }
}

func listPackagesHydrator(ctx context.Context, d *plugin.QueryData, h *plugin.HydrateData) (interface{}, error) {
        logger := plugin.Logger(ctx)
        config := GetConfig(d.Connection)
        client := CortexHTTPClient(ctx, config)
        hydratorWriter := QueryDataWriter{d}
        packageTag := d.EqualsQuals["package_tag"].GetStringValue()
        logger.Info("listPackageHydrator", "packageTag", packageTag)
        return nil, listPackages(ctx, client, &hydratorWriter, packageTag)
}

func listPackages(ctx context.Context, client *req.Client, writer HydratorWriter, packageTag string) error {
        logger := plugin.Logger(ctx)
        var response []Package
        logger.Debug("listPackages")
        resp := client.
                Get("/api/v1/catalog/{tag}/packages").
                SetPathParam("tag", packageTag).
                // Options
                SetQueryParam("yaml", "false").
                Do(ctx)

        // Check for HTTP errors
        if resp.IsErrorState() {
                logger.Error("listPackages", "Status", resp.Status, "Body", resp.String())
                return fmt.Errorf("error from cortex API %s: %s", resp.Status, resp.String())
        }
        logger.Error("listPackages url", resp.Request.URL.String())

        // Unmarshal the response and check for unmarshal errors
        err := resp.Into(&response)
        if err != nil {
                logger.Error("listPackages", "Error", err)
                return err
        }

        // Stream each row from the response, stop if we hit the limit
        for _, result := range response {
                // send the item to steampipe
                row := CortexPackageRow{
                        Name:        result.Name,
                        PackageType: result.PackageType,
                        PackageTag:  packageTag,
                        Version:     result.Version,
                }
                writer.StreamListItem(ctx, row)
                // Context can be cancelled due to manual cancellation or the limit has been hit
                if writer.RowsRemaining(ctx) == 0 {
                        return nil
                }
        }

        return nil
}

1	package cortex
2
3	import (
4	"context"
5	"fmt"
6
7	"github.com/imroc/req/v3"
8	"github.com/turbot/steampipe-plugin-sdk/v5/grpc/proto"
9	"github.com/turbot/steampipe-plugin-sdk/v5/plugin"
10	)
11
12	type Package struct {
13	DateCreated string `yaml:"dateCreated"`
14	Id string `yaml:"id"`
15	Name string `yaml:"name"`
16	PackageType string `yaml:"packageType"`
17	Version string `yaml:"version"`
18	}
19
20	type CortexPackageRow struct {
21	PackageTag string
22	PackageType string
23	Name string
24	Version string
25	}
26
NEW 27	func tableCortexPackage() *plugin.Table {	×
NEW 28	return &plugin.Table{	×
NEW 29	Name: "cortex_packages",	×
NEW 30	Description: "Cortex list package api.",	×
NEW 31	List: &plugin.ListConfig{	×
NEW 32	Hydrate: listPackagesHydrator,	×
NEW 33	KeyColumns: []*plugin.KeyColumn{	×
NEW 34	{Name: "package_tag", Require: plugin.Required},	×
NEW 35	},	×
NEW 36	},	×
NEW 37	Columns: []*plugin.Column{	×
NEW 38	{Name: "package_tag", Type: proto.ColumnType_STRING, Description: "Package Tag."},	×
NEW 39	{Name: "package_type", Type: proto.ColumnType_STRING, Description: "Package Type."},	×
NEW 40	{Name: "name", Type: proto.ColumnType_STRING, Description: "Package Name."},	×
NEW 41	{Name: "version", Type: proto.ColumnType_STRING, Description: "Version."},	×
NEW 42	},	×
NEW 43	}	×
NEW 44	}	×
45
NEW 46	func listPackagesHydrator(ctx context.Context, d plugin.QueryData, h plugin.HydrateData) (interface{}, error) {	×
NEW 47	logger := plugin.Logger(ctx)	×
NEW 48	config := GetConfig(d.Connection)	×
NEW 49	client := CortexHTTPClient(ctx, config)	×
NEW 50	hydratorWriter := QueryDataWriter{d}	×
NEW 51	packageTag := d.EqualsQuals["package_tag"].GetStringValue()	×
NEW 52	logger.Info("listPackageHydrator", "packageTag", packageTag)	×
NEW 53	return nil, listPackages(ctx, client, &hydratorWriter, packageTag)	×
NEW 54	}	×
55
NEW 56	func listPackages(ctx context.Context, client *req.Client, writer HydratorWriter, packageTag string) error {	×
NEW 57	logger := plugin.Logger(ctx)	×
NEW 58	var response []Package	×
NEW 59	logger.Debug("listPackages")	×
NEW 60	resp := client.	×
NEW 61	Get("/api/v1/catalog/{tag}/packages").	×
NEW 62	SetPathParam("tag", packageTag).	×
NEW 63	// Options	×
NEW 64	SetQueryParam("yaml", "false").	×
NEW 65	Do(ctx)	×
NEW 66		×
NEW 67	// Check for HTTP errors	×
NEW 68	if resp.IsErrorState() {	×
NEW 69	logger.Error("listPackages", "Status", resp.Status, "Body", resp.String())	×
NEW 70	return fmt.Errorf("error from cortex API %s: %s", resp.Status, resp.String())	×
NEW 71	}	×
NEW 72	logger.Error("listPackages url", resp.Request.URL.String())	×
NEW 73		×
NEW 74	// Unmarshal the response and check for unmarshal errors	×
NEW 75	err := resp.Into(&response)	×
NEW 76	if err != nil {	×
NEW 77	logger.Error("listPackages", "Error", err)	×
NEW 78	return err	×
NEW 79	}	×
80
81	// Stream each row from the response, stop if we hit the limit
NEW 82	for _, result := range response {	×
NEW 83	// send the item to steampipe	×
NEW 84	row := CortexPackageRow{	×
NEW 85	Name: result.Name,	×
NEW 86	PackageType: result.PackageType,	×
NEW 87	PackageTag: packageTag,	×
NEW 88	Version: result.Version,	×
NEW 89	}	×
NEW 90	writer.StreamListItem(ctx, row)	×
NEW 91	// Context can be cancelled due to manual cancellation or the limit has been hit	×
NEW 92	if writer.RowsRemaining(ctx) == 0 {	×
NEW 93	return nil	×
NEW 94	}	×
95	}
96
NEW 97	return nil	×
98	}

Smirl / steampipe-plugin-cortex / 15168027867

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous