1 package net.pterodactylus.rhynodge.webpages.weather.wettercom
3 import net.pterodactylus.rhynodge.Filter
4 import net.pterodactylus.rhynodge.State
5 import net.pterodactylus.rhynodge.states.FailedState
6 import net.pterodactylus.rhynodge.states.HtmlState
7 import net.pterodactylus.rhynodge.webpages.weather.HourState
8 import net.pterodactylus.rhynodge.webpages.weather.WeatherState
9 import net.pterodactylus.rhynodge.webpages.weather.WindDirection
10 import net.pterodactylus.rhynodge.webpages.weather.toWindDirection
11 import org.jsoup.nodes.Document
12 import org.jsoup.nodes.Element
13 import java.time.LocalDateTime
14 import java.time.ZoneId.of
15 import java.time.ZonedDateTime
16 import java.time.format.DateTimeFormatter
19 * HTML parser for [wetter.com](https://www.wetter.com/).
21 * @author [David ‘Bombe’ Roden](mailto:bombe@pterodactylus.net)
23 class WetterComFilter : Filter {
25 private val dateTimeFormatter = DateTimeFormatter.ofPattern("dd.MM.yyyy HH:mm")
27 override fun filter(state: State): State {
28 if (state.success().not()) {
29 return FailedState.from(state)
31 if (state !is HtmlState) {
32 throw IllegalArgumentException("")
35 return parseWetterComState(state)
38 private fun parseWetterComState(state: HtmlState): State {
39 val dateTime = parseDateTime(state.document()) ?: return FailedState(IllegalArgumentException("no date present"))
40 val wetterComState = WeatherState("wetter.com", dateTime)
41 parseHourStates(state.document()).forEach { wetterComState += it }
45 private fun parseDateTime(document: Document): ZonedDateTime? {
46 val dateElement = document.select("#furtherDetails .portable-mb h3")
47 .single()?.text()?.split(",")?.get(1)?.trim() ?: return null
48 val timeElement = document.select(".weather-strip--detail .delta.palm-hide")
49 .first()?.text()?.split(" ")?.first() ?: return null
50 return LocalDateTime.from(dateTimeFormatter.parse("%s %s".format(dateElement, timeElement))).atZone(of("Europe/Berlin"))
53 private fun parseHourStates(document: Document): List<HourState> {
54 return document.select(".weather-strip--detail").mapIndexed { index, element -> parseHourState(index, element) }
57 private fun parseHourState(index: Int, hourElement: Element): HourState {
60 temperature = hourElement.temperature,
61 rainProbability = hourElement.rainProbability,
62 rainAmount = hourElement.rainAmount,
63 windDirection = hourElement.windDirection,
64 windSpeed = hourElement.windSpeed,
65 description = hourElement.description,
66 image = hourElement.imageUrl
70 private val Element.temperature: Int
71 get() = extractText(".weather-strip__2 .item-weathericon .palm-hide").filter { (it >= '0') and (it <= '9') }.toInt()
73 private val Element.rainProbability: Double
74 get() = extractText(".weather-strip__3 .text--left:eq(0) .flag__body span:eq(0)").filter { (it >= '0') and (it <= '9') }.toDouble() / 100.0
76 private val Element.rainAmount: Double
77 get() = extractText(".weather-strip__3 .text--left:eq(0) .flag__body span:eq(1)").trim().split(" ").getOrNull(1)?.toDouble() ?: 0.0
79 private val Element.windDirection: WindDirection
80 get() = extractText(".weather-strip__3 .text--left:eq(1) .flag__body span:eq(0)").trim().split(",")[0].toWindDirection()
82 private val Element.windSpeed: Int
83 get() = extractText(".weather-strip__3 .text--left:eq(1) .flag__body span:eq(0)").split(Regex("[, ]+"))[1].toInt()
85 private val Element.description: String
86 get() = extractText(".weather-strip__1 .vhs-text--small")
88 private val Element.imageUrl: String
89 get() = select(".weather-strip__2 img").firstOrNull()?.attr("src") ?: ""
91 private fun Element.extractText(selector: String) = select(selector).firstOrNull()?.text() ?: ""