From 360335191a33173b83fd4b3993b4519f8adb8ed8 Mon Sep 17 00:00:00 2001
From: DanB <danb@cgrates.org>
Date: Thu, 8 Dec 2016 13:03:30 +0100
Subject: [PATCH] Migrator with CallCost conversion

---
 migrator/costdetails.go      | 49 +++++++++++++++++++++++++++++++++---
 migrator/costdetails_test.go | 13 +++++++---
 migrator/migrator.go         |  2 +-
 3 files changed, 56 insertions(+), 8 deletions(-)

diff --git a/migrator/costdetails.go b/migrator/costdetails.go
index 89d78ed28..d0b37d770 100644
--- a/migrator/costdetails.go
+++ b/migrator/costdetails.go
@@ -23,10 +23,6 @@ import (
 	"github.com/cgrates/cgrates/engine"
 )
 
-type CallCostMigrator interface {
-	AsCallCost() (*engine.CallCost, error)
-}
-
 type v1CallCost struct {
 	Direction, Category, Tenant, Subject, Account, Destination, TOR string
 	Cost                                                            float64
@@ -69,5 +65,50 @@ type v1UnitInfo struct {
 
 func (v1cc *v1CallCost) AsCallCost() (cc *engine.CallCost, err error) {
 	cc = new(engine.CallCost)
+	cc.Direction = v1cc.Direction
+	cc.Category = v1cc.Category
+	cc.Tenant = v1cc.Tenant
+	cc.Account = v1cc.Account
+	cc.Subject = v1cc.Subject
+	cc.Destination = v1cc.Destination
+	cc.TOR = v1cc.TOR
+	cc.Cost = v1cc.Cost
+	cc.Timespans = make(engine.TimeSpans, len(v1cc.Timespans))
+	for i, v1ts := range v1cc.Timespans {
+		cc.Timespans[i] = &engine.TimeSpan{TimeStart: v1ts.TimeStart,
+			TimeEnd:        v1ts.TimeEnd,
+			Cost:           v1ts.Cost,
+			RateInterval:   v1ts.RateInterval,
+			DurationIndex:  v1ts.DurationIndex,
+			Increments:     make(engine.Increments, len(v1ts.Increments)),
+			MatchedSubject: v1ts.MatchedSubject,
+			MatchedPrefix:  v1ts.MatchedPrefix,
+			MatchedDestId:  v1ts.MatchedDestId,
+			RatingPlanId:   v1ts.RatingPlanId,
+		}
+		for j, v1Incrm := range v1ts.Increments {
+			cc.Timespans[i].Increments[j] = &engine.Increment{
+				Duration:       v1Incrm.Duration,
+				Cost:           v1Incrm.Cost,
+				CompressFactor: v1Incrm.CompressFactor,
+				BalanceInfo: &engine.DebitInfo{
+					AccountID: v1Incrm.BalanceInfo.AccountId,
+				},
+			}
+			if v1Incrm.BalanceInfo.UnitBalanceUuid != "" {
+				cc.Timespans[i].Increments[j].BalanceInfo.Unit = &engine.UnitInfo{
+					UUID:          v1Incrm.BalanceInfo.UnitBalanceUuid,
+					Value:         v1Incrm.UnitInfo.Quantity,
+					DestinationID: v1Incrm.UnitInfo.DestinationId,
+					TOR:           v1Incrm.UnitInfo.TOR,
+				}
+			} else if v1Incrm.BalanceInfo.MoneyBalanceUuid != "" {
+				cc.Timespans[i].Increments[j].BalanceInfo.Monetary = &engine.MonetaryInfo{
+					UUID: v1Incrm.BalanceInfo.MoneyBalanceUuid,
+					//Value: v1Incrm.UnitInfo.Quantity,
+				}
+			}
+		}
+	}
 	return
 }
diff --git a/migrator/costdetails_test.go b/migrator/costdetails_test.go
index 08f188d74..84d8297de 100644
--- a/migrator/costdetails_test.go
+++ b/migrator/costdetails_test.go
@@ -19,7 +19,6 @@ package migrator
 
 import (
 	"encoding/json"
-	"fmt"
 	"testing"
 )
 
@@ -31,7 +30,12 @@ func TestV1CostDetailsAsCostDetails1(t *testing.T) {
 	if err := json.Unmarshal([]byte(v1TmspsStr1), &v1tmsps); err != nil {
 		t.Error(err)
 	}
-	fmt.Printf("Timespans: %+v\n", v1tmsps[0])
+	v1CC := &v1CallCost{Timespans: v1tmsps}
+	if _, err := v1CC.AsCallCost(); err != nil {
+		t.Error(err)
+	}
+	// ToDo: Test here the content
+
 }
 
 func TestV1CostDetailsAsCostDetails2(t *testing.T) {
@@ -39,5 +43,8 @@ func TestV1CostDetailsAsCostDetails2(t *testing.T) {
 	if err := json.Unmarshal([]byte(v1TmspsStr2), &v1tmsps); err != nil {
 		t.Error(err)
 	}
-	fmt.Printf("Timespans: %+v\n", v1tmsps[0])
+	v1CC := &v1CallCost{Timespans: v1tmsps}
+	if _, err := v1CC.AsCallCost(); err != nil {
+		t.Error(err)
+	}
 }
diff --git a/migrator/migrator.go b/migrator/migrator.go
index 7416f0b23..e825b8034 100644
--- a/migrator/migrator.go
+++ b/migrator/migrator.go
@@ -70,7 +70,7 @@ func (m *Migrator) migrateCostDetails() (err error) {
 			utils.UndefinedVersion,
 			"version number is not defined for CostDetails model")
 	}
-	if vrs[utils.COST_DETAILS] != 1 {
+	if vrs[utils.COST_DETAILS] != 1 { // Right now we only support migrating from version 1
 		return
 	}
 	storSQL := m.storDB.(*engine.SQLStorage)