es.go

package cmds

import (
	"context"
	"encoding/json"
	"fmt"
	"github.com/elastic/go-elasticsearch/v8"
	layers2 "github.com/go-go-golems/escuse-me/pkg/cmds/layers"
	"github.com/go-go-golems/glazed/pkg/cmds"
	"github.com/go-go-golems/glazed/pkg/cmds/layers"
	"github.com/go-go-golems/glazed/pkg/cmds/parameters"
	"github.com/go-go-golems/glazed/pkg/middlewares"
	"github.com/go-go-golems/glazed/pkg/middlewares/row"
	"github.com/go-go-golems/glazed/pkg/settings"
	"github.com/go-go-golems/glazed/pkg/types"
	"github.com/pkg/errors"
	"io"
)

type InfoCommand struct {
	*cmds.CommandDescription
}

var _ cmds.GlazeCommand = &InfoCommand{}

func NewInfoCommand() (*InfoCommand, error) {
	glazedParameterLayer, err := settings.NewGlazedParameterLayers()
	if err != nil {
		return nil, errors.Wrap(err, "could not create Glazed parameter layer")
	}

	esParameterLayer, err := layers2.NewESParameterLayer()
	if err != nil {
		return nil, errors.Wrap(err, "could not create ES parameter layer")
	}

	return &InfoCommand{
		CommandDescription: cmds.NewCommandDescription(
			"info",
			cmds.WithShort("Prints information about the ES server"),
			cmds.WithFlags(
				parameters.NewParameterDefinition(
					"full",
					parameters.ParameterTypeBool,
					parameters.WithHelp("Prints the full version response"),
					parameters.WithDefault(false),
				),
			),
			cmds.WithLayersList(
				glazedParameterLayer,
				esParameterLayer,
			),
		),
	}, nil
}

type InfoSettings struct {
	Full bool `glazed.parameter:"full"`
}

func (i *InfoCommand) RunIntoGlazeProcessor(
	ctx context.Context,
	parsedLayers *layers.ParsedLayers,
	gp middlewares.Processor,
) error {
	s := &InfoSettings{}
	err := parsedLayers.InitializeStruct(layers.DefaultSlug, s)
	if err != nil {
		return err
	}

	es, err := layers2.NewESClientFromParsedLayers(parsedLayers)
	if err != nil {
		return err
	}

	gp.(*middlewares.TableProcessor).AddRowMiddleware(
		row.NewReorderColumnOrderMiddleware(
			[]string{"client_version", "version", "cluster_name"},
		),
	)

	clientVersion := elasticsearch.Version
	res, err := es.Info()
	if err != nil {
		return err
	}

	defer func(Body io.ReadCloser) {
		err := Body.Close()
		if err != nil {
			fmt.Println(err)
		}
	}(res.Body)

	// read all body
	body, err := io.ReadAll(res.Body)
	if err != nil {
		return err
	}

	body_ := types.NewRow()
	err = json.Unmarshal(body, &body_)
	if err != nil {
		return err
	}

	if !s.Full {
		version_, ok := body_.Get("version")
		if !ok {
			return errors.New("could not find version in response")
		}
		version := version_.(map[string]interface{})
		body_.Set("version", version["number"])
	}
	body_.Set("client_version", clientVersion)

	err = gp.AddRow(ctx, body_)
	if err != nil {
		return err
	}
	return nil
}