// Copyright 2013 Beego Authors
// Copyright 2014 The Macaron Authors
// Copyright 2024 The Forgejo Authors
//
// 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 memcache

import (
	"fmt"
	"strings"

	"github.com/bradfitz/gomemcache/memcache"

	"code.forgejo.org/go-chi/cache"
)

// Cacher represents a memcache cache adapter implementation.
type Cacher struct {
	c *memcache.Client
}

func NewItem(key string, data []byte, expire int32) *memcache.Item {
	return &memcache.Item{
		Key:        key,
		Value:      data,
		Expiration: expire,
	}
}

// Put puts value into cache with key and expire time.
// If expired is 0, it lives forever.
func (c *Cacher) Put(key string, val any, expire int64) error {
	var v []byte
	switch val := val.(type) {
	case string:
		v = []byte(val)
	case []byte:
		v = val
	default:
		// Best effort.
		v = []byte(fmt.Sprint(val))
	}

	return c.c.Set(NewItem(key, v, int32(expire)))
}

// Get gets cached value by given key.
func (c *Cacher) Get(key string) any {
	item, err := c.c.Get(key)
	if err != nil {
		return nil
	}
	return string(item.Value)
}

// Delete deletes cached value by given key.
func (c *Cacher) Delete(key string) error {
	if err := c.c.Delete(key); err != nil && err != memcache.ErrCacheMiss {
		return err
	}
	return nil
}

// Incr increases cached int-type value by given key as a counter.
func (c *Cacher) Incr(key string) error {
	_, err := c.c.Increment(key, 1)
	return err
}

// Decr decreases cached int-type value by given key as a counter.
func (c *Cacher) Decr(key string) error {
	_, err := c.c.Decrement(key, 1)
	return err
}

// IsExist returns true if cached value exists.
func (c *Cacher) IsExist(key string) bool {
	_, err := c.c.Get(key)
	return err == nil
}

// Flush deletes all cached data.
func (c *Cacher) Flush() error {
	return c.c.FlushAll()
}

// StartAndGC starts GC routine based on config string settings.
// AdapterConfig: 127.0.0.1:9090;127.0.0.1:9091
func (c *Cacher) StartAndGC(opt cache.Options) error {
	c.c = memcache.New(strings.Split(opt.AdapterConfig, ";")...)
	return nil
}

// Ping tests if the cache is alive.
func (c *Cacher) Ping() error {
	return cache.GenericPing(c)
}

func init() {
	cache.Register("memcache", &Cacher{})
}