cgrates/cdrs/fscdr.go

/*
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 (
	"encoding/json"
	"errors"
	"github.com/cgrates/cgrates/utils"
	"strconv"
	"time"
)

const (
	// Freswitch event property names
	CDR_MAP      = "variables"
	DIRECTION    = "direction"
	ORIG_ID      = "sip_call_id" //- originator_id - match cdrs
	SUBJECT      = "cgr_subject"
	ACCOUNT      = "cgr_account"
	DESTINATION  = "cgr_destination"
	REQTYPE      = "cgr_reqtype" //prepaid or postpaid
	TOR          = "cgr_tor"
	UUID         = "uuid" // -Unique ID for this call leg
	CSTMID       = "cgr_cstmid"
	CALL_DEST_NR = "dialed_extension"
	PARK_TIME    = "start_epoch"
	ANSWER_TIME   = "answer_epoch"
	HANGUP_TIME     = "end_epoch"
	DURATION     = "billsec"
	USERNAME     = "user_name"
	FS_IP        = "sip_local_network_addr"
)

type FSCdr map[string]string

func (fsCdr FSCdr) New(body []byte) (utils.CDR, error) {
	fsCdr = make(map[string]string)
	var tmp map[string]interface{}
	var err error
	if err = json.Unmarshal(body, &tmp); err == nil {
		if variables, ok := tmp[CDR_MAP]; ok {
			if variables, ok := variables.(map[string]interface{}); ok {
				for k, v := range variables {
					fsCdr[k] = v.(string)
				}
			}
			return fsCdr, nil
		}
	}
	return nil, err
}

func (fsCdr FSCdr) GetCgrId() string {
	return utils.FSCgrId(fsCdr[UUID])
}
func (fsCdr FSCdr) GetAccId() string {
	return fsCdr[UUID]
}
func (fsCdr FSCdr) GetCdrHost() string {
	return fsCdr[FS_IP]
}
func (fsCdr FSCdr) GetDirection() string {
	//TODO: implement direction, not related to FS_DIRECTION but traffic towards or from subject/account
	return "OUT"
}
func (fsCdr FSCdr) GetOrigId() string {
	return fsCdr[ORIG_ID]
}
func (fsCdr FSCdr) GetSubject() string {
	return utils.FirstNonEmpty(fsCdr[SUBJECT], fsCdr[USERNAME])
}
func (fsCdr FSCdr) GetAccount() string {
	return utils.FirstNonEmpty(fsCdr[ACCOUNT], fsCdr[USERNAME])
}

// Charging destination number
func (fsCdr FSCdr) GetDestination() string {
	return utils.FirstNonEmpty(fsCdr[DESTINATION], fsCdr[CALL_DEST_NR])
}

func (fsCdr FSCdr) GetTOR() string {
	return utils.FirstNonEmpty(fsCdr[TOR], cfg.DefaultTOR)
}

func (fsCdr FSCdr) GetTenant() string {
	return utils.FirstNonEmpty(fsCdr[CSTMID], cfg.DefaultTenant)
}
func (fsCdr FSCdr) GetReqType() string {
	return utils.FirstNonEmpty(fsCdr[REQTYPE], cfg.DefaultReqType)
}
func (fsCdr FSCdr) GetExtraFields() map[string]string {
	return nil // ToDo: Add and extract from config
}
func (fsCdr FSCdr) GetFallbackSubj() string {
	return cfg.DefaultSubject
}
func (fsCdr FSCdr) GetAnswerTime() (t time.Time, err error) {
	st, err := strconv.ParseInt(fsCdr[ANSWER_TIME], 0, 64)
	t = time.Unix(0, st*1000)
	return
}
func (fsCdr FSCdr) GetHangupTime() (t time.Time, err error) {
	st, err := strconv.ParseInt(fsCdr[HANGUP_TIME], 0, 64)
	t = time.Unix(0, st*1000)
	return
}

// Extracts duration as considered by the telecom switch
func (fsCdr FSCdr) GetDuration() int64 {
	dur, _ := strconv.ParseInt(fsCdr[DURATION], 0, 64)
	return dur
}

func (fsCdr FSCdr) Store() (result string, err error) {
	result += fsCdr.GetCgrId() + "|"
	result += fsCdr.GetAccId() + "|"
	result += fsCdr.GetCdrHost() + "|"
	result += fsCdr.GetDirection() + "|"
	result += fsCdr.GetOrigId() + "|"
	result += fsCdr.GetSubject() + "|"
	result += fsCdr.GetAccount() + "|"
	result += fsCdr.GetDestination() + "|"
	result += fsCdr.GetTOR() + "|"
	result += fsCdr.GetAccId() + "|"
	result += fsCdr.GetTenant() + "|"
	result += fsCdr.GetReqType() + "|"
	st, err := fsCdr.GetAnswerTime()
	if err != nil {
		return "", err
	}
	result += strconv.FormatInt(st.UnixNano(), 10) + "|"
	et, err := fsCdr.GetHangupTime()
	if err != nil {
		return "", err
	}
	result += strconv.FormatInt(et.UnixNano(), 10) + "|"
	result += strconv.FormatInt(fsCdr.GetDuration(), 10) + "|"
	result += fsCdr.GetFallbackSubj() + "|"
	return
}

func (fsCdr FSCdr) Restore(input string) error {
	return errors.New("Not implemented")
}