mirror of
https://github.com/sonr-io/sonr.git
synced 2026-08-02 17:31:39 +00:00
Prad NukalaandGitHub
bf94277b0f
- **refactor: refactor DID module types and move to controller package** - **refactor: move controller creation and resolution logic to keeper** - **refactor: update imports to reflect controller package move** - **refactor: update protobuf definitions for DID module** - **docs: update proto README to reflect changes** - **refactor: move hway to gateway, update node modules, and refactor pkl generation** - **build: update pkl-gen task to use new pkl file paths** - **refactor: refactor DWN WASM build and deployment process** - **refactor: refactor DID controller implementation to use account-based storage** - **refactor: move DID controller interface to base file and update implementation** - **chore: migrate to google protobuf** - **feat: Add v0.52.0 Interfaces for Acc Abstraction** - **refactor: replace public_key with public_key_hex in Assertion message** - **refactor: remove unused PubKey, JSONWebKey, and RawKey message types and related code**
69 lines
2.0 KiB
Go
69 lines
2.0 KiB
Go
package ctx
|
|
|
|
import (
|
|
"github.com/go-webauthn/webauthn/protocol"
|
|
"github.com/labstack/echo/v4"
|
|
"github.com/segmentio/ksuid"
|
|
)
|
|
|
|
// CookieKey is a type alias for string.
|
|
type CookieKey string
|
|
|
|
const (
|
|
// CookieKeySessionID is the key for the session ID cookie.
|
|
CookieKeySessionID CookieKey = "session.id"
|
|
|
|
// CookieKeySessionChal is the key for the session challenge cookie.
|
|
CookieKeySessionChal CookieKey = "session.chal"
|
|
|
|
// CookieKeySonrAddr is the key for the Sonr address cookie.
|
|
CookieKeySonrAddr CookieKey = "sonr.addr"
|
|
|
|
// CookieKeySonrDID is the key for the Sonr DID cookie.
|
|
CookieKeySonrDID CookieKey = "sonr.did"
|
|
|
|
// CookieKeyVaultCID is the key for the Vault CID cookie.
|
|
CookieKeyVaultCID CookieKey = "vault.cid"
|
|
|
|
// CookieKeyVaultSchema is the key for the Vault schema cookie.
|
|
CookieKeyVaultSchema CookieKey = "vault.schema"
|
|
)
|
|
|
|
// String returns the string representation of the CookieKey.
|
|
func (c CookieKey) String() string {
|
|
return string(c)
|
|
}
|
|
|
|
// GetSessionID returns the session ID from the cookies.
|
|
func GetSessionID(c echo.Context) string {
|
|
// Attempt to read the session ID from the "session" cookie
|
|
sessionID, err := ReadCookie(c, CookieKeySessionID)
|
|
if err != nil {
|
|
// Generate a new KSUID if the session cookie is missing or invalid
|
|
WriteCookie(c, CookieKeySessionID, ksuid.New().String())
|
|
}
|
|
return sessionID
|
|
}
|
|
|
|
// GetSessionChallenge returns the session challenge from the cookies.
|
|
func GetSessionChallenge(c echo.Context) (*protocol.URLEncodedBase64, error) {
|
|
// TODO: Implement a way to regenerate the challenge if it is invalid.
|
|
chal := new(protocol.URLEncodedBase64)
|
|
// Attempt to read the session challenge from the "session" cookie
|
|
sessionChal, err := ReadCookie(c, CookieKeySessionChal)
|
|
if err != nil {
|
|
// Generate a new challenge if the session cookie is missing or invalid
|
|
ch, errb := protocol.CreateChallenge()
|
|
if errb != nil {
|
|
return nil, err
|
|
}
|
|
WriteCookie(c, CookieKeySessionChal, ch.String())
|
|
return &ch, nil
|
|
}
|
|
err = chal.UnmarshalJSON([]byte(sessionChal))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return chal, nil
|
|
}
|