/*
Real-time Online/Offline Charging System (OCS) for Telecom & ISP environments
Copyright (C) ITsysCOM GmbH

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 migrator

import (
	"encoding/json"
	"fmt"
	"time"

	"github.com/cgrates/cgrates/engine"
	"github.com/cgrates/cgrates/utils"
)

func (m *Migrator) migrateCurrentCDRs() (err error) {
	if m.sameStorDB { // no move
		return
	}
	cdrs, _, err := m.storDBIn.StorDB().GetCDRs(new(utils.CDRsFilter), false)
	if err != nil {
		return err
	}
	for _, cdr := range cdrs {
		if err := m.storDBOut.StorDB().SetCDR(cdr, true); err != nil {
			return err
		}
		m.stats[utils.CDRs]++
	}
	return
}

func (m *Migrator) migrateCDRs() (err error) {
	var vrs engine.Versions
	current := engine.CurrentStorDBVersions()
	if vrs, err = m.getVersions(utils.CDRs); err != nil {
		return
	}
	migrated := true
	var v2 *engine.CDR
	for {
		version := vrs[utils.CDRs]
		for {
			switch version {
			default:
				return fmt.Errorf("Unsupported version %v", version)
			case current[utils.CDRs]:
				migrated = false
				if m.sameStorDB {
					break
				}
				if err = m.migrateCurrentCDRs(); err != nil {
					return
				}
			case 1:
				if v2, err = m.migrateV1CDRs(); err != nil && err != utils.ErrNoMoreData {
					return
				}
				version = 2
			}
			if version == current[utils.CDRs] || err == utils.ErrNoMoreData {
				break
			}
		}
		if err == utils.ErrNoMoreData || !migrated {
			break
		}

		if !m.dryRun {
			//set action plan
			if err = m.storDBOut.StorDB().SetCDR(v2, true); err != nil {
				return
			}
		}
		m.stats[utils.CDRs]++
	}
	// All done, update version wtih current one
	if err = m.setVersions(utils.CDRs); err != nil {
		return
	}
	return m.ensureIndexesStorDB(engine.ColCDRs)
}

func (m *Migrator) removeV1CDRs() (err error) {
	var v1CDR *v1Cdrs
	if v1CDR, err = m.storDBIn.getV1CDR(); err != nil {
		return err
	}
	if v1CDR == nil {
		return
	}
	if err = m.storDBIn.remV1CDRs(v1CDR); err != nil {
		return
	}
	return
}

func (m *Migrator) migrateV1CDRs() (cdr *engine.CDR, err error) {
	var v1CDR *v1Cdrs
	if v1CDR, err = m.storDBIn.getV1CDR(); err != nil {
		return nil, err
	}
	if v1CDR == nil {
		return
	}
	cdr = v1CDR.V1toV2Cdr()
	return
}

type v1Cdrs struct {
	CGRID       string
	RunID       string
	OrderID     int64             // Stor order id used as export order id
	OriginHost  string            // represents the IP address of the host generating the CDR (automatically populated by the server)
	Source      string            // formally identifies the source of the CDR (free form field)
	OriginID    string            // represents the unique accounting id given by the telecom switch generating the CDR
	ToR         string            // type of record, meta-field, should map to one of the TORs hardcoded inside the server <*voice|*data|*sms|*generic>
	RequestType string            // matching the supported request types by the **CGRateS**, accepted values are hardcoded in the server <prepaid|postpaid|pseudoprepaid|rated>.
	Tenant      string            // tenant whom this record belongs
	Category    string            // free-form filter for this record, matching the category defined in rating profiles.
	Account     string            // account id (accounting subsystem) the record should be attached to
	Subject     string            // rating subject (rating subsystem) this record should be attached to
	Destination string            // destination to be charged
	SetupTime   time.Time         // set-up time of the event. Supported formats: datetime RFC3339 compatible, SQL datetime (eg: MySQL), unix timestamp.
	AnswerTime  time.Time         // answer time of the event. Supported formats: datetime RFC3339 compatible, SQL datetime (eg: MySQL), unix timestamp.
	Usage       time.Duration     // event usage information (eg: in case of tor=*voice this will represent the total duration of a call)
	ExtraFields map[string]string // Extra fields to be stored in CDR
	ExtraInfo   string            // Container for extra information related to this CDR, eg: populated with error reason in case of error on calculation
	Partial     bool              // Used for partial record processing by ERs
	Rated       bool              // Mark the CDR as rated so we do not process it during rating
	CostSource  string            // The source of this cost
	Cost        float64
}

func (v1Cdr *v1Cdrs) V1toV2Cdr() (cdr *engine.CDR) {
	cdr = &engine.CDR{
		CGRID:       v1Cdr.CGRID,
		RunID:       v1Cdr.RunID,
		OrderID:     v1Cdr.OrderID,
		OriginHost:  v1Cdr.OriginHost,
		Source:      v1Cdr.Source,
		OriginID:    v1Cdr.OriginID,
		ToR:         v1Cdr.ToR,
		RequestType: v1Cdr.RequestType,
		Tenant:      v1Cdr.Tenant,
		Category:    v1Cdr.Category,
		Account:     v1Cdr.Account,
		Subject:     v1Cdr.Subject,
		Destination: v1Cdr.Destination,
		SetupTime:   v1Cdr.SetupTime,
		AnswerTime:  v1Cdr.AnswerTime,
		Usage:       v1Cdr.Usage,
		ExtraFields: make(map[string]string),
		ExtraInfo:   v1Cdr.ExtraInfo,
		Partial:     v1Cdr.Partial,
		PreRated:    v1Cdr.Rated,
		CostSource:  v1Cdr.CostSource,
		Cost:        v1Cdr.Cost,
	}
	if v1Cdr.ExtraFields != nil {
		for key, value := range v1Cdr.ExtraFields {
			cdr.ExtraFields[key] = value
		}
	}
	return
}

func NewV1CDRFromCDRSql(cdrSql *engine.CDRsql) (cdr *v1Cdrs, err error) {
	cdr = new(v1Cdrs)
	cdr.CGRID = cdrSql.Cgrid
	cdr.RunID = cdrSql.RunID
	cdr.OriginHost = cdrSql.OriginHost
	cdr.Source = cdrSql.Source
	cdr.OriginID = cdrSql.OriginID
	cdr.OrderID = cdrSql.ID
	cdr.ToR = cdrSql.TOR
	cdr.RequestType = cdrSql.RequestType
	cdr.Tenant = cdrSql.Tenant
	cdr.Category = cdrSql.Category
	cdr.Account = cdrSql.Account
	cdr.Subject = cdrSql.Subject
	cdr.Destination = cdrSql.Destination
	cdr.SetupTime = cdrSql.SetupTime
	cdr.AnswerTime = cdrSql.AnswerTime
	cdr.Usage = time.Duration(cdrSql.Usage)
	cdr.CostSource = cdrSql.CostSource
	cdr.Cost = cdrSql.Cost
	cdr.ExtraInfo = cdrSql.ExtraInfo
	if cdrSql.ExtraFields != "" {
		if err = json.Unmarshal([]byte(cdrSql.ExtraFields), &cdr.ExtraFields); err != nil {
			return nil, err
		}
	}
	return
}

func (cdr *v1Cdrs) AsCDRsql() (cdrSql *engine.CDRsql) {
	cdrSql = new(engine.CDRsql)
	cdrSql.Cgrid = cdr.CGRID
	cdrSql.RunID = cdr.RunID
	cdrSql.OriginHost = cdr.OriginHost
	cdrSql.Source = cdr.Source
	cdrSql.OriginID = cdr.OriginID
	cdrSql.TOR = cdr.ToR
	cdrSql.RequestType = cdr.RequestType
	cdrSql.Tenant = cdr.Tenant
	cdrSql.Category = cdr.Category
	cdrSql.Account = cdr.Account
	cdrSql.Subject = cdr.Subject
	cdrSql.Destination = cdr.Destination
	cdrSql.SetupTime = cdr.SetupTime
	cdrSql.AnswerTime = cdr.AnswerTime
	cdrSql.Usage = cdr.Usage.Nanoseconds()
	cdrSql.ExtraFields = utils.ToJSON(cdr.ExtraFields)
	cdrSql.CostSource = cdr.CostSource
	cdrSql.Cost = cdr.Cost
	cdrSql.ExtraInfo = cdr.ExtraInfo
	cdrSql.CreatedAt = time.Now()
	return
}