ru.pocketbyte.locolaser.resource.ResourcesSet.kt Maven / Gradle / Ivy

Go to download
/*
 * Copyright © 2017 Denis Shurygin. All rights reserved.
 * Licensed under the Apache License, Version 2.0
 */

package ru.pocketbyte.locolaser.resource

import ru.pocketbyte.locolaser.config.ExtraParams
import ru.pocketbyte.locolaser.resource.entity.ResMap
import ru.pocketbyte.locolaser.resource.entity.merge
import ru.pocketbyte.locolaser.resource.formatting.FormattingType
import ru.pocketbyte.locolaser.resource.formatting.MixedFormattingType
import ru.pocketbyte.locolaser.resource.formatting.NoFormattingType
import ru.pocketbyte.locolaser.summary.FileSummary

import java.io.IOException

class ResourcesSet(
    private val resources: Set,
    private val main: Resources? = null
) : Resources {

    override val formattingType: FormattingType
        get() {
            val firstItemType = if (resources.isEmpty()) {
                NoFormattingType
            } else {
                resources.elementAt(0).formattingType
            }

            return if (resources.find { it.formattingType != firstItemType } != null) {
                MixedFormattingType
            } else {
                firstItemType
            }
        }

    override fun read(locales: Set?, extraParams: ExtraParams?): ResMap? {
        return resources.fold(null as? ResMap) { resMap, source ->
            resMap.merge(source.read(locales, extraParams))
        }
    }

    @Throws(IOException::class)
    override fun write(resMap: ResMap, extraParams: ExtraParams?) {
        if (main != null) {
            main.write(resMap, extraParams)
        } else {
            resources.forEach {
                it.write(resMap, extraParams)
            }
        }
    }

    override fun summaryForLocale(locale: String): FileSummary {
        var bytes: Long = 0
        val hash = StringBuilder()

        resources.map { it.summaryForLocale(locale) }.forEach {
            bytes += it?.bytes ?: 0
            hash.append(it?.hash)
        }

        resources.forEach { resource ->
            resource.summaryForLocale(locale).let {
                bytes += it?.bytes ?: 0
                hash.append(it?.hash)
            }
        }

        return FileSummary(bytes, hash.toString())
    }
}