add index create,delete, enhance index show, catch empty index arg

pkg/es/index.go

@@ -20,11 +20,39 @@ import (
 	"context"
 	"fmt"
 	"log/slog"
+	"strconv"
+	"time"
 
 	"codeberg.org/scip/esctl/pkg/cfg"
+	"github.com/elastic/go-elasticsearch/v9/typedapi/esdsl"
 	"github.com/elastic/go-elasticsearch/v9/typedapi/types/enums/healthstatus"
 )
 
+type Settings struct {
+	wait_for_active_shards string
+	number_of_shards       int
+	number_of_replicas     int
+}
+
+// used for completion
+func IndexNames(conf *cfg.Config) ([]string, error) {
+	res, err := conf.DefaultCluster.ES.Cat.Indices().
+		Header("content-type", "application/json").
+		Header("accept", "application/json").
+		Do(context.Background())
+
+	if err != nil {
+		return nil, fmt.Errorf("Error getting indicies: %s", err)
+	}
+
+	indices := make([]string, len(res))
+	for idx, index := range res {
+		indices[idx] = *index.Index
+	}
+
+	return indices, nil
+}
+
 func IndexList(conf *cfg.Config) error {
 	cat := conf.DefaultCluster.ES.Cat.Indices().
 		// we need to add custom request headers, required for older ES instances
@@ -70,6 +98,10 @@ func IndexList(conf *cfg.Config) error {
 }
 
 func IndexShow(conf *cfg.Config, index string) error {
+	if index == "" {
+		return fmt.Errorf("no index specified")
+	}
+
 	res, err := conf.DefaultCluster.ES.Indices.Get(index).
 		// we need to add custom request headers, required for older ES instances
 		Header("content-type", "application/json").
@@ -81,7 +113,73 @@ func IndexShow(conf *cfg.Config, index string) error {
 
 	slog.Debug("ES result", "index", res)
 
-	// FIXME: add table printer like SnapshotShow
+	table := NewTable(2, 5)
+	table.Addheaders("field", "value")
+
+	ts, err := strconv.ParseInt(res[index].Settings.Index.CreationDate.(string), 10, 64)
+	if err != nil {
+		ts = 0
+	}
+
+	created := time.Unix(ts/1000, 0)
+
+	table.entries = [][]string{
+		{"name", index},
+		{"replicas", *res[index].Settings.Index.NumberOfReplicas},
+		{"shards", *res[index].Settings.Index.NumberOfShards},
+		{"created", created.Format("2006-01-02 15:04:05")},
+		{"uuid", *res[index].Settings.Index.Uuid},
+	}
+
+	table.PrintMarkdown()
+
+	return nil
+}
+
+func IndexCreate(conf *cfg.Config, index string) error {
+	if index == "" {
+		return fmt.Errorf("no index specified")
+	}
+
+	settings := esdsl.NewIndexSettings()
+
+	create := conf.DefaultCluster.ES.Indices.Create(index).
+		Header("content-type", "application/json").
+		Header("accept", "application/json")
+
+	if conf.Wait {
+		create.WaitForActiveShards("all")
+	}
+
+	if conf.Shards > 0 {
+		settings = settings.NumberOfShards(strconv.Itoa(conf.Shards))
+	}
+	if conf.Replicas > 0 {
+		settings = settings.NumberOfReplicas(strconv.Itoa(conf.Replicas))
+	}
+
+	_, err := create.Settings(settings).
+		Do(context.Background())
+
+	if err != nil {
+		return fmt.Errorf("failed to create index: %s", err)
+	}
+
+	return nil
+}
+
+func IndexDelete(conf *cfg.Config, index string) error {
+	if index == "" {
+		return fmt.Errorf("no index specified")
+	}
+
+	_, err := conf.DefaultCluster.ES.Indices.Delete(index).
+		Header("content-type", "application/json").
+		Header("accept", "application/json").
+		Do(context.Background())
+	if err != nil {
+		return fmt.Errorf("failed to delete index: %s", err)
+	}
 
 	return nil
 }