multitenantStack/controllers/baseAPI.go

package controllers

import (
	"database/sql"
	"fmt"
	"multitenantStack/models"
	companydb "multitenantStack/services/companydb"
	"time"

	"github.com/astaxie/beego/orm"
)

// BaseAPIController operations for APIs
type BaseAPIController struct {
	BaseController
}

// SessionData holds all vars for a user from user company map
type SessionData struct {
	Email         string
	CompanyName   string
	CompanyUserID int
	Exp           time.Time
}

var jwtSession SessionData
var currentUser *models.CompanyUser
var companyDB *sql.DB
var o orm.Ormer

//var database sql.database

// Prepare parse all requests that come after this controller for valid auth
func (c *BaseAPIController) Prepare() {

	tokenString := c.Ctx.Request.Header.Get("X-JWTtoken")

	if tokenString == "" {
		c.ServeJSONError("No Token provided")
		return
	}

	token, db, err := companydb.GetDatabase(tokenString)
	if err != nil {
		c.ServeJSONError("Token invalid")
		return
	}

	jwtSession.Email = token["email"].(string)
	jwtSession.CompanyName = token["companyName"].(string)
	companyUserIDFloat := token["companyUserID"].(float64)
	jwtSession.CompanyUserID = int(companyUserIDFloat)
	ExpFloat := token["exp"].(float64)
	jwtSession.Exp = time.Unix(int64(ExpFloat), 0)

	companyDB = db
	o, err = orm.NewOrmWithDB("postgres", "company", companyDB)
	if err != nil {
		fmt.Println(err.Error())
		c.ServeJSONError("internal")
		return
	}

	companyUser, err := models.GetCompanyUserById(o, jwtSession.CompanyUserID)
	if err != nil {
		c.ServeJSONError("Error retrieving Company User")
		return
	}
	currentUser = companyUser
}