fints4k/fints4javaLib/src/main/kotlin/net/dankito/fints/messages/MessageBuilder.kt

package net.dankito.fints.messages

import net.dankito.fints.messages.datenelemente.implementierte.*
import net.dankito.fints.messages.datenelemente.implementierte.signatur.Sicherheitsfunktion
import net.dankito.fints.messages.datenelemente.implementierte.tan.TanProcess
import net.dankito.fints.messages.nachrichten.Nachricht
import net.dankito.fints.messages.segmente.ISegmentNumberGenerator
import net.dankito.fints.messages.segmente.Segment
import net.dankito.fints.messages.segmente.SegmentNumberGenerator
import net.dankito.fints.messages.segmente.implementierte.*
import net.dankito.fints.util.FinTsUtils


/**
 * Takes the Segments of they payload, may signs and encrypts them, calculates message size,
 * adds the message header and closing, and formats the whole message to string.
 */
open class MessageBuilder(protected val generator: ISegmentNumberGenerator = SegmentNumberGenerator(),
                          protected val utils: FinTsUtils = FinTsUtils()) {

    companion object {
        const val MessageHeaderLength = 30
        const val MessageClosingLength = 11
        const val AddedSeparatorsLength = 3
    }


    /**
     * Um Kunden die Möglichkeit zu geben, sich anonym anzumelden, um sich bspw. über die
     * angebotenen Geschäftsvorfälle fremder Kreditinstitute (von denen sie keine BPD besitzen)
     * zu informieren bzw. nicht-signierungspflichtige Aufträge bei fremden Kreditinstituten
     * einreichen zu können, kann sich der Kunde anonym (als Gast) anmelden.
     *
     * Bei anonymen Dialogen werden Nachrichten weder signiert, noch können sie verschlüsselt und komprimiert werden.
     */
    open fun createAnonymousDialogInitMessage(
        bankCountryCode: Int,
        bankCode: String,
        productName: String,
        productVersion: String
    ): String {

        val customerId = KundenID.Anonymous

        return createMessage(false, false, bankCountryCode, bankCode, customerId, listOf(
            IdentifikationsSegment(generator.resetSegmentNumber(1), bankCountryCode, bankCode, customerId, KundensystemID.Anonymous, KundensystemStatusWerte.NichtBenoetigt),
            Verarbeitungsvorbereitung(generator.getNextSegmentNumber(), BPDVersion.VersionNotReceivedYet, UPDVersion.VersionNotReceivedYet, Dialogsprache.Default, productName, productVersion)
        ))
    }

    open fun createDialogInitMessage(
        bankCountryCode: Int,
        bankCode: String,
        customerId: String,
        customerSystemId: String,
        status: KundensystemStatusWerte,
        bpdVersion: Int,
        updVersion: Int,
        language: Dialogsprache,
        productName: String,
        productVersion: String
    ): String {

        return createMessage(true, true, bankCountryCode, bankCode, customerId, listOf(
            IdentifikationsSegment(generator.resetSegmentNumber(2), bankCountryCode, bankCode, customerId, customerSystemId, status),
            Verarbeitungsvorbereitung(generator.getNextSegmentNumber(), bpdVersion, updVersion, language, productName, productVersion),
            ZweiSchrittTanEinreichung(generator.getNextSegmentNumber(), TanProcess.TanProcess4, "HKIDN")
        ))
    }


    open fun createMessage(signMessage: Boolean, encryptMessage: Boolean, bankCountryCode: Int, bankCode: String, customerId: String,
                           payloadSegments: List<Segment>): String {

        var payload = payloadSegments
        val partyIdentification = "0"
        val date = utils.formatDateTodayAsInt()
        val time = utils.formatTimeNowAsInt()

        if (signMessage) {
            payload = signPayload(2, partyIdentification, date, time, bankCountryCode, bankCode, customerId, payload)
        }

        if (encryptMessage) {
            payload = encryptPayload(partyIdentification, date, time, bankCountryCode, bankCode, customerId, payload)
        }

        val formattedPayload = formatPayload(payload)

        val messageSize = formattedPayload.length + MessageHeaderLength + MessageClosingLength + AddedSeparatorsLength
        val messageNumber = Nachrichtennummer.FirstMessageNumber

        val header = Nachrichtenkopf(ISegmentNumberGenerator.FirstSegmentNumber, messageSize, "0", messageNumber)

        val closing = Nachrichtenabschluss(generator.getNextSegmentNumber(), messageNumber)

        return listOf(header.format(), formattedPayload, closing.format())
            .joinToString(Nachricht.SegmentSeparator, postfix = Nachricht.SegmentSeparator)
    }


    protected open fun signPayload(headerSegmentNumber: Int, partyIdentification: String, date: Int, time: Int,
                                   bankCountryCode: Int, bankCode: String, customerId: String,
                                   payloadSegments: List<Segment>): List<Segment> {
        val controlReference = "1" // TODO

        val signatureHeader = PinTanSignaturkopf(
            headerSegmentNumber,
            Sicherheitsfunktion.PIN_TAN_911, // TODO
            controlReference,
            "0",
            utils.formatDateTodayAsInt(),
            utils.formatTimeNowAsInt(),
            bankCountryCode,
            bankCode,
            customerId
        )

        val signatureClosing = Signaturabschluss(
            generator.getNextSegmentNumber(),
            controlReference,
            "12345" // TODO
        )

        return listOf(signatureHeader, *payloadSegments.toTypedArray(), signatureClosing)
    }


    private fun encryptPayload(partyIdentification: String, date: Int, time: Int,
                               bankCountryCode: Int, bankCode: String, customerId: String, payload: List<Segment>): List<Segment> {

        val encryptionHeader = PinTanVerschluesselungskopf(partyIdentification, date, time, bankCountryCode, bankCode, customerId)

        val encryptedData = VerschluesselteDaten(formatPayload(payload) + Nachricht.SegmentSeparator)

        return listOf(encryptionHeader, encryptedData)
    }


    protected open fun formatPayload(payload: List<Segment>): String {
        return payload.joinToString(Nachricht.SegmentSeparator) { it.format() }
    }

}