basic common app packages

jwt/jwt.go

@@ -0,0 +1,48 @@
+// Copyright 2024 Patial Tech (Ankit Patial).
+// All rights reserved.
+
+package jwt
+
+import (
+	"crypto/ed25519"
+	"errors"
+	"log"
+	"maps"
+	"time"
+
+	"github.com/golang-jwt/jwt/v5"
+)
+
+func Sign(key ed25519.PrivateKey, claims map[string]any, issuer string, d time.Duration) (string, error) {
+	cl := jwt.MapClaims{
+		"iss": issuer,
+		"iat": jwt.NewNumericDate(time.Now().UTC()),
+		"exp": jwt.NewNumericDate(time.Now().Add(d)),
+	}
+	maps.Copy(cl, claims)
+
+	t := jwt.NewWithClaims(jwt.SigningMethodEdDSA, cl)
+	return t.SignedString(key)
+}
+
+func Parse(key ed25519.PrivateKey, tokenString string, issuer string) (jwt.MapClaims, error) {
+	token, err := jwt.Parse(
+		tokenString,
+		func(token *jwt.Token) (any, error) {
+			return key.Public(), nil
+		},
+		jwt.WithValidMethods([]string{jwt.SigningMethodEdDSA.Alg()}),
+		jwt.WithIssuer(issuer),
+		jwt.WithIssuedAt(),
+		jwt.WithExpirationRequired(),
+	)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	if claims, ok := token.Claims.(jwt.MapClaims); ok {
+		return claims, nil
+	} else {
+		return nil, errors.New("no claims found")
+	}
+}

jwt/jwt_test.go

@@ -0,0 +1,57 @@
+package jwt
+
+import (
+	"fmt"
+	"testing"
+	"time"
+
+	"code.patial.tech/go/appcore/crypto"
+)
+
+/*
+-----BEGIN PUBLIC KEY-----
+MCowBQYDK2VwAyEA9JTCYl3OQwuVTSf0PsBkmgJSt7e5Tbk3jKnB90vDqXA=
+-----END PUBLIC KEY-----
+
+-----BEGIN PRIVATE KEY-----
+MC4CAQAwBQYDK2VwBCIEIMMkYUKJ9P0gp+Rm9mR4i0KUBT9nFUzxzxjH7sC0xq/F
+-----END PRIVATE KEY-----
+*/
+
+func TestSign2(t *testing.T) {
+	key, err := crypto.ParseEdPrivateKey([]byte(`-----BEGIN PRIVATE KEY-----
+MC4CAQAwBQYDK2VwBCIEIMMkYUKJ9P0gp+Rm9mR4i0KUBT9nFUzxzxjH7sC0xq/F
+-----END PRIVATE KEY-----`))
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	s, err := Sign(key, map[string]any{"name": "ankit", "age": 25}, "blackdu", time.Second)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	println(s)
+}
+
+func TestExprired(t *testing.T) {
+	privKey, err := crypto.ParseEdPrivateKey([]byte(`-----BEGIN PRIVATE KEY-----
+MC4CAQAwBQYDK2VwBCIEIMMkYUKJ9P0gp+Rm9mR4i0KUBT9nFUzxzxjH7sC0xq/F
+-----END PRIVATE KEY-----`))
+	if err != nil {
+		t.Error(err)
+	}
+
+	// payload := "eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJhZ2UiOjI1LCJleHAiOjE3NDQxMjAxNzUsImlhdCI6MTc0NDEyMDE3NCwiaXNzIjoiYmxhY2tkdSIsIm5hbWUiOiJhbmtpdCJ9.W6OQHMRdcRiPS398p8u0vLjpq34oxYPDengillXSFEDXJVXOkzl0ncCpju0yuMOhrQLRRG0EJLKfoFcAxsbpDA"
+	payload, err := Sign(privKey, map[string]any{"name": "ankit", "age": 25}, "blackdu", time.Second)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	claims, err := Parse(privKey, payload, "blackdu")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	fmt.Printf("%v", claims)
+}