update durpcli

cmd/auth/generateToken.go

@@ -0,0 +1,104 @@
+package auth
+
+import (
+	"encoding/json"
+	"fmt"
+	"io/ioutil"
+	"net/http"
+	"strings"
+
+	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
+)
+
+type accessTokenResponse struct {
+	AccessToken string `json:"access_token"`
+	TokenType   string `json:"token_type"`
+	ExpiresIn   int    `json:"expires_in"`
+	IDToken     string `json:"id_token"`
+}
+
+var (
+	clientID  string
+	grantType string
+	url       string
+	username  string
+	password  string
+)
+
+var generateTokenCmd = &cobra.Command{
+	Use:   "generateToken",
+	Short: "Prints disk usage of the current directory",
+	Long:  ``,
+	Run: func(cmd *cobra.Command, args []string) {
+
+		if clientID == "" {
+			clientID = viper.GetViper().GetString("auth.clientID")
+		}
+		if grantType == "" {
+			grantType = viper.GetViper().GetString("auth.grantType")
+		}
+		if url == "" {
+			url = viper.GetViper().GetString("auth.url")
+		}
+		if username == "" {
+			username = viper.GetViper().GetString("auth.username")
+		}
+		if password == "" {
+			password = viper.GetViper().GetString("auth.password")
+		}
+
+		generateToken(clientID, grantType, url, username, password)
+	},
+}
+
+func init() {
+	generateTokenCmd.Flags().StringVarP(&clientID, "clientID", "c", "", "The ClientID")
+	generateTokenCmd.Flags().StringVarP(&grantType, "grantType", "g", "client_credentials", "The Grant Type")
+	generateTokenCmd.Flags().StringVarP(&url, "url", "", "", "Token URL")
+	generateTokenCmd.Flags().StringVarP(&username, "username", "u", "", "username")
+	generateTokenCmd.Flags().StringVarP(&password, "password", "p", "", "password")
+
+	AuthCmd.AddCommand(generateTokenCmd)
+}
+
+func generateToken(clientID string, grantType string, url string, username string, password string) {
+
+	formData := fmt.Sprintf("grant_type=%s&client_id=%s&username=%s&password=%s",
+		grantType, clientID, username, password)
+
+	client := &http.Client{}
+
+	req, err := http.NewRequest("POST", url, strings.NewReader(formData))
+	if err != nil {
+		fmt.Println("Error creating request:", err)
+		return
+	}
+
+	req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
+
+	resp, err := client.Do(req)
+	if err != nil {
+		fmt.Println("Error sending request:", err)
+		return
+	}
+	defer resp.Body.Close()
+
+	body, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		fmt.Println("Error reading response:", err)
+		return
+	}
+
+	var response accessTokenResponse
+	err = json.Unmarshal(body, &response)
+	if err != nil {
+		fmt.Println("Error parsing response:", err)
+		return
+	}
+
+	fmt.Println("Access_Token:", response.AccessToken)
+	fmt.Println("Token_Type:", response.TokenType)
+	fmt.Println("Expires_In:", response.ExpiresIn)
+	fmt.Println("ID_Token:", response.IDToken)
+}