gopay/internal/context/context.go

// Copyright 2024 perp (supernets)
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package context

import (
	"git.supernets.org/perp/gopay/internal/config"
	"git.supernets.org/perp/gopay/internal/database"
	"git.supernets.org/perp/gopay/internal/models"
	"github.com/gin-gonic/gin"
)

// Request context
type Context struct {
	*gin.Context
	*gin.Engine
	Db     *database.Database
	Config *config.Config
}

// Return a new Context
func New(cfg *config.Config, db *database.Database) *Context {
	// Create engine
	engine := gin.New()
	engine.Use(gin.Recovery())

	// Log level
	if cfg.Log.Level == "debug" {
		engine.Use(gin.Logger())
	}

	return &Context{
		Engine: engine,
		Db:     db,
		Config: cfg,
	}
}

// Return a Gin Context
func (ctx *Context) API(handler func(ctx *Context)) func(*gin.Context) {
	return func(gctx *gin.Context) {
		context := Context{
			Context: gctx,
			Db:      ctx.Db,
			Config:  ctx.Config,
		}
		handler(&context)
	}
}

// Return an Error
func (ctx *Context) Error(err string) models.Error {
	return models.Error{
		Type: err,
	}
}

// Return a Token
func (ctx *Context) Token(token string) models.Token {
	return models.Token{
		Token: token,
	}
}