/*
Rating system designed to be used in VoIP Carriers World
Copyright (C) 2013 ITsysCOM

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>
*/

package cdrs

import (
	"fmt"
	"github.com/cgrates/cgrates/config"
	"github.com/cgrates/cgrates/engine"
	"github.com/cgrates/cgrates/mediator"
	"github.com/cgrates/cgrates/utils"
	"io/ioutil"
	"net/http"
)

var (
	cfg     *config.CGRConfig // Share the configuration with the rest of the package
	storage engine.CdrStorage
	medi    *mediator.Mediator
)

func fsCdrHandler(w http.ResponseWriter, r *http.Request) {
	body, _ := ioutil.ReadAll(r.Body)
	if fsCdr, err := new(FSCdr).New(body); err == nil {
		storage.SetCdr(fsCdr)
		go func() { //FS will not send us hangup_complete until we have send back the answer to CDR, so we need to handle mediation async
			if cfg.CDRSMediator == "internal" {
				medi.MediateRawCDR(fsCdr)
			} else {
				//TODO: use the connection to mediator
			}
		}()
	} else {
		engine.Logger.Err(fmt.Sprintf("Could not create CDR entry: %s", err.Error()))
	}
}

func cgrCdrHandler(w http.ResponseWriter, r *http.Request) {
	if cgrCdr, err := utils.NewCgrCdrFromHttpReq(r); err == nil {
		storage.SetCdr(cgrCdr)
		if cfg.CDRSMediator == "internal" {
			errMedi := medi.MediateRawCDR(cgrCdr)
			if errMedi != nil {
				engine.Logger.Err(fmt.Sprintf("Could not run mediation on CDR: %s", errMedi.Error()))
			}
		} else {
			//TODO: use the connection to mediator
		}
	} else {
		engine.Logger.Err(fmt.Sprintf("Could not create CDR entry: %s", err.Error()))
	}
}

type CDRS struct{}

func New(s engine.CdrStorage, m *mediator.Mediator, c *config.CGRConfig) *CDRS {
	storage = s
	medi = m
	cfg = c
	return &CDRS{}
}

func (cdrs *CDRS) StartCapturingCDRs() {
	http.HandleFunc("/cgr", cgrCdrHandler)            // Attach CGR CDR Handler
	http.HandleFunc("/freeswitch_json", fsCdrHandler) // Attach FreeSWITCH JSON CDR Handler
	http.ListenAndServe(cfg.CDRSListen, nil)
}