tlinx-gettext.kotlinx-gettext-plugin.0.4.1-dev-1-ir.source-code.PoEntry.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kotlinx-gettext-plugin Show documentation

Kotlin Compiler plugin to extract strings for i18n in Gettext format

There is a newer version: 0.6.1

Show newest version

/*
 * Copyright 2022 Victor Kropp
 *
 * 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 name.kropp.kotlinx.gettext

import java.io.ByteArrayOutputStream
import java.io.PrintStream

/**
 * Single entry of [PoFile]
 */
data class PoEntry(
    /**
     * Translator comments
     */
    val comments: List,
    /**
     * Comments extracted from the source code
     */
    val extractedComments: List,
    /**
     * Reference to the source code where this entry has been extracted from
     */
    val references: List,
    /**
     * Flags
     */
    val flags: String?,
    /**
     * Previous entries of the same entry
     */
    val previous: List,
    /**
     * Contextual information
     */
    val context: String?,
    /**
     * Original message, usually an untranslated string
     */
    val text: String,
    /**
     * Plural form of original untranslated string
     */
    val plural: String?,
    /**
     * Different cases of translated string for different plural forms
     */
    val cases: List
) {
    fun write(out: PrintStream) {
        for (comment in comments) {
            out.println("# $comment")
        }
        for (comment in extractedComments) {
            out.println("#. $comment")
        }
        for (reference in references) {
            out.println("#: $reference")
        }
        if (flags != null) {
            out.println("#, $flags")
        }
        for (prev in previous) {
            out.println("#| $prev")
        }
        if (context != null) {
            out.println("msgctxt \"${context.escape()}\"")
        }
        out.println("msgid \"${text.escape()}\"")
        if (plural != null) {
            out.println("msgid_plural \"${plural.escape()}\"")
            cases.forEachIndexed { idx, msg ->
                out.println("msgstr[$idx] \"${msg.escape()}\"")
            }
        } else {
            if (cases.isEmpty()) {
                out.println("msgstr \"\"")
            } else {
                out.println("msgstr \"${cases[0].escape()}\"")
            }
        }
    }

    private fun String.escape(): String {
        return replace("\n", "\\n").replace("\"", "\\\"")
    }

    override fun toString(): String {
        return ByteArrayOutputStream().use {
            PrintStream(it, false, Charsets.UTF_8).use { printer -> write(printer) }
            it.toString()
        }
    }
}