giedomak/Telepath

/**
 * Copyright (C) 2016-2017 - All rights reserved.
 * This file is part of the telepath project which is released under the GPLv3 license.
 * See file LICENSE.txt or go to http://www.gnu.org/licenses/gpl.txt for full license details.
 * You may use, distribute and modify this code under the terms of the GPLv3 license.
 */

package com.github.giedomak.telepath.datamodels.integrations

import com.github.giedomak.telepath.datamodels.graph.Node
import com.github.giedomak.telepath.datamodels.graph.Path
import com.github.giedomak.telepath.datamodels.graph.PathPrefix
import com.pathdb.pathIndex.models.ImmutablePathPrefix
import kotlin.streams.toList

/**
 * Convert our data classes from and to the ones from [com.pathdb].
 */
object PathDBWrapper {

    fun toPath(path: Path): com.pathdb.pathIndex.models.Path {
        return com.pathdb.pathIndex.models.ImmutablePath.of(path.pathId, toNodeList(path.nodes))
    }

    fun fromPath(path: com.pathdb.pathIndex.models.Path): Path {
        return Path(path.pathId, fromNodeList(path.nodes))
    }

    fun toPathPrefix(pathPrefix: PathPrefix): com.pathdb.pathIndex.models.PathPrefix {
        return ImmutablePathPrefix.of(pathPrefix.pathId, toNodeList(pathPrefix.nodes))
    }

    private fun toNode(node: Node): com.pathdb.pathIndex.models.Node {
        return com.pathdb.pathIndex.models.Node(node.id)
    }

    private fun fromNode(node: com.pathdb.pathIndex.models.Node): Node {
        return Node(node.id)
    }

    private fun toNodeList(nodes: List<Node>): List<com.pathdb.pathIndex.models.Node> {
        return nodes.stream()
                .map { toNode(it) }
                .toList()
    }

    private fun fromNodeList(nodes: List<com.pathdb.pathIndex.models.Node>): List<Node> {
        return nodes.stream()
                .map { fromNode(it) }
                .toList()
    }
}