Initial commit

context.go

@@ -0,0 +1,26 @@
+package csrf
+
+import (
+	"context"
+)
+
+type contextKey int8
+
+const (
+	tokenContextKey contextKey = 1
+)
+
+func setCSRFToken(ctx context.Context, token string) context.Context {
+	newCtx := context.WithValue(ctx, tokenContextKey, token)
+	return newCtx
+}
+
+func GetCSRFToken(ctx context.Context) string {
+	value := ctx.Value(tokenContextKey)
+	strValue, ok := value.(string)
+	if !ok {
+		return ""
+	}
+
+	return strValue
+}