com.jtransc.sourcemaps.sourcemaps.kt Maven / Gradle / Ivy

Go to download
package com.jtransc.sourcemaps

import com.jtransc.json.Json
import java.util.*

object Sourcemaps {
	fun decodeRaw(str: String): List>> {
		return str.split(";").map {
			it.split(",").map { Base64Vlq.decode(it) }
		}
	}

	fun encodeRaw(items: List>>): String {
		return items.map {
			it.map { Base64Vlq.encode(it) }.joinToString(",")
		}.joinToString(";")
	}

	data class MappingItem(val sourceIndex: Int, val sourceLine: Int, val sourceColumn: Int, val targetColumn: Int)
	data class MappingRow(val mappings: List)
	data class MappingFile(val rows: List)

	data class TargetMapping(val name: String, val line: Int, val column: Int)

	data class SourceMap(
		val version: Int = 3,
		val file: String = "unknown.js",
		val sourceRoot: String = "file:///",
		val sources: List = arrayListOf(),
		val names: List = arrayListOf(),
		val mappings: String = ""
	) {
		val mappingFile = decode(mappings)

		fun decodePos(line: Int, column: Int = 0): TargetMapping? {
			for (item in mappingFile.rows.getOrNull(line)?.mappings ?: listOf()) {
				if (item.targetColumn >= column) {
					return TargetMapping(sourceRoot + sources[item.sourceIndex], item.sourceLine, item.sourceColumn)
				}
			}
			return null
		}
	}

	fun decodeFile(str: String) = Json.decodeTo(str)

	fun decode(str: String): MappingFile {
		var sourceLine = 0
		var sourceIndex = 0
		var sourceColumn = 0

		return MappingFile(decodeRaw(str).map { row ->
			var targetColumn = 0

			MappingRow(row.flatMap { info ->
				if (info.size >= 4) {
					targetColumn += info[0]
					sourceIndex += info[1]
					sourceLine += info[2]
					sourceColumn += info[3]
					listOf(MappingItem(sourceIndex, sourceLine, sourceColumn, targetColumn))
				} else {
					listOf()
				}
			})
		})
	}

	fun encode(mapping: MappingFile): String {
		var sourceLine = 0
		var sourceIndex = 0
		var sourceColumn = 0
		val lists = mapping.rows.map { row ->
			var targetColumn = 0
			row.mappings.map {
				listOf(
					it.targetColumn - targetColumn,
					it.sourceIndex - sourceIndex,
					it.sourceLine - sourceLine,
					it.sourceColumn - sourceColumn
				).apply {
					sourceIndex = it.sourceIndex
					sourceLine = it.sourceLine
					sourceColumn = it.sourceColumn
					targetColumn = it.targetColumn
				}
			}
		}
		return encodeRaw(lists)
	}

	fun encodeFile(targetPath: String, targetContent: String, source: String, mappings: HashMap): String {
		//(0 until targetContent.count { it == '\n' }).map {}
		val mapping = MappingFile((0 until (mappings.keys.max() ?: 1)).map {
			val targetLine = mappings[it]
			MappingRow(if (targetLine == null) listOf() else listOf(MappingItem(0, targetLine, 0, 0)))
		})
		return Json.encode(mapOf(
			"version" to 3,
			"file" to targetPath,
			"sources" to arrayListOf(source),
			"names" to arrayListOf(),
			"mappings" to encode(mapping)
		), prettify = true)
	}

	fun encodeFile(files: List, mappings: Map): String {
		val mapping = MappingFile((0 until (mappings.keys.max() ?: 1)).map {
			val item = mappings[it]
			MappingRow(if (item == null) listOf() else listOf(item))
		})

		return Json.encode(mapOf(
			"version" to 3,
			"file" to "program.js",
			"sources" to files,
			"names" to arrayListOf(),
			"mappings" to encode(mapping)
		), prettify = true)
	}
}