package common

import (
	"context"
	"database/sql"
)

type Key uint8

const (
	keyDB           Key = 0
	keyDebug        Key = 1
	keyAuthor       Key = 2
	KeyCopaingState Key = 3
)

func SetDB(ctx context.Context, db *sql.DB) context.Context {
	return context.WithValue(ctx, keyDB, db)
}

func GetDB(ctx context.Context) *sql.DB {
	raw := ctx.Value(keyDB)
	if raw == nil {
		return nil
	}
	return raw.(*sql.DB)
}

func SetDebug(ctx context.Context, b bool) context.Context {
	return context.WithValue(ctx, keyDebug, b)
}

func IsDebug(ctx context.Context) bool {
	raw := ctx.Value(keyDebug)
	if raw == nil {
		return false
	}
	return raw.(bool)
}

func SetAuthor(ctx context.Context, s string) context.Context {
	return context.WithValue(ctx, keyAuthor, s)
}

func GetAuthor(ctx context.Context) string {
	raw := ctx.Value(keyAuthor)
	if raw == nil {
		return ""
	}
	return raw.(string)
}