src/main/kotlin/net/pterodactylus/rhynodge/webpages/weather/wettercom/WetterComFilter.kt

   1 package net.pterodactylus.rhynodge.webpages.weather.wettercom
   2
   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 org.jsoup.nodes.Document
   8 import org.jsoup.nodes.Element
   9 import java.time.LocalDateTime
  10 import java.time.ZoneId.of
  11 import java.time.ZonedDateTime
  12 import java.time.format.DateTimeFormatter
  13
  14 /**
  15  * HTML parser for [wetter.com](https://www.wetter.com/).
  16  *
  17  * @author [David ‘Bombe’ Roden](mailto:bombe@pterodactylus.net)
  18  */
  19 class WetterComFilter : Filter {
  20
  21     private val dateTimeFormatter = DateTimeFormatter.ofPattern("dd.MM.yyyy HH:mm")
  22
  23     override fun filter(state: State): State {
  24         if (state.success().not()) {
  25             return FailedState.from(state)
  26         }
  27         if (state !is HtmlState) {
  28             throw IllegalArgumentException("")
  29         }
  30
  31         return parseWetterComState(state)
  32     }
  33
  34     private fun parseWetterComState(state: HtmlState): State {
  35         val dateTime = parseDateTime(state.document()) ?: return FailedState(IllegalArgumentException("no date present"))
  36         val wetterComState = WetterComState(dateTime)
  37         parseHourStates(state.document()).forEach { wetterComState.addHour(it) }
  38         return wetterComState
  39     }
  40
  41     private fun parseDateTime(document: Document): ZonedDateTime? {
  42         val dateElement = document.select("#furtherDetails .portable-mb h3")
  43                 .single()?.text()?.split(",")?.get(1)?.trim() ?: return null
  44         val timeElement = document.select(".weather-strip--detail .delta.palm-hide")
  45                 .first()?.text()?.split(" ")?.first() ?: return null
  46         return LocalDateTime.from(dateTimeFormatter.parse("%s %s".format(dateElement, timeElement))).atZone(of("Europe/Berlin"))
  47     }
  48
  49     private fun parseHourStates(document: Document): List<HourState> {
  50         return document.select(".weather-strip--detail").mapIndexed { index, element -> parseHourState(index, element) }
  51     }
  52
  53     private fun parseHourState(index: Int, hourElement: Element): HourState {
  54         return HourState.atHour(index)
  55                 .temperature(parseTemperature(hourElement))
  56                 .rainProbability(parseRainProbability(hourElement))
  57                 .rainAmount(parseRainAmount(hourElement))
  58                 .windFrom(parseWindDirection(hourElement))
  59                 .at(parseWindSpeed(hourElement))
  60                 .describedAs(parseDescription(hourElement))
  61                 .withImage(parseImageUrl(hourElement))
  62                 .build()
  63     }
  64
  65     private fun parseTemperature(hourElement: Element) =
  66             hourElement.extractText(".weather-strip__2 .item-weathericon .palm-hide").filter { (it >= '0') and (it <= '9') }.toDouble()
  67
  68     private fun parseRainProbability(hourElement: Element) =
  69             hourElement.extractText(".weather-strip__3 .text--left:eq(0) .flag__body span:eq(0)").filter { (it >= '0') and (it <= '9') }.toDouble() / 100.0
  70
  71     private fun parseRainAmount(hourElement: Element) =
  72             hourElement.extractText(".weather-strip__3 .text--left:eq(0) .flag__body span:eq(1)").trim().split(" ").getOrNull(1)?.toDouble() ?: 0.0
  73
  74     private fun parseWindDirection(hourElement: Element) =
  75             hourElement.extractText(".weather-strip__3 .text--left:eq(1) .flag__body span:eq(0)").trim().split(",")[0].toWindDirection()
  76
  77     private fun parseWindSpeed(hourElement: Element) =
  78             hourElement.extractText(".weather-strip__3 .text--left:eq(1) .flag__body span:eq(0)").split(Regex("[, ]+"))[1].toDouble()
  79
  80     private fun parseDescription(hourElement: Element) =
  81             hourElement.extractText(".weather-strip__1 .vhs-text--small")
  82
  83     private fun parseImageUrl(hourElement: Element) =
  84             hourElement.select(".weather-strip__2 img").firstOrNull()?.attr("src") ?: ""
  85
  86     private fun Element.extractText(selector: String) = select(selector).firstOrNull()?.text() ?: ""
  87
  88 }