From fd594506e2fd2a7a8df40c7a26bcaec668e681de Mon Sep 17 00:00:00 2001
From: Viacheslav Gonkivskyi <viacheslavgonkivskyi@gmail.com>
Date: Thu, 14 Sep 2023 15:10:06 +0300
Subject: [PATCH] feat(rpc): add admin cmd

---
 cmd/celestia/admin.go | 88 +++++++++++++++++++++++++++++++++++++++++++
 cmd/celestia/logs.go  | 48 -----------------------
 cmd/celestia/rpc.go   |  4 +-
 3 files changed, 90 insertions(+), 50 deletions(-)
 create mode 100644 cmd/celestia/admin.go
 delete mode 100644 cmd/celestia/logs.go

diff --git a/cmd/celestia/admin.go b/cmd/celestia/admin.go
new file mode 100644
index 0000000000..89f30f508e
--- /dev/null
+++ b/cmd/celestia/admin.go
@@ -0,0 +1,88 @@
+package main
+
+import (
+	"fmt"
+	"strings"
+
+	"github.com/filecoin-project/go-jsonrpc/auth"
+	"github.com/spf13/cobra"
+
+	"github.com/celestiaorg/celestia-node/cmd"
+)
+
+var nodeInfoCmd = &cobra.Command{
+	Use:   "node.info",
+	Args:  cobra.NoArgs,
+	Short: "Returns administrative information about the node.",
+	RunE: func(c *cobra.Command, args []string) error {
+		client, err := rpcClient(c.Context())
+		if err != nil {
+			return err
+		}
+		info, err := client.Node.Info(c.Context())
+		return printOutput(info, err, nil)
+	},
+}
+
+var logCmd = &cobra.Command{
+	Use:  cmd.LogLevelFlag,
+	Args: cobra.MinimumNArgs(1),
+	Short: "Allows to set log level for module to in format <module>:<level>" +
+		"`DEBUG, INFO, WARN, ERROR, DPANIC, PANIC, FATAL and their lower-case forms`.\n" +
+		"To set all modules to a particular level `*:<log.level>` should be passed",
+	RunE: func(c *cobra.Command, args []string) error {
+		client, err := rpcClient(c.Context())
+		if err != nil {
+			return err
+		}
+
+		for _, ll := range args {
+			params := strings.Split(ll, ":")
+			if len(params) != 2 {
+				return fmt.Errorf("cmd: %s arg must be in form <module>:<level>,"+
+					"e.g. pubsub:debug", cmd.LogLevelFlag)
+			}
+
+			if err = client.Node.LogLevelSet(c.Context(), params[0], params[1]); err != nil {
+				return err
+			}
+		}
+		return nil
+	},
+}
+
+var verifyCmd = &cobra.Command{
+	Use:   "permissions",
+	Args:  cobra.ExactArgs(1),
+	Short: "Returns the permissions assigned to the given token.",
+
+	RunE: func(c *cobra.Command, args []string) error {
+		client, err := rpcClient(c.Context())
+		if err != nil {
+			return err
+		}
+
+		perms, err := client.Node.AuthVerify(c.Context(), args[0])
+		return printOutput(perms, err, nil)
+	},
+}
+
+var authCmd = &cobra.Command{
+	Use:   "set.permissions",
+	Args:  cobra.MinimumNArgs(1),
+	Short: "Signs and returns a new token with the given permissions.",
+	RunE: func(c *cobra.Command, args []string) error {
+		client, err := rpcClient(c.Context())
+		if err != nil {
+			return err
+		}
+
+		perms := make([]auth.Permission, len(args))
+		for i, p := range args {
+			perms[i] = (auth.Permission)(p)
+		}
+
+		result, err := client.Node.AuthNew(c.Context(), perms)
+		return printOutput(result, err, nil)
+	},
+}
diff --git a/cmd/celestia/logs.go b/cmd/celestia/logs.go
deleted file mode 100644
index ac302ff6dd..0000000000
--- a/cmd/celestia/logs.go
+++ /dev/null
@@ -1,48 +0,0 @@
-package main
-
-import (
-	"fmt"
-	"strings"
-
-	"github.com/spf13/cobra"
-
-	"github.com/celestiaorg/celestia-node/cmd"
-)
-
-var logCmd = &cobra.Command{
-	Use:  cmd.LogLevelFlag,
-	Args: cobra.ExactArgs(1),
-	Short: "Allows to set log level for all modules to " +
-		"`DEBUG, INFO, WARN, ERROR, DPANIC, PANIC, FATAL and their lower-case forms`",
-
-	RunE: func(c *cobra.Command, args []string) error {
-		client, err := rpcClient(c.Context())
-		if err != nil {
-			return err
-		}
-		return client.Node.LogLevelSet(c.Context(), "*", args[0])
-	},
-}
-
-var logModuleCmd = &cobra.Command{
-	Use:   cmd.LogLevelModuleFlag,
-	Args:  cobra.MinimumNArgs(1),
-	Short: "Allows to set log level for a particular module in format <module>:<level>",
-	RunE: func(c *cobra.Command, args []string) error {
-		client, err := rpcClient(c.Context())
-		if err != nil {
-			return err
-		}
-		for _, ll := range args {
-			params := strings.Split(ll, ":")
-			if len(params) != 2 {
-				return fmt.Errorf("cmd: %s arg must be in form <module>:<level>,"+
-					"e.g. pubsub:debug", cmd.LogLevelModuleFlag)
-			}
-			if err = client.Node.LogLevelSet(c.Context(), params[0], params[1]); err != nil {
-				return err
-			}
-		}
-		return nil
-	},
-}
diff --git a/cmd/celestia/rpc.go b/cmd/celestia/rpc.go
index 1516211c3e..bce7b2b736 100644
--- a/cmd/celestia/rpc.go
+++ b/cmd/celestia/rpc.go
@@ -23,7 +23,7 @@ import (
 	"github.com/celestiaorg/celestia-node/state"
 )
 
-const authEnvKey = "CELESTIA_NODE_AUTH_TOKEN" //nolint:gosec
+const authEnvKey = "CELESTIA_NODE_AUTH_TOKEN"
 
 var requestURL string
 var authTokenFlag string
@@ -60,7 +60,7 @@ func init() {
 		false,
 		"Print JSON-RPC request along with the response",
 	)
-	rpcCmd.AddCommand(logCmd, logModuleCmd)
+	rpcCmd.AddCommand(nodeInfoCmd, logCmd, verifyCmd, authCmd)
 	rpcCmd.AddCommand(blobCmd)
 	rpcCmd.AddCommand(p2pCmd)
 	rpcCmd.AddCommand(dasCmd)