types

type MemoryOAuth2TokenStore struct {
	Tokens map[string]*oauth2.Token
	Config *oauth2.Config
}

type MemoryTokenStore struct {
	Tokens map[string]string
}

type OAuth2TokenStore interface {
	SetOAuth2Config(*oauth2.Config)
	Set(string, *oauth2.Token)
	Get(string) oauth2.TokenSource
}

type SSHSessionOauthHandler interface {
	HandleAuth(w http.ResponseWriter, r *http.Request)
	HandleAuthCallback(w http.ResponseWriter, r *http.Request)
	HandleUser(w http.ResponseWriter, r *http.Request)
	// Return nil to call SessionHandler directly
	DefaultCommand() []string
	SSHSessionCommandHandler(ssh.Session, *exec.Cmd) error
	SessionHandler(ssh.Session)
}

type TokenStore interface {
	Get(sessionID string) string
	Set(sessionID, value string)
}