GigaProjects

package com.gigaprojects.gigaweather.data

import androidx.room.Entity
import androidx.room.PrimaryKey
import androidx.room.Index

@Entity(
    tableName = "locations",
    indices = [
        Index(value = ["latitude", "longitude"], unique = true)
    ]
)
data class LocationEntity(
    @PrimaryKey(autoGenerate = true)
    val id: Long = 0,
    val name: String,
    val latitude: Double,
    val longitude: Double,
    val weatherData: String? = null,
    val lastUpdated: Long = 0,
    val notificationsEnabled: Boolean = false,
    val notificationTime: String = "08:00",
    val changeAlertsEnabled: Boolean = false,
    val changeAlertInterval: String = "3",
    val selected: Boolean = false,
    val isDefault: Boolean = false
) {
    val currentTemp: Double?
        get() {
            return try {
                weatherData?.let { data ->
                    val obj = org.json.JSONObject(data)
                    when {
                        obj.has("current_weather") -> obj.getJSONObject("current_weather").getDouble("temperature")
                        obj.has("current") -> obj.getJSONObject("current").getDouble("temp_c")
                        obj.has("timelines") -> {
                            val timelines = obj.getJSONObject("timelines")
                            if (timelines.has("minutely")) timelines.getJSONArray("minutely").getJSONObject(0).getJSONObject("values").getDouble("temperature")
                            else timelines.getJSONArray("daily").getJSONObject(0).getJSONObject("values").getDouble("temperatureAvg")
                        }
                        obj.has("currentConditions") -> obj.getJSONObject("currentConditions").getDouble("temp")
                        obj.has("temperature_2m") -> obj.getDouble("temperature_2m")
                        else -> null
                    }
                }
            } catch (e: Exception) {
                null
            }
        }
    
    val currentWeatherCode: Int?
        get() {
            return try {
                weatherData?.let { data ->
                    val obj = org.json.JSONObject(data)
                    when {
                        obj.has("current_weather") -> obj.getJSONObject("current_weather").getInt("weathercode")
                        obj.has("current") -> {
                            val current = obj.getJSONObject("current")
                            if (current.has("condition")) current.getJSONObject("condition").getInt("code")
                            else current.optInt("weathercode", 0)
                        }
                        obj.has("timelines") -> obj.getJSONObject("timelines").getJSONArray("daily").getJSONObject(0).getJSONObject("values").getInt("weatherCodeMax")
                        obj.has("currentConditions") -> 0 // Visual crossing needs mapping
                        obj.has("weathercode") -> obj.getInt("weathercode")
                        else -> null
                    }
                }
            } catch (e: Exception) {
                null
            }
        }

    val isDay: Boolean
        get() {
            return try {
                weatherData?.let { data ->
                    val obj = org.json.JSONObject(data)
                    if (obj.has("current")) {
                        obj.getJSONObject("current").optInt("is_day", 1) == 1
                    } else if (obj.has("current_weather")) {
                        obj.getJSONObject("current_weather").optInt("is_day", 1) == 1
                    } else true
                } ?: true
            } catch (e: Exception) {
                true
            }
        }

    val provider: String
        get() {
            val data = weatherData ?: return "unknown"
            return when {
                data.contains("\"current_weather\":") -> "open_meteo"
                data.contains("\"current\":") -> "weatherapi"
                data.contains("\"timelines\":") -> "tomorrow.io"
                data.contains("\"currentConditions\":") -> "visualcrossing"
                else -> "open_meteo"
            }
        }

}