package rpcstats

import (
	"context"
	"sync"
)

// PublicAPI represents a set of APIs from the namespace.
type PublicAPI struct {
	s *Service
}

// NewAPI creates an instance of the API.
func NewAPI(s *Service) *PublicAPI {
	return &PublicAPI{s: s}
}

// Reset resets RPC usage stats
func (api *PublicAPI) Reset(context context.Context) {
	resetStats()
}

type RPCStats struct {
	Total            uint            `json:"total"`
	CounterPerMethod map[string]uint `json:"methods"`
}

// GetStats returns RPC usage stats
func (api *PublicAPI) GetStats(context context.Context) (RPCStats, error) {
	total, perMethod, perMethodPerTag := getStats()

	counterPerMethod := make(map[string]uint)
	perMethod.Range(func(key, value interface{}) bool {
		counterPerMethod[key.(string)] = value.(uint)
		return true
	})
	perMethodPerTag.Range(func(key, value interface{}) bool {
		tag := key.(string)
		methods := value.(*sync.Map)
		methods.Range(func(key, value interface{}) bool {
			method := key.(string)
			count := value.(uint)
			counterPerMethod[method+"_"+tag] = count
			return true
		})
		return true
	})

	return RPCStats{
		Total:            total,
		CounterPerMethod: counterPerMethod,
	}, nil
}