Mirrors/grayjay
1
0
Fork 0
mirror of https://gitlab.futo.org/videostreaming/grayjay.git synced 2025-05-05 17:14:37 +02:00
Code Issues Packages Projects Releases Wiki Activity

revert style changes

Browse Source
This commit is contained in:
Kai DeLorenzo 2024-11-19 11:45:54 -05:00
parent 8f6eac7ca2
commit 4dcc2dd0ca
No known key found for this signature in database
1 changed files with 124 additions and 109 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

211
app/src/main/java/com/futo/platformplayer/parsers/HLS.kt
View File

@ -46,53 +46,36 @@ class HLS {
} }
} }
return MasterPlaylist( return MasterPlaylist(variantPlaylists, mediaRenditions, sessionDataList, independentSegments)
variantPlaylists, mediaRenditions, sessionDataList, independentSegments
)
} }
fun parseVariantPlaylist(content: String, sourceUrl: String): VariantPlaylist { fun parseVariantPlaylist(content: String, sourceUrl: String): VariantPlaylist {
val lines = content.lines() val lines = content.lines()
val version = val version = lines.find { it.startsWith("#EXT-X-VERSION:") }?.substringAfter(":")?.toIntOrNull()
lines.find { it.startsWith("#EXT-X-VERSION:") }?.substringAfter(":")?.toIntOrNull() val targetDuration = lines.find { it.startsWith("#EXT-X-TARGETDURATION:") }?.substringAfter(":")?.toIntOrNull()
val targetDuration = val mediaSequence = lines.find { it.startsWith("#EXT-X-MEDIA-SEQUENCE:") }?.substringAfter(":")?.toLongOrNull()
lines.find { it.startsWith("#EXT-X-TARGETDURATION:") }?.substringAfter(":") val discontinuitySequence = lines.find { it.startsWith("#EXT-X-DISCONTINUITY-SEQUENCE:") }?.substringAfter(":")?.toIntOrNull()
?.toIntOrNull() val programDateTime = lines.find { it.startsWith("#EXT-X-PROGRAM-DATE-TIME:") }?.substringAfter(":")?.let {
val mediaSequence =
lines.find { it.startsWith("#EXT-X-MEDIA-SEQUENCE:") }?.substringAfter(":")
?.toLongOrNull()
val discontinuitySequence =
lines.find { it.startsWith("#EXT-X-DISCONTINUITY-SEQUENCE:") }?.substringAfter(":")
?.toIntOrNull()
val programDateTime =
lines.find { it.startsWith("#EXT-X-PROGRAM-DATE-TIME:") }?.substringAfter(":")
?.let {
ZonedDateTime.parse(it, DateTimeFormatter.ISO_DATE_TIME) ZonedDateTime.parse(it, DateTimeFormatter.ISO_DATE_TIME)
} }
val playlistType = val playlistType = lines.find { it.startsWith("#EXT-X-PLAYLIST-TYPE:") }?.substringAfter(":")
lines.find { it.startsWith("#EXT-X-PLAYLIST-TYPE:") }?.substringAfter(":") val streamInfo = lines.find { it.startsWith("#EXT-X-STREAM-INF:") }?.let { parseStreamInfo(it) }
val streamInfo =
lines.find { it.startsWith("#EXT-X-STREAM-INF:") }?.let { parseStreamInfo(it) }
val segments = mutableListOf<Segment>() val segments = mutableListOf<Segment>()
var currentSegment: MediaSegment? = null var currentSegment: MediaSegment? = null
lines.forEach { line -> lines.forEach { line ->
when { when {
line.startsWith("#EXTINF:") -> { line.startsWith("#EXTINF:") -> {
val duration = val duration = line.substringAfter(":").substringBefore(",").toDoubleOrNull()
line.substringAfter(":").substringBefore(",").toDoubleOrNull()
?: throw Exception("Invalid segment duration format") ?: throw Exception("Invalid segment duration format")
currentSegment = MediaSegment(duration = duration) currentSegment = MediaSegment(duration = duration)
} }
line == "#EXT-X-DISCONTINUITY" -> { line == "#EXT-X-DISCONTINUITY" -> {
segments.add(DiscontinuitySegment()) segments.add(DiscontinuitySegment())
} }
line =="#EXT-X-ENDLIST" -> { line =="#EXT-X-ENDLIST" -> {
segments.add(EndListSegment()) segments.add(EndListSegment())
} }
else -> { else -> {
currentSegment?.let { currentSegment?.let {
it.uri = resolveUrl(sourceUrl, line) it.uri = resolveUrl(sourceUrl, line)
@ -106,60 +89,40 @@ class HLS {
return VariantPlaylist(version, targetDuration, mediaSequence, discontinuitySequence, programDateTime, playlistType, streamInfo, segments) return VariantPlaylist(version, targetDuration, mediaSequence, discontinuitySequence, programDateTime, playlistType, streamInfo, segments)
} }
fun parseAndGetVideoSources( fun parseAndGetVideoSources(source: Any, content: String, url: String): List<HLSVariantVideoUrlSource> {
source: Any, content: String, url: String
): List<HLSVariantVideoUrlSource> {
val masterPlaylist: MasterPlaylist val masterPlaylist: MasterPlaylist
try { try {
masterPlaylist = parseMasterPlaylist(content, url) masterPlaylist = parseMasterPlaylist(content, url)
return masterPlaylist.getVideoSources() return masterPlaylist.getVideoSources()
} catch (e: Throwable) { } catch (e: Throwable) {
if (content.lines().any { it.startsWith("#EXTINF:") }) { if (content.lines().any { it.startsWith("#EXTINF:") }) {
return when (source) { return if (source is IHLSManifestSource) {
is IHLSManifestSource -> { listOf(HLSVariantVideoUrlSource("variant", 0, 0, "application/vnd.apple.mpegurl", "", null, 0, false, url))
listOf( } else if (source is IHLSManifestAudioSource) {
HLSVariantVideoUrlSource("variant", 0, 0, "application/vnd.apple.mpegurl", "", null, 0, false, url)
)
}
is IHLSManifestAudioSource -> {
listOf() listOf()
} } else {
else -> {
throw NotImplementedError() throw NotImplementedError()
} }
}
} else { } else {
throw e throw e
} }
} }
} }
fun parseAndGetAudioSources( fun parseAndGetAudioSources(source: Any, content: String, url: String): List<HLSVariantAudioUrlSource> {
source: Any, content: String, url: String
): List<HLSVariantAudioUrlSource> {
val masterPlaylist: MasterPlaylist val masterPlaylist: MasterPlaylist
try { try {
masterPlaylist = parseMasterPlaylist(content, url) masterPlaylist = parseMasterPlaylist(content, url)
return masterPlaylist.getAudioSources() return masterPlaylist.getAudioSources()
} catch (e: Throwable) { } catch (e: Throwable) {
if (content.lines().any { it.startsWith("#EXTINF:") }) { if (content.lines().any { it.startsWith("#EXTINF:") }) {
return when (source) { return if (source is IHLSManifestSource) {
is IHLSManifestSource -> {
listOf() listOf()
} } else if (source is IHLSManifestAudioSource) {
listOf(HLSVariantAudioUrlSource("variant", 0, "application/vnd.apple.mpegurl", "", "", null, false, url))
is IHLSManifestAudioSource -> { } else {
listOf(
HLSVariantAudioUrlSource("variant", 0, "application/vnd.apple.mpegurl", "", "", null, false, url)
)
}
else -> {
throw NotImplementedError() throw NotImplementedError()
} }
}
} else { } else {
throw e throw e
} }
@ -183,7 +146,15 @@ class HLS {
private fun parseStreamInfo(content: String): StreamInfo { private fun parseStreamInfo(content: String): StreamInfo {
val attributes = parseAttributes(content) val attributes = parseAttributes(content)
return StreamInfo( return StreamInfo(
bandwidth = attributes["BANDWIDTH"]?.toIntOrNull(), resolution = attributes["RESOLUTION"], codecs = attributes["CODECS"], frameRate = attributes["FRAME-RATE"], videoRange = attributes["VIDEO-RANGE"], audio = attributes["AUDIO"], video = attributes["VIDEO"], subtitles = attributes["SUBTITLES"], closedCaptions = attributes["CLOSED-CAPTIONS"] bandwidth = attributes["BANDWIDTH"]?.toIntOrNull(),
resolution = attributes["RESOLUTION"],
codecs = attributes["CODECS"],
frameRate = attributes["FRAME-RATE"],
videoRange = attributes["VIDEO-RANGE"],
audio = attributes["AUDIO"],
video = attributes["VIDEO"],
subtitles = attributes["SUBTITLES"],
closedCaptions = attributes["CLOSED-CAPTIONS"]
) )
} }
@ -191,7 +162,14 @@ class HLS {
val attributes = parseAttributes(line) val attributes = parseAttributes(line)
val uri = attributes["URI"]?.let { resolveUrl(baseUrl, it) } val uri = attributes["URI"]?.let { resolveUrl(baseUrl, it) }
return MediaRendition( return MediaRendition(
type = attributes["TYPE"], uri = uri, groupID = attributes["GROUP-ID"], language = attributes["LANGUAGE"], name = attributes["NAME"], isDefault = attributes["DEFAULT"]?.yesNoToBoolean(), isAutoSelect = attributes["AUTOSELECT"]?.yesNoToBoolean(), isForced = attributes["FORCED"]?.yesNoToBoolean() type = attributes["TYPE"],
uri = uri,
groupID = attributes["GROUP-ID"],
language = attributes["LANGUAGE"],
name = attributes["NAME"],
isDefault = attributes["DEFAULT"]?.yesNoToBoolean(),
isAutoSelect = attributes["AUTOSELECT"]?.yesNoToBoolean(),
isForced = attributes["FORCED"]?.yesNoToBoolean()
) )
} }
@ -224,55 +202,99 @@ class HLS {
private val _quoteList = listOf("GROUP-ID", "NAME", "URI", "CODECS", "AUDIO", "VIDEO") private val _quoteList = listOf("GROUP-ID", "NAME", "URI", "CODECS", "AUDIO", "VIDEO")
private fun shouldQuote(key: String, value: String?): Boolean { private fun shouldQuote(key: String, value: String?): Boolean {
if (value == null) return false if (value == null)
return false;
if (value.contains(',')) return true if (value.contains(','))
return true;
return _quoteList.contains(key) return _quoteList.contains(key)
} }
private fun appendAttributes(stringBuilder: StringBuilder, vararg attributes: Pair<String, String?>) {
private fun appendAttributes( attributes.filter { it.second != null }
stringBuilder: StringBuilder, vararg attributes: Pair<String, String?> .joinToString(",") {
) {
attributes.filter { it.second != null }.joinToString(",") {
val value = it.second val value = it.second
"${it.first}=${if (shouldQuote(it.first, it.second)) "\"$value\"" else value}" "${it.first}=${if (shouldQuote(it.first, it.second)) "\"$value\"" else value}"
}.let { if (it.isNotEmpty()) stringBuilder.append(it) } }
.let { if (it.isNotEmpty()) stringBuilder.append(it) }
} }
} }
data class SessionData( data class SessionData(
val dataId: String, val value: String val dataId: String,
val value: String
) { ) {
fun toM3U8Line(): String = buildString { fun toM3U8Line(): String = buildString {
append("#EXT-X-SESSION-DATA:") append("#EXT-X-SESSION-DATA:")
appendAttributes( appendAttributes(this,
this, "DATA-ID" to dataId, "VALUE" to value "DATA-ID" to dataId,
"VALUE" to value
) )
append("\n") append("\n")
} }
} }
data class StreamInfo(val bandwidth: Int?, val resolution: String?, val codecs: String?, val frameRate: String?, val videoRange: String?, val audio: String?, val video: String?, val subtitles: String?, val closedCaptions: String?) { data class StreamInfo(
val bandwidth: Int?,
val resolution: String?,
val codecs: String?,
val frameRate: String?,
val videoRange: String?,
val audio: String?,
val video: String?,
val subtitles: String?,
val closedCaptions: String?
) {
fun toM3U8Line(): String = buildString { fun toM3U8Line(): String = buildString {
append("#EXT-X-STREAM-INF:") append("#EXT-X-STREAM-INF:")
appendAttributes(this, "BANDWIDTH" to bandwidth?.toString(), "RESOLUTION" to resolution, "CODECS" to codecs, "FRAME-RATE" to frameRate, "VIDEO-RANGE" to videoRange, "AUDIO" to audio, "VIDEO" to video, "SUBTITLES" to subtitles, "CLOSED-CAPTIONS" to closedCaptions) appendAttributes(this,
"BANDWIDTH" to bandwidth?.toString(),
"RESOLUTION" to resolution,
"CODECS" to codecs,
"FRAME-RATE" to frameRate,
"VIDEO-RANGE" to videoRange,
"AUDIO" to audio,
"VIDEO" to video,
"SUBTITLES" to subtitles,
"CLOSED-CAPTIONS" to closedCaptions
)
append("\n") append("\n")
} }
} }
data class MediaRendition( data class MediaRendition(
val type: String?, val uri: String?, val groupID: String?, val language: String?, val name: String?, val isDefault: Boolean?, val isAutoSelect: Boolean?, val isForced: Boolean? val type: String?,
val uri: String?,
val groupID: String?,
val language: String?,
val name: String?,
val isDefault: Boolean?,
val isAutoSelect: Boolean?,
val isForced: Boolean?
) { ) {
fun toM3U8Line(): String = buildString { fun toM3U8Line(): String = buildString {
append("#EXT-X-MEDIA:") append("#EXT-X-MEDIA:")
appendAttributes(this, "TYPE" to type, "URI" to uri, "GROUP-ID" to groupID, "LANGUAGE" to language, "NAME" to name, "DEFAULT" to isDefault.toYesNo(), "AUTOSELECT" to isAutoSelect.toYesNo(), "FORCED" to isForced.toYesNo()) appendAttributes(this,
"TYPE" to type,
"URI" to uri,
"GROUP-ID" to groupID,
"LANGUAGE" to language,
"NAME" to name,
"DEFAULT" to isDefault.toYesNo(),
"AUTOSELECT" to isAutoSelect.toYesNo(),
"FORCED" to isForced.toYesNo()
)
append("\n") append("\n")
} }
} }
data class MasterPlaylist(val variantPlaylistsRefs: List<VariantPlaylistReference>, val mediaRenditions: List<MediaRendition>, val sessionDataList: List<SessionData>, val independentSegments: Boolean) { data class MasterPlaylist(
val variantPlaylistsRefs: List<VariantPlaylistReference>,
val mediaRenditions: List<MediaRendition>,
val sessionDataList: List<SessionData>,
val independentSegments: Boolean
) {
fun buildM3U8(): String { fun buildM3U8(): String {
val builder = StringBuilder() val builder = StringBuilder()
builder.append("#EXTM3U\n") builder.append("#EXTM3U\n")
@ -305,14 +327,8 @@ class HLS {
height = resolutionTokens[1].toIntOrNull() height = resolutionTokens[1].toIntOrNull()
} }
val suffix = listOf( val suffix = listOf(it.streamInfo.video, it.streamInfo.codecs).mapNotNull { x -> x?.ifEmpty { null } }.joinToString(", ")
it.streamInfo.video, it.streamInfo.codecs HLSVariantVideoUrlSource(suffix, width ?: 0, height ?: 0, "application/vnd.apple.mpegurl", it.streamInfo.codecs ?: "", it.streamInfo.bandwidth, 0, false, it.url)
).mapNotNull { x -> x?.ifEmpty { null } }.joinToString(", ")
HLSVariantVideoUrlSource(
suffix, width ?: 0, height
?: 0, "application/vnd.apple.mpegurl", it.streamInfo.codecs
?: "", it.streamInfo.bandwidth, 0, false, it.url
)
} }
} }
@ -322,13 +338,9 @@ class HLS {
return@mapNotNull null return@mapNotNull null
} }
val suffix = listOf(it.language, it.groupID).mapNotNull { x -> x?.ifEmpty { null } } val suffix = listOf(it.language, it.groupID).mapNotNull { x -> x?.ifEmpty { null } }.joinToString(", ")
.joinToString(", ")
return@mapNotNull when (it.type) { return@mapNotNull when (it.type) {
"AUDIO" -> HLSVariantAudioUrlSource(it.name?.ifEmpty { "Audio (${suffix})" } "AUDIO" -> HLSVariantAudioUrlSource(it.name?.ifEmpty { "Audio (${suffix})" } ?: "Audio (${suffix})", 0, "application/vnd.apple.mpegurl", "", it.language ?: "", null, false, it.uri)
?: "Audio (${suffix})", 0, "application/vnd.apple.mpegurl", "", it.language
?: "", null, false, it.uri)
else -> null else -> null
} }
} }
@ -340,12 +352,9 @@ class HLS {
return@mapNotNull null return@mapNotNull null
} }
val suffix = listOf(it.language, it.groupID).mapNotNull { x -> x?.ifEmpty { null } } val suffix = listOf(it.language, it.groupID).mapNotNull { x -> x?.ifEmpty { null } }.joinToString(", ")
.joinToString(", ")
return@mapNotNull when (it.type) { return@mapNotNull when (it.type) {
"SUBTITLE" -> HLSVariantSubtitleUrlSource(it.name?.ifEmpty { "Subtitle (${suffix})" } "SUBTITLE" -> HLSVariantSubtitleUrlSource(it.name?.ifEmpty { "Subtitle (${suffix})" } ?: "Subtitle (${suffix})", it.uri, "application/vnd.apple.mpegurl")
?: "Subtitle (${suffix})", it.uri, "application/vnd.apple.mpegurl")
else -> null else -> null
} }
} }
@ -360,7 +369,14 @@ class HLS {
} }
data class VariantPlaylist( data class VariantPlaylist(
val version: Int?, val targetDuration: Int?, val mediaSequence: Long?, val discontinuitySequence: Int?, val programDateTime: ZonedDateTime?, val playlistType: String?, val streamInfo: StreamInfo?, val segments: List<Segment> val version: Int?,
val targetDuration: Int?,
val mediaSequence: Long?,
val discontinuitySequence: Int?,
val programDateTime: ZonedDateTime?,
val playlistType: String?,
val streamInfo: StreamInfo?,
val segments: List<Segment>
) { ) {
fun buildM3U8(): String = buildString { fun buildM3U8(): String = buildString {
append("#EXTM3U\n") append("#EXTM3U\n")
@ -369,9 +385,7 @@ class HLS {
mediaSequence?.let { append("#EXT-X-MEDIA-SEQUENCE:$it\n") } mediaSequence?.let { append("#EXT-X-MEDIA-SEQUENCE:$it\n") }
discontinuitySequence?.let { append("#EXT-X-DISCONTINUITY-SEQUENCE:$it\n") } discontinuitySequence?.let { append("#EXT-X-DISCONTINUITY-SEQUENCE:$it\n") }
playlistType?.let { append("#EXT-X-PLAYLIST-TYPE:$it\n") } playlistType?.let { append("#EXT-X-PLAYLIST-TYPE:$it\n") }
programDateTime?.let { programDateTime?.let { append("#EXT-X-PROGRAM-DATE-TIME:${it.format(DateTimeFormatter.ISO_DATE_TIME)}\n") }
append("#EXT-X-PROGRAM-DATE-TIME:${it.format(DateTimeFormatter.ISO_DATE_TIME)}\n")
}
streamInfo?.let { append(it.toM3U8Line()) } streamInfo?.let { append(it.toM3U8Line()) }
segments.forEach { segment -> segments.forEach { segment ->
@ -385,7 +399,8 @@ class HLS {
} }
data class MediaSegment ( data class MediaSegment (
val duration: Double, var uri: String = "" val duration: Double,
var uri: String = ""
) : Segment() { ) : Segment() {
override fun toM3U8Line(): String = buildString { override fun toM3U8Line(): String = buildString {
append("#EXTINF:${duration},\n") append("#EXTINF:${duration},\n")