/**
 * ProjectLaogai
 *
 * Copyright 2019  <seil0@mosad.xyz>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
 * MA 02110-1301, USA.
 *
 */

package org.mosad.seil0.projectlaogai.hsoparser

import org.jsoup.Jsoup
import java.util.*

class MensaParser {

    /**
     * returns the mensa menu for the a week
     */
    fun getMensaMenu(menuLink: String): ArrayList<Meal> {
        val mealList = ArrayList<Meal>()
        val menuHTML = Jsoup.connect(menuLink).get()

        menuHTML.select("#speiseplan-tabs").select("div.tab-content").select("div.menu-tagesplan").forEachIndexed { _, element ->
            val day = element.select("h3").text()
            for (i in 0 .. (element.select("div.row  h4").size - 1)) {
                try {
                    val heading = element.select("div.row  h4")[i].text()
                    val parts = ArrayList<String>(element.select("div.row").select("div.menu-info")[i].html().substringBefore("<span").replace("<br>", " ").split("\n"))
                    val additives = element.select("div.row").select("div.menu-info")[i].select("span.show-with-allergenes").text()

                    mealList.add(Meal(day, heading, parts, additives))
                } catch (e: Exception) {
                    //println("Oooups! Something went wrong: ${e.printStackTrace()}")
                }
            }
        }

        return  mealList
    }

    /**
     * return the link of the menus of the next week
     */
    fun getMenuLinkNextWeek(menuLink: String): String {
        val menuHTML = Jsoup.connect(menuLink).get()

        return "https://www.swfr.de" + menuHTML.select("#speiseplan-tabs").select("a.next-week").attr("href")
    }

    /**
     * return the mensa menu of a given day (Mon - Sat)
     */
    fun getMensaMenuDay(mealList: ArrayList<Meal>, day: Int): ArrayList<Meal> {
        val dayMenus = ArrayList<Meal>()
        val strDay: String = when (day) {
            Calendar.MONDAY -> "Mon"
            Calendar.TUESDAY -> "Die"
            Calendar.WEDNESDAY -> "Mit"
            Calendar.THURSDAY -> "Don"
            Calendar.FRIDAY -> "Fre"
            Calendar.SATURDAY -> "Sam"
            else -> "TODAY"
        }

        for (meal in mealList) {
            if (meal.day.contains(strDay))
                dayMenus.add(meal)
        }
        return dayMenus
    }
}