summaryrefslogtreecommitdiff
path: root/formats/json/commonMain/src/kotlinx/serialization/json/internal/Composers.kt
blob: 2bc080f9405cd75fc743de108317298ec6ca81ab (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
/*
 * Copyright 2017-2021 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license.
 */

@file:OptIn(ExperimentalSerializationApi::class)
package kotlinx.serialization.json.internal

import kotlinx.serialization.ExperimentalSerializationApi
import kotlinx.serialization.json.Json
import kotlin.jvm.JvmField

internal fun Composer(sb: JsonStringBuilder, json: Json): Composer =
    if (json.configuration.prettyPrint) ComposerWithPrettyPrint(sb, json) else Composer(sb)

@OptIn(ExperimentalSerializationApi::class)
internal open class Composer(@JvmField internal val sb: JsonStringBuilder) {
    var writingFirst = true
        protected set

    open fun indent() {
        writingFirst = true
    }

    open fun unIndent() = Unit

    open fun nextItem() {
        writingFirst = false
    }

    open fun space() = Unit

    fun print(v: Char) = sb.append(v)
    fun print(v: String) = sb.append(v)
    open fun print(v: Float) = sb.append(v.toString())
    open fun print(v: Double) = sb.append(v.toString())
    open fun print(v: Byte) = sb.append(v.toLong())
    open fun print(v: Short) = sb.append(v.toLong())
    open fun print(v: Int) = sb.append(v.toLong())
    open fun print(v: Long) = sb.append(v)
    open fun print(v: Boolean) = sb.append(v.toString())
    fun printQuoted(value: String): Unit = sb.appendQuoted(value)
}

@ExperimentalUnsignedTypes
internal class ComposerForUnsignedNumbers(sb: JsonStringBuilder) : Composer(sb) {
    override fun print(v: Int) {
        return super.print(v.toUInt().toString())
    }

    override fun print(v: Long) {
        return super.print(v.toULong().toString())
    }

    override fun print(v: Byte) {
        return super.print(v.toUByte().toString())
    }

    override fun print(v: Short) {
        return super.print(v.toUShort().toString())
    }
}

internal class ComposerWithPrettyPrint(
    sb: JsonStringBuilder,
    private val json: Json
) : Composer(sb) {
    private var level = 0

    override fun indent() {
        writingFirst = true
        level++
    }

    override fun unIndent() {
        level--
    }

    override fun nextItem() {
        writingFirst = false
        print("\n")
        repeat(level) { print(json.configuration.prettyPrintIndent) }
    }

    override fun space() {
        print(' ')
    }
}