create authctl cli tool

Author: shiv-tyagi

cmd/authctl/main.go

@@ -0,0 +1,30 @@
+package main
+
+import (
+	"fmt"
+	"os"
+
+	"github.com/spf13/cobra"
+	"github.com/ubuntu/authd/cmd/authctl/user"
+)
+
+const cmdName = "authctl"
+
+var rootCmd = &cobra.Command{
+	Use:   fmt.Sprintf("%s COMMAND", cmdName),
+	Short: "CLI tool to interact with authd services",
+	Long:  "authctl is a CLI tool which can be used to interact with various authd services.",
+	Args:  cobra.NoArgs,
+	Run:   func(cmd *cobra.Command, args []string) {},
+}
+
+func init() {
+	rootCmd.AddCommand(user.UserCmd)
+}
+
+func main() {
+	if err := rootCmd.Execute(); err != nil {
+		fmt.Fprintf(os.Stderr, "Error while executing authctl '%s'\n", err)
+		os.Exit(1)
+	}
+}

cmd/authctl/user/disable.go

@@ -0,0 +1,35 @@
+package user
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/spf13/cobra"
+	"github.com/ubuntu/authd/internal/consts"
+	"github.com/ubuntu/authd/internal/proto/authd"
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/credentials/insecure"
+)
+
+var DisableCmd = &cobra.Command{
+	Use:   "disable",
+	Short: "Disable a user to log in",
+	Args:  cobra.ExactArgs(1),
+	RunE: func(cmd *cobra.Command, args []string) error {
+		fmt.Printf("Disabling user %s\n", args[0])
+		conn, err := grpc.NewClient("unix://"+consts.DefaultSocketPath, grpc.WithTransportCredentials(insecure.NewCredentials()))
+
+		if err != nil {
+			return err
+		}
+
+		client := authd.NewNSSClient(conn)
+		_, err = client.EnablePasswd(context.Background(), &authd.EnablePasswdRequest{Name: args[0]})
+
+		if err != nil {
+			return err
+		}
+
+		return nil
+	},
+}

cmd/authctl/user/enable.go

@@ -0,0 +1,35 @@
+package user
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/spf13/cobra"
+	"github.com/ubuntu/authd/internal/consts"
+	"github.com/ubuntu/authd/internal/proto/authd"
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/credentials/insecure"
+)
+
+var EnableCmd = &cobra.Command{
+	Use:   "enable",
+	Short: "Enable a user to log in",
+	Args:  cobra.ExactArgs(1),
+	RunE: func(cmd *cobra.Command, args []string) error {
+		fmt.Printf("Enabling user %s\n", args[0])
+		conn, err := grpc.NewClient("unix://"+consts.DefaultSocketPath, grpc.WithTransportCredentials(insecure.NewCredentials()))
+
+		if err != nil {
+			return err
+		}
+
+		client := authd.NewNSSClient(conn)
+		_, err = client.EnablePasswd(context.Background(), &authd.EnablePasswdRequest{Name: args[0]})
+
+		if err != nil {
+			return err
+		}
+
+		return nil
+	},
+}

cmd/authctl/user/user.go

@@ -0,0 +1,17 @@
+package user
+
+import (
+	"github.com/spf13/cobra"
+)
+
+var UserCmd = &cobra.Command{
+	Use:   "user",
+	Short: "Commands retaled to working with users",
+	Args:  cobra.NoArgs,
+	Run:   func(cmd *cobra.Command, args []string) {},
+}
+
+func init() {
+	UserCmd.AddCommand(DisableCmd)
+	UserCmd.AddCommand(EnableCmd)
+}