55 lines
1.2 KiB
Go
55 lines
1.2 KiB
Go
|
package cmd
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"log/slog"
|
||
|
|
"os"
|
||
|
|
|
||
|
|
"git.a71.su/Andrew71/pye/auth"
|
||
|
|
"github.com/golang-jwt/jwt/v5"
|
||
|
|
"github.com/spf13/cobra"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
verifyToken string
|
||
|
|
verifyFile string
|
||
|
|
)
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
verifyCmd.Flags().StringVarP(&verifyToken, "token", "t", "", "token to verify")
|
||
|
|
verifyCmd.MarkFlagRequired("token")
|
||
|
|
verifyCmd.Flags().StringVarP(&verifyFile, "file", "f", "", "file to use")
|
||
|
|
rootCmd.AddCommand(verifyCmd)
|
||
|
|
}
|
||
|
|
|
||
|
|
var verifyCmd = &cobra.Command{
|
||
|
|
Use: "verify",
|
||
|
|
Short: "Verify a JWT token",
|
||
|
|
Long: `Pass a JWT token and a path to PEM-encoded file with a public key
|
||
|
|
to verify whether it is legit.`,
|
||
|
|
Run: verifyFunc,
|
||
|
|
}
|
||
|
|
|
||
|
|
// TODO: Better name.
|
||
|
|
func verifyFunc(cmd *cobra.Command, args []string) {
|
||
|
|
if verifyToken == "" {
|
||
|
|
fmt.Println("Empty token supplied!")
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
var t *jwt.Token
|
||
|
|
var err error
|
||
|
|
if verifyFile == "" {
|
||
|
|
fmt.Println("No PEM file supplied, assuming local")
|
||
|
|
t, err = auth.VerifyLocalJWT(verifyToken)
|
||
|
|
} else {
|
||
|
|
key, err_k := os.ReadFile(verifyFile)
|
||
|
|
if err_k != nil {
|
||
|
|
slog.Error("error reading file", "error", err, "file", verifyFile)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
t, err = auth.VerifyJWT(verifyToken, key)
|
||
|
|
}
|
||
|
|
slog.Info("result", "token", t, "error", err, "ok", err == nil)
|
||
|
|
}
|