From 45c9d0c1204362ae172fdcd0ddc5685a0be4f9d4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Miroslav=20=C5=A0ediv=C3=BD?= <sedivy.miro@gmail.com>
Date: Mon, 15 Mar 2021 13:34:14 +0100
Subject: [PATCH] add API TOKEN to sessions.

---
 internal/config/session.go  |  7 +++++++
 internal/session/manager.go | 33 ++++++++++++++++++++++++++++-----
 2 files changed, 35 insertions(+), 5 deletions(-)

diff --git a/internal/config/session.go b/internal/config/session.go
index b7de9c84..06664165 100644
--- a/internal/config/session.go
+++ b/internal/config/session.go
@@ -7,6 +7,7 @@ import (
 
 type Session struct {
 	ImplicitHosting bool
+	APIToken        string
 }
 
 func (Session) Init(cmd *cobra.Command) error {
@@ -15,9 +16,15 @@ func (Session) Init(cmd *cobra.Command) error {
 		return err
 	}
 
+	cmd.PersistentFlags().String("api_token", "", "API token for interacting with external services")
+	if err := viper.BindPFlag("api_token", cmd.PersistentFlags().Lookup("api_token")); err != nil {
+		return err
+	}
+
 	return nil
 }
 
 func (s *Session) Set() {
 	s.ImplicitHosting = viper.GetBool("implicit_hosting")
+	s.APIToken = viper.GetString("api_token")
 }
diff --git a/internal/session/manager.go b/internal/session/manager.go
index eeacce0f..e4e82f0a 100644
--- a/internal/session/manager.go
+++ b/internal/session/manager.go
@@ -14,7 +14,7 @@ import (
 )
 
 func New(config *config.Session) *SessionManagerCtx {
-	return &SessionManagerCtx{
+	manager := &SessionManagerCtx{
 		logger:     log.With().Str("module", "session").Logger(),
 		config:     config,
 		host:       nil,
@@ -24,6 +24,23 @@ func New(config *config.Session) *SessionManagerCtx {
 		sessionsMu: sync.Mutex{},
 		emmiter:    events.New(),
 	}
+
+	// create API session
+	if config.APIToken != "" {
+		manager.apiSession = &SessionCtx{
+			id:      "API",
+			token:   config.APIToken,
+			manager: manager,
+			logger:  manager.logger.With().Str("session_id", "API").Logger(),
+			profile: types.MemberProfile{
+				Name:     "API Session",
+				IsAdmin:  true,
+				CanLogin: true,
+			},
+		}
+	}
+
+	return manager
 }
 
 type SessionManagerCtx struct {
@@ -35,6 +52,7 @@ type SessionManagerCtx struct {
 	sessions   map[string]*SessionCtx
 	sessionsMu sync.Mutex
 	emmiter    events.EventEmmiter
+	apiSession *SessionCtx
 }
 
 func (manager *SessionManagerCtx) Create(id string, profile types.MemberProfile) (types.Session, string, error) {
@@ -58,7 +76,7 @@ func (manager *SessionManagerCtx) Create(id string, profile types.MemberProfile)
 		id:      id,
 		token:   token,
 		manager: manager,
-		logger:  manager.logger.With().Str("id", id).Logger(),
+		logger:  manager.logger.With().Str("session_id", id).Logger(),
 		profile: profile,
 	}
 
@@ -124,11 +142,16 @@ func (manager *SessionManagerCtx) GetByToken(token string) (types.Session, bool)
 	id, ok := manager.tokens[token]
 	manager.sessionsMu.Unlock()
 
-	if !ok {
-		return nil, false
+	if ok {
+		return manager.Get(id)
 	}
 
-	return manager.Get(id)
+	// is API session
+	if manager.apiSession != nil && manager.apiSession.token == token {
+		return manager.apiSession, true
+	}
+
+	return nil, false
 }
 
 func (manager *SessionManagerCtx) List() []types.Session {