/*
 * Copyright 2019-present Open Networking Foundation

 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at

 * http://www.apache.org/licenses/LICENSE-2.0

 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package resourcemanager 

import (
    "encoding/json"
    "errors"
    "fmt"
    "strconv"
    "strings"

    ponrmgr "github.com/opencord/voltha-go/adapters/common/ponresourcemanager"
    "github.com/opencord/voltha-go/common/log"
    "github.com/opencord/voltha-go/db/kvstore"
    "github.com/opencord/voltha-go/db/model"
    "github.com/opencord/voltha-go/protos/openolt"
)

CONST KVSTORE_TIMEOUT = 5
const BASE_PATH_KV_STORE = "service/voltha/openolt/{%s}" // service/voltha/openolt/<device_id>

type OpenOltResourceMgr struct {
    DeviceID    string         //OLT device id
    HostAndPort string         // Host and port of the kv store to connect to
    Args        string         // args
    KVStore     *model.Backend // backend kv store connection handle
    DeviceType  string
    Host        string        // Host ip of the kv store
    Port        int           // port of the kv store
    DevInfo     *openolt.DeviceInfo // device information
    // array of pon resource managers per interface technology
    ResourceMgrs map[uint32]*ponrmgr.PONResourceManager
}

func newKVClient(storeType string, address string, timeout uint32) (kvstore.Client, error) {
    log.Infow("kv-store-type", log.Fields{"store": storeType})
    switch storeType {
    case "consul":
        return kvstore.NewConsulClient(address, int(timeout))
    case "etcd":
        return kvstore.NewEtcdClient(address, int(timeout))
    }
    return nil, errors.New("unsupported-kv-store")
}

func SetKVClient(Backend string, Host string, Port int, DeviceID string) *model.Backend {
    addr := Host + ":" + strconv.Itoa(Port)
    // TODO : Make sure direct call to NewBackend is working fine with backend , currently there is some
    // issue between kv store and backend , core is not calling NewBackend directly
    kvClient, err := newKVClient(Backend, addr, KVSTORE_TIMEOUT) 
    if err != nil {
        log.Fatalw("Failed to init KV client\n", log.Fields{"err": err})
        return nil
    }
    kvbackend := &model.Backend{
        Client:     kvClient,
        StoreType:  Backend,
        Host:       Host,
        Port:       Port,
        Timeout:    KVSTORE_TIMEOUT,
        PathPrefix: fmt.Sprintf(BASE_PATH_KV_STORE, DeviceID)}

    return kvbackend
}

func NewResourceMgr(DeviceID string, HostPort string, DeviceType string, DevInfo *openolt.DeviceInfo) *OpenOltResourceMgr {

    /* init a New resource maanger instance which in turn instantiates pon resource manager
       instances according to technology. Initializes the default resource ranges for all
       the reources.
    */
    var ResourceMgr OpenOltResourceMgr
    log.Debugf("Init new resource manager , host_port: %s, deviceid: %s", HostPort, DeviceID)
    ResourceMgr.HostAndPort = HostPort
    ResourceMgr.DeviceType = DeviceType  
    ResourceMgr.DevInfo = DevInfo
    IpPort := strings.Split(HostPort, ":")
    ResourceMgr.Host = IpPort[0]
    ResourceMgr.Port, _ = strconv.Atoi(IpPort[1])

    Backend := "etcd" /* TODO  remove this once we get backend store from registry*/
    ResourceMgr.KVStore = SetKVClient(Backend, ResourceMgr.Host,
        ResourceMgr.Port, DeviceID)
    if ResourceMgr.KVStore == nil {
        log.Error("Failed to setup KV store")
    }
    Ranges := make(map[string]*openolt.DeviceInfo_DeviceResourceRanges)
    RsrcMgrsByTech := make(map[string]*ponrmgr.PONResourceManager)
    ResourceMgr.ResourceMgrs = make(map[uint32]*ponrmgr.PONResourceManager)

    // TODO self.args = registry('main').get_args()

    /*
       If a legacy driver returns protobuf without any ranges,s synthesize one from
       the legacy global per-device informaiton. This, in theory, is temporary until
       the legacy drivers are upgrade to support pool ranges.
    */
    if DevInfo.Ranges == nil {
        var ranges openolt.DeviceInfo_DeviceResourceRanges
        ranges.Technology = DevInfo.GetTechnology()

        NumPONPorts := DevInfo.GetPonPorts()
        var index uint32
        for index = 0; index < NumPONPorts; index++ {
            ranges.IntfIds = append(ranges.IntfIds, index)
        }

        var Pool *openolt.DeviceInfo_DeviceResourceRanges_Pool
        Pool.Type = openolt.DeviceInfo_DeviceResourceRanges_Pool_ONU_ID
        Pool.Start = DevInfo.OnuIdStart
        Pool.End = DevInfo.OnuIdEnd
        Pool.Sharing = openolt.DeviceInfo_DeviceResourceRanges_Pool_DEDICATED_PER_INTF
        ranges.Pools = append(ranges.Pools, Pool)

        Pool.Type = openolt.DeviceInfo_DeviceResourceRanges_Pool_ALLOC_ID
        Pool.Start = DevInfo.AllocIdStart
        Pool.End = DevInfo.AllocIdEnd
        Pool.Sharing = openolt.DeviceInfo_DeviceResourceRanges_Pool_SHARED_BY_ALL_INTF_ALL_TECH
        ranges.Pools = append(ranges.Pools, Pool)

        Pool.Type = openolt.DeviceInfo_DeviceResourceRanges_Pool_GEMPORT_ID
        Pool.Start = DevInfo.GemportIdStart
        Pool.End = DevInfo.GemportIdEnd
        Pool.Sharing = openolt.DeviceInfo_DeviceResourceRanges_Pool_SHARED_BY_ALL_INTF_ALL_TECH
        ranges.Pools = append(ranges.Pools, Pool)

        Pool.Type = openolt.DeviceInfo_DeviceResourceRanges_Pool_FLOW_ID
        Pool.Start = DevInfo.FlowIdStart
        Pool.End = DevInfo.FlowIdEnd
        Pool.Sharing = openolt.DeviceInfo_DeviceResourceRanges_Pool_SHARED_BY_ALL_INTF_ALL_TECH
        ranges.Pools = append(ranges.Pools, Pool)
    }

    //Create a separate Resource Manager instance for each range. This assumes that
    // each technology is represented by only a single range
    var GlobalPONRsrcMgr *ponrmgr.PONResourceManager
    var err error
    for _, TechRange := range DevInfo.Ranges {
        technology := TechRange.Technology
        log.Debugf("Device info technology %s", technology)
        Ranges[technology] = TechRange
        RsrcMgrsByTech[technology], err = ponrmgr.NewPONResourceManager(technology, DeviceType, DeviceID,
            Backend, ResourceMgr.Host, ResourceMgr.Port)
        if err != nil {
            log.Errorf("Failed to create pon resource manager instacnce for technology %s", technology)
            return nil
        }
        //resource_mgrs_by_tech[technology] = resource_mgr
        if GlobalPONRsrcMgr == nil {
            GlobalPONRsrcMgr = RsrcMgrsByTech[technology]
        }
        for IntfId := range TechRange.IntfIds {
            ResourceMgr.ResourceMgrs[uint32(IntfId)] = RsrcMgrsByTech[technology]
        }
        //self.initialize_device_resource_range_and_pool(resource_mgr, global_resource_mgr, arange)
        InitializeDeviceResourceRangeAndPool(RsrcMgrsByTech[technology], GlobalPONRsrcMgr,
            TechRange, DevInfo)
    }
    // After we have initialized resource ranges, initialize the
    // resource pools accordingly.
    for _, PONRMgr := range RsrcMgrsByTech {
        PONRMgr.InitDeviceResourcePool()
    }
    return &ResourceMgr
}

func InitializeDeviceResourceRangeAndPool(PONRMgr *ponrmgr.PONResourceManager, GlobalPONRMgr *ponrmgr.PONResourceManager,
    TechRange *openolt.DeviceInfo_DeviceResourceRanges, DevInfo *openolt.DeviceInfo) {

    // init the resource range pool according to the sharing type

    log.Debugf("Resource range pool init for technology %s", PONRMgr.Technology)
    //first load from KV profiles
    status := PONRMgr.InitResourceRangesFromKVStore()
    if status == false {
        log.Debugf("Failed to load resource ranges from KV store for tech %s", PONRMgr.Technology)
    }

    /*
       Then apply device specific information. If KV doesn't exist
       or is broader than the device, the device's informationw ill
       dictate the range limits
    */
    log.Debugf("Using device info to init pon resource ranges for tech", PONRMgr.Technology)

    ONUIDStart := DevInfo.OnuIdStart
    ONUIDEnd := DevInfo.OnuIdEnd
    ONUIDShared := openolt.DeviceInfo_DeviceResourceRanges_Pool_DEDICATED_PER_INTF
    ONUIDSharedPoolID := uint32(0)
    AllocIDStart := DevInfo.AllocIdStart
    AllocIDEnd := DevInfo.AllocIdEnd
    AllocIDShared := openolt.DeviceInfo_DeviceResourceRanges_Pool_SHARED_BY_ALL_INTF_ALL_TECH // TODO EdgeCore/BAL limitation
    AllocIDSharedPoolID := uint32(0)
    GEMPortIDStart := DevInfo.GemportIdStart
    GEMPortIDEnd := DevInfo.GemportIdEnd
    GEMPortIDShared := openolt.DeviceInfo_DeviceResourceRanges_Pool_SHARED_BY_ALL_INTF_ALL_TECH // TODO EdgeCore/BAL limitation
    GEMPortIDSharedPoolID := uint32(0)
    FlowIDStart := DevInfo.FlowIdStart
    FlowIDEnd := DevInfo.FlowIdEnd
    FlowIDShared := openolt.DeviceInfo_DeviceResourceRanges_Pool_SHARED_BY_ALL_INTF_ALL_TECH // TODO EdgeCore/BAL limitation
    FlowIDSharedPoolID := uint32(0)

    var GlobalPoolID uint32
    var FirstIntfPoolID uint32
    var SharedPoolID uint32

    for _, FirstIntfPoolID = range TechRange.IntfIds {
        break
    }

    for _, RangePool := range TechRange.Pools {
        if RangePool.Sharing == openolt.DeviceInfo_DeviceResourceRanges_Pool_SHARED_BY_ALL_INTF_ALL_TECH {
            SharedPoolID = GlobalPoolID
        } else if RangePool.Sharing == openolt.DeviceInfo_DeviceResourceRanges_Pool_SHARED_BY_ALL_INTF_SAME_TECH {
            SharedPoolID = FirstIntfPoolID
        } else {
            SharedPoolID = 0
        }
        if RangePool.Type == openolt.DeviceInfo_DeviceResourceRanges_Pool_ONU_ID {
            ONUIDStart = RangePool.Start
            ONUIDEnd = RangePool.End
            ONUIDShared = RangePool.Sharing
            ONUIDSharedPoolID = SharedPoolID
        } else if RangePool.Type == openolt.DeviceInfo_DeviceResourceRanges_Pool_ALLOC_ID {
            AllocIDStart = RangePool.Start
            AllocIDEnd = RangePool.End
            AllocIDShared = RangePool.Sharing
            AllocIDSharedPoolID = SharedPoolID
        } else if RangePool.Type == openolt.DeviceInfo_DeviceResourceRanges_Pool_GEMPORT_ID {
            GEMPortIDStart = RangePool.Start
            GEMPortIDEnd = RangePool.End
            GEMPortIDShared = RangePool.Sharing
            GEMPortIDSharedPoolID = SharedPoolID
        } else if RangePool.Type == openolt.DeviceInfo_DeviceResourceRanges_Pool_FLOW_ID {
            FlowIDStart = RangePool.Start
            FlowIDEnd = RangePool.End
            FlowIDShared = RangePool.Sharing
            FlowIDSharedPoolID = SharedPoolID
        }
    }

    log.Debugw("Device info init", log.Fields{"technology": TechRange.Technology,
        "onu_id_start": ONUIDStart, "onu_id_end": ONUIDEnd, "onu_id_shared_pool_id": ONUIDSharedPoolID,
        "alloc_id_start": AllocIDStart, "alloc_id_end": AllocIDEnd,
        "alloc_id_shared_pool_id": AllocIDSharedPoolID,
        "gemport_id_start":        GEMPortIDStart, "gemport_id_end": GEMPortIDEnd,
        "gemport_id_shared_pool_id": GEMPortIDSharedPoolID,
        "flow_id_start":             FlowIDStart,
        "flow_id_end_idx":           FlowIDEnd,
        "flow_id_shared_pool_id":    FlowIDSharedPoolID,
        "intf_ids":                  TechRange.IntfIds,
        "uni_id_start":              0,
        "uni_id_end_idx":/*MaxUNIIDperONU()*/ 1})

    PONRMgr.InitDefaultPONResourceRanges(ONUIDStart, ONUIDEnd, ONUIDSharedPoolID,
        AllocIDStart, AllocIDEnd, AllocIDSharedPoolID,
        GEMPortIDStart, GEMPortIDEnd, GEMPortIDSharedPoolID,
        FlowIDStart, FlowIDEnd, FlowIDSharedPoolID, 0, 1,
        DevInfo.PonPorts, TechRange.IntfIds)

    // For global sharing, make sure to refresh both local and global resource manager instances' range

    if ONUIDShared == openolt.DeviceInfo_DeviceResourceRanges_Pool_SHARED_BY_ALL_INTF_ALL_TECH {
        GlobalPONRMgr.UpdateRanges(ponrmgr.ONU_ID_START_IDX, ONUIDStart, ponrmgr.ONU_ID_END_IDX, ONUIDEnd,
            "", 0, nil)
        PONRMgr.UpdateRanges(ponrmgr.ONU_ID_START_IDX, ONUIDStart, ponrmgr.ONU_ID_END_IDX, ONUIDEnd,
            "", 0, GlobalPONRMgr)
    }
    if AllocIDShared == openolt.DeviceInfo_DeviceResourceRanges_Pool_SHARED_BY_ALL_INTF_ALL_TECH {
        GlobalPONRMgr.UpdateRanges(ponrmgr.ALLOC_ID_START_IDX, AllocIDStart, ponrmgr.ALLOC_ID_END_IDX, AllocIDEnd,
            "", 0, nil)

        PONRMgr.UpdateRanges(ponrmgr.ALLOC_ID_START_IDX, AllocIDStart, ponrmgr.ALLOC_ID_END_IDX, AllocIDEnd,
            "", 0, GlobalPONRMgr)
    }
    if GEMPortIDShared == openolt.DeviceInfo_DeviceResourceRanges_Pool_SHARED_BY_ALL_INTF_ALL_TECH {
        GlobalPONRMgr.UpdateRanges(ponrmgr.GEMPORT_ID_START_IDX, GEMPortIDStart, ponrmgr.GEMPORT_ID_END_IDX, GEMPortIDEnd,
            "", 0, nil)
        PONRMgr.UpdateRanges(ponrmgr.GEMPORT_ID_START_IDX, GEMPortIDStart, ponrmgr.GEMPORT_ID_END_IDX, GEMPortIDEnd,
            "", 0, GlobalPONRMgr)
    }
    if FlowIDShared == openolt.DeviceInfo_DeviceResourceRanges_Pool_SHARED_BY_ALL_INTF_ALL_TECH {
        GlobalPONRMgr.UpdateRanges(ponrmgr.FLOW_ID_START_IDX, FlowIDStart, ponrmgr.FLOW_ID_END_IDX, FlowIDEnd,
            "", 0, nil)
        PONRMgr.UpdateRanges(ponrmgr.FLOW_ID_START_IDX, FlowIDStart, ponrmgr.FLOW_ID_END_IDX, FlowIDEnd,
            "", 0, GlobalPONRMgr)
    }

    // Make sure loaded range fits the platform bit encoding ranges
    PONRMgr.UpdateRanges(ponrmgr.UNI_ID_START_IDX, 0, ponrmgr.UNI_ID_END_IDX /* TODO =OpenOltPlatform.MAX_UNIS_PER_ONU-1*/, 1, "", 0, nil)
}

/* TODO
def __del__(self):
        self.log.info("clearing-device-resource-pool")
        for key, resource_mgr in self.resource_mgrs.iteritems():
            resource_mgr.clear_device_resource_pool()

    def assert_pon_id_limit(self, pon_intf_id):
        assert pon_intf_id in self.resource_mgrs

    def assert_onu_id_limit(self, pon_intf_id, onu_id):
        self.assert_pon_id_limit(pon_intf_id)
        self.resource_mgrs[pon_intf_id].assert_resource_limits(onu_id, PONResourceManager.ONU_ID)

    @property
    def max_uni_id_per_onu(self):
        return 0 #OpenOltPlatform.MAX_UNIS_PER_ONU-1, zero-based indexing Uncomment or override to make default multi-uni

    def assert_uni_id_limit(self, pon_intf_id, onu_id, uni_id):
        self.assert_onu_id_limit(pon_intf_id, onu_id)
        self.resource_mgrs[pon_intf_id].assert_resource_limits(uni_id, PONResourceManager.UNI_ID)
*/

func (RsrcMgr *OpenOltResourceMgr) GetONUID(PONIntfID uint32) (uint32, error) {

    // Get ONU id for a provided pon interface ID.

    ONUID, err := RsrcMgr.ResourceMgrs[PONIntfID].GetResourceID(PONIntfID,
        ponrmgr.ONU_ID, 1)
    if err != nil {
        log.Errorf("Failed to get resource for interface %d for type %s",
            PONIntfID, ponrmgr.ONU_ID)
        return ONUID[0], err
    }
    if ONUID != nil {
        RsrcMgr.ResourceMgrs[PONIntfID].InitResourceMap(fmt.Sprintf("%d,%d", PONIntfID, ONUID))
    }

    return ONUID[0], err
}

func (RsrcMgr *OpenOltResourceMgr) GetFlowID(PONIntfID uint32, ONUID uint32, UNIID uint32,
    FlowStoreCookie interface{},
    FlowCategory interface{}) (uint32, error) {

    // Get flow ID for a given pon interface id, onu id and uni id.

    var err error
    FlowPath := fmt.Sprintf("%d,%d,%d", PONIntfID, ONUID, UNIID)
    FlowIDs := RsrcMgr.ResourceMgrs[PONIntfID].GetCurrentFlowIDsForOnu(FlowPath)
    if FlowIDs != nil {
        /* TODO once the flow id info structure is known
                for Flow := range FlowIDs {
                    FlowInfo := RsrcMgr.ResourceMgrs[PONIntfID].GetFlowIDInfo(FlowPath, Flow)
                    for Info := range FlowInfo {
                        if FlowCategory != nil &&
                            Info[FlowCategory] == FlowCategory {
                            return 0, Flow
                        }
                        if FlowStoreCookie != nil &&
                            Info[FlowStoreCookie] == FlowStoreCookie {
                            return 0, Flow
                        }
                    }
                }
        */
    }
    FlowIDs, err = RsrcMgr.ResourceMgrs[PONIntfID].GetResourceID(PONIntfID,
        ponrmgr.FLOW_ID, 1)
    if err != nil {
        log.Errorf("Failed to get resource for interface %d for type %s",
            PONIntfID, ponrmgr.FLOW_ID)
        return FlowIDs[0], err
    }
    if FlowIDs != nil {
        RsrcMgr.ResourceMgrs[PONIntfID].UpdateFlowIDForOnu(FlowPath, FlowIDs[0], true)
    }

    return FlowIDs[0], err
}

func (RsrcMgr *OpenOltResourceMgr) GetAllocID(IntfID uint32, ONUID uint32) uint32 {

    // Get alloc id for a given pon interface id and onu id.
    var err error
    IntfONUID := fmt.Sprintf("%d,%d", IntfID, ONUID)
    AllocID := RsrcMgr.ResourceMgrs[IntfID].GetCurrentAllocIDForOnu(IntfONUID)
    if AllocID != nil {
        // Since we support only one alloc_id for the ONU at the moment,
        // return the first alloc_id in the list, if available, for that
        // ONU.
        return AllocID[0]
    }
    AllocID, err = RsrcMgr.ResourceMgrs[IntfID].GetResourceID(IntfID,
        ponrmgr.ALLOC_ID, 1)

    if AllocID == nil || err != nil {
        log.Error("Failed to allocate alloc id")
        return 0
    }
    // update the resource map on KV store with the list of alloc_id
    // allocated for the pon_intf_onu_id tuple
    err = RsrcMgr.ResourceMgrs[IntfID].UpdateAllocIdsForOnu(IntfONUID, AllocID)
    if err != nil {
        log.Error("Failed to update Alloc ID")
        return 0
    }
    return AllocID[0]
}

func (RsrcMgr *OpenOltResourceMgr) UpdateAllocIdsForOnu(PONPort uint32, ONUID uint32,
    UNIID uint32, AllocID []uint32) error {

    /* update alloc ids in kv store for a given pon interface id,
       onu id and uni id.
    */
    IntfONUID := fmt.Sprintf("%d,%d,%d", PONPort, ONUID, UNIID)
    return RsrcMgr.ResourceMgrs[PONPort].UpdateAllocIdsForOnu(IntfONUID,
        AllocID)
}
func (RsrcMgr *OpenOltResourceMgr) GetCurrentGEMPortIDsForOnu(IntfID uint32, ONUID uint32,
    UNIID uint32) []uint32 {

    /* Get gem ports for given pon interface , onu id and uni id. */

    IntfONUID := fmt.Sprintf("%d,%d,%d", IntfID, ONUID, UNIID)
    GEMPortID := RsrcMgr.ResourceMgrs[IntfID].GetCurrentGEMPortIDsForOnu(IntfONUID)
    if GEMPortID != nil {
        // Since we support only one alloc_id for the ONU at the moment,
        // return the first alloc_id in the list, if available, for that
        // ONU.
        return GEMPortID
    }
    return nil
}

func (RsrcMgr *OpenOltResourceMgr) GetCurrentAllocIDForOnu(IntfID uint32, ONUID uint32) uint32 {

    /* Get alloc ids for given pon interface and onu id. */

    IntfONUID := fmt.Sprintf("%d,%d", IntfID, ONUID)
    AllocID := RsrcMgr.ResourceMgrs[IntfID].GetCurrentAllocIDForOnu(IntfONUID)
    if AllocID != nil {
        // Since we support only one alloc_id for the ONU at the moment,
        // return the first alloc_id in the list, if available, for that
        // ONU.
        return AllocID[0]
    }
    return 0
}

func (RsrcMgr *OpenOltResourceMgr) UpdateGEMportsPonportToOnuMapOnKVStore(GEMPorts []uint32, PonPort uint32,
    ONUID uint32, UNIID uint32) error {

    /* Update onu and uni id associated with the gem port to the kv store. */
    var IntfGEMPortPath string
    Data := fmt.Sprintf("%d %d", ONUID, UNIID)
    for _, GEM := range GEMPorts {
        IntfGEMPortPath = fmt.Sprintf("%d,%d", PonPort, GEM)
        Val, err := json.Marshal(Data)
        if err != nil {
            log.Error("failed to Marshal")
            return err
        }
        // This information is used when packet_indication is received and
        // we need to derive the ONU Id for which the packet arrived based
        // on the pon_intf and gemport available in the packet_indication
        if err = RsrcMgr.KVStore.Put(IntfGEMPortPath, Val); err != nil {
            log.Errorf("Failed to update resource %s", IntfGEMPortPath)
            return err
        }
    }
    return nil
}

func (RsrcMgr *OpenOltResourceMgr) GetONUUNIfromPONPortGEMPort(PONPort uint32, GEMPort uint32) []uint32 {

    /* get the onu and uni id for a given gem port. */
    IntfGEMPortPath := fmt.Sprintf("%d,%d", PONPort, GEMPort)
    var GEMPortIDs []uint32
    var Data string
    Value, err := RsrcMgr.KVStore.Get(IntfGEMPortPath)
    if err == nil {
        if Value != nil {
            Val, _ := kvstore.ToByte(Value.Value)
            if err = json.Unmarshal(Val, &Data); err != nil {
                log.Error("Failed to unmarshal")
                return nil
            }
            IDs := strings.Split(Data, " ")
            for _, port := range IDs {
                Intport, _ := strconv.Atoi(port)
                GEMPortIDs = append(GEMPortIDs, uint32(Intport))
            }
        }
    }
    return GEMPortIDs
}

func (RsrcMgr *OpenOltResourceMgr) GetGEMPortID(PONPort uint32, ONUID uint32,
    UNIID uint32, NumOfPorts uint32) ([]uint32, error) {

    /* Get gem port id for a particular pon port, onu id
       and uni id.
    */

    var err error
    IntfONUID := fmt.Sprintf("%d,%d,%d", PONPort, ONUID, UNIID)

    GEMPortList := RsrcMgr.ResourceMgrs[PONPort].GetCurrentGEMPortIDsForOnu(IntfONUID)
    if GEMPortList != nil {
        return GEMPortList, nil
    }

    GEMPortList, err = RsrcMgr.ResourceMgrs[PONPort].GetResourceID(PONPort,
        ponrmgr.GEMPORT_ID, NumOfPorts)
    if err != nil && GEMPortList == nil {
        log.Errorf("Failed to get gem port id for %s", IntfONUID)
        return nil, err
    }

    // update the resource map on KV store with the list of gemport_id
    // allocated for the pon_intf_onu_id tuple
    err = RsrcMgr.ResourceMgrs[PONPort].UpdateGEMPortIDsForOnu(IntfONUID,
        GEMPortList)
    if err != nil {
        log.Errorf("Failed to update GEM ports to kv store for %s", IntfONUID)
        return nil, err
    }
    RsrcMgr.UpdateGEMportsPonportToOnuMapOnKVStore(GEMPortList, PONPort,
        ONUID, UNIID)
    return GEMPortList, err
}

func (RsrcMgr *OpenOltResourceMgr) UpdateGEMPortIDsForOnu(PONPort uint32, ONUID uint32,
    UNIID uint32, GEMPortList []uint32) error {

    /* Update gemport ids on to kv store for a given pon port,
       onu id and uni id.
    */
    IntfONUID := fmt.Sprintf("%d,%d,%d", PONPort, ONUID, UNIID)
    return RsrcMgr.ResourceMgrs[PONPort].UpdateGEMPortIDsForOnu(IntfONUID,
        GEMPortList)

}
func (RsrcMgr *OpenOltResourceMgr) FreeONUID(IntfID uint32, ONUID []uint32) {

    /* Free onu id for a particular interface.*/
    RsrcMgr.ResourceMgrs[IntfID].FreeResourceID(IntfID, ponrmgr.ONU_ID, ONUID)

    var IntfONUID string
    for _, onu := range ONUID {
        IntfONUID = fmt.Sprintf("%d,%d", IntfID, onu)
        RsrcMgr.ResourceMgrs[IntfID].RemoveResourceMap(IntfONUID)
    }
    return
}

func (RsrcMgr *OpenOltResourceMgr) FreeFlowID(IntfID uint32, ONUID uint32,
    UNIID uint32, FlowID []uint32) {

    /* Free flow id for a given interface, onu id and uni id.*/

    RsrcMgr.ResourceMgrs[IntfID].FreeResourceID(IntfID, ponrmgr.FLOW_ID, FlowID)

    var IntfONUID string
    var err error
    for _, flow := range FlowID {
        IntfONUID = fmt.Sprintf("%d,%d,%d", IntfID, ONUID, UNIID)
        err = RsrcMgr.ResourceMgrs[IntfID].UpdateFlowIDForOnu(IntfONUID, flow, false)
        if err != nil {
            log.Error("Failed to Update flow id infor for %s", IntfONUID)
        }
        RsrcMgr.ResourceMgrs[IntfID].RemoveFlowIDInfo(IntfONUID, flow)
    }
    return
}

func (RsrcMgr *OpenOltResourceMgr) FreePONResourcesForONU(IntfID uint32, ONUID uint32) {

    /* Free pon resources for a given pon interface and onu id. */

    var ONUIDs []uint32
    ONUIDs = append(ONUIDs, ONUID)
    IntfONUID := fmt.Sprintf("%d,%d", IntfID, ONUID)

    AllocIDs := RsrcMgr.ResourceMgrs[IntfID].GetCurrentAllocIDForOnu(IntfONUID)

    RsrcMgr.ResourceMgrs[IntfID].FreeResourceID(IntfID,
        ponrmgr.ALLOC_ID,
        AllocIDs)

    GEMPortIDs := RsrcMgr.ResourceMgrs[IntfID].GetCurrentGEMPortIDsForOnu(IntfONUID)
    RsrcMgr.ResourceMgrs[IntfID].FreeResourceID(IntfID,
        ponrmgr.GEMPORT_ID,
        GEMPortIDs)

    FlowIDs := RsrcMgr.ResourceMgrs[IntfID].GetCurrentFlowIDsForOnu(IntfONUID)
    RsrcMgr.ResourceMgrs[IntfID].FreeResourceID(IntfID,
        ponrmgr.FLOW_ID,
        FlowIDs)
    RsrcMgr.ResourceMgrs[IntfID].FreeResourceID(IntfID,
        ponrmgr.ONU_ID,
        ONUIDs)

    // Clear resource map associated with (pon_intf_id, gemport_id) tuple.
    RsrcMgr.ResourceMgrs[IntfID].RemoveResourceMap(IntfONUID)

    // Clear the ONU Id associated with the (pon_intf_id, gemport_id) tuple.
    for _, GEM := range GEMPortIDs {
        RsrcMgr.KVStore.Delete(fmt.Sprintf("%d,%d", IntfID, GEM))
    }
}

/* TODO once the flow id info structure is known
def is_flow_cookie_on_kv_store(self, intf_id, onu_id, uni_id, flow_store_cookie):
  '''
  Note: For flows which trap from the NNI and not really associated with any particular
  ONU (like LLDP), the onu_id and uni_id is set as -1. The intf_id is the NNI intf_id.
  '''
  intf_onu_id = (intf_id, onu_id, uni_id)
  try:
      flow_ids = self.resource_mgrs[intf_id]. \
          get_current_flow_ids_for_onu(intf_onu_id)
      if flow_ids is not None:
          for flow_id in flow_ids:
              flows = self.get_flow_id_info(intf_id, onu_id, uni_id, flow_id)
              assert (isinstance(flows, list))
              for flow in flows:
                  if flow['flow_store_cookie'] == flow_store_cookie:
                      return True
  except Exception as e:
      self.log.error("error-retrieving-flow-info", e=e)

  return False
*/