""".. title:: CF6 Data as GeoJSON

Return to `API Services </api/#json>`_

Changelog
---------

- 2024-08-15: Initial documentation release

Example Usage
-------------

Request CF6 data for January 1st, 2024:

https://mesonet.agron.iastate.edu/geojson/cf6.py?dt=2024-01-01

Same request, but JSON(P)

https://mesonet.agron.iastate.edu/geojson/cf6.py?dt=2024-01-01&callback=foo

Same request, but return as a comma delimited file

https://mesonet.agron.iastate.edu/geojson/cf6.py?dt=2024-01-01&fmt=csv

"""

from datetime import date
from typing import Annotated

import simplejson as json
from pydantic import Field
from pyiem.database import get_sqlalchemy_conn, sql_helper
from pyiem.reference import TRACE_VALUE
from pyiem.webutil import CGIModel, iemapp
from simplejson import encoder

from iemweb.fields import CALLBACK_FIELD
from iemweb.util import get_ct, json_response_dict

encoder.FLOAT_REPR = lambda o: format(o, ".2f")


class Schema(CGIModel):
    """See how we are called."""

    dt: Annotated[
        date,
        Field(default_factory=date.today, description="Date to query for"),
    ]
    callback: CALLBACK_FIELD = None
    fmt: Annotated[
        str,
        Field(
            description="Format of output",
            pattern=r"^(geojson|csv)$",
        ),
    ] = "geojson"


def departure(ob, climo):
    """Compute a departure value"""
    if ob is None or climo is None:
        return "M"
    return ob - climo


def int_sanitize(val):
    """convert to Ms"""
    if val is None:
        return "M"
    if val == TRACE_VALUE:
        return "T"
    return int(val)


def f1_sanitize(val):
    """convert to Ms"""
    if val is None:
        return "M"
    if val == TRACE_VALUE:
        return "T"
    return round(val, 1)


def f2_sanitize(val):
    """convert to Ms"""
    if val is None:
        return "M"
    if val == TRACE_VALUE:
        return "T"
    return round(val, 2)


def get_data(conn, ts, fmt):
    """Get the data for this timestamp"""
    data = json_response_dict(
        {
            "type": "FeatureCollection",
            "features": [],
        }
    )
    # Fetch the daily values
    res = conn.execute(
        sql_helper("""
    select station, name, product, state, wfo, valid,
    round(st_x(geom)::numeric, 4)::float as st_x,
    round(st_y(geom)::numeric, 4)::float as st_y,
    high, low, avg_temp, dep_temp, hdd, cdd, precip, snow, snowd_12z,
    avg_smph, max_smph, avg_drct, minutes_sunshine, possible_sunshine,
    cloud_ss, wxcodes, gust_smph, gust_drct
    from cf6_data c JOIN stations s on (c.station = s.id)
    WHERE s.network = 'NWSCLI' and c.valid = :ts
    """),
        {"ts": ts},
    )
    for i, row in enumerate(res.mappings()):
        data["features"].append(
            {
                "type": "Feature",
                "id": i,
                "properties": {
                    "station": row["station"],
                    "state": row["state"],
                    "valid": row["valid"].strftime("%Y-%m-%d"),
                    "wfo": row["wfo"],
                    "link": f"/api/1/nwstext/{row['product']}",
                    "product": row["product"],
                    "name": row["name"],
                    "high": int_sanitize(row["high"]),
                    "low": int_sanitize(row["low"]),
                    "avg_temp": f1_sanitize(row["avg_temp"]),
                    "dep_temp": f1_sanitize(row["dep_temp"]),
                    "hdd": int_sanitize(row["hdd"]),
                    "cdd": int_sanitize(row["cdd"]),
                    "precip": f2_sanitize(row["precip"]),
                    "snow": f1_sanitize(row["snow"]),
                    "snowd_12z": f1_sanitize(row["snowd_12z"]),
                    "avg_smph": f1_sanitize(row["avg_smph"]),
                    "max_smph": f1_sanitize(row["max_smph"]),
                    "avg_drct": int_sanitize(row["avg_drct"]),
                    "minutes_sunshine": int_sanitize(row["minutes_sunshine"]),
                    "possible_sunshine": int_sanitize(
                        row["possible_sunshine"]
                    ),
                    "cloud_ss": f1_sanitize(row["cloud_ss"]),
                    "wxcodes": row["wxcodes"],
                    "gust_smph": f1_sanitize(row["gust_smph"]),
                    "gust_drct": int_sanitize(row["gust_drct"]),
                },
                "geometry": {
                    "type": "Point",
                    "coordinates": [row["st_x"], row["st_y"]],
                },
            }
        )
    if fmt == "geojson":
        return json.dumps(data, ensure_ascii=False)
    cols = (
        "station,valid,name,state,wfo,high,low,avg_temp,dep_temp,hdd,cdd,"
        "precip,snow,snowd_12z,avg_smph,max_smph,avg_drct,minutes_sunshine,"
        "possible_sunshine,cloud_ss,wxcodes,gust_smph,gust_drct"
    )
    res = cols + "\n"
    for feat in data["features"]:
        for col in cols.split(","):
            val = feat["properties"][col]
            if isinstance(val, list | tuple):
                res += "%s," % (" ".join([str(s) for s in val]),)
            else:
                res += "%s," % (val,)
        res += "\n"
    return res


def get_mckey(environ: dict) -> str:
    """Figure out our memcache key."""
    return f"/geojson/cf6/{environ['dt']:%Y%m%d}?fmt={environ['fmt']}"


@iemapp(
    help=__doc__,
    schema=Schema,
    memcachekey=get_mckey,
    memcacheexpire=300,
    content_type=get_ct,
)
def application(environ, start_response):
    """see how we are called"""
    dt = environ["dt"]
    fmt = environ["fmt"]
    headers = [("Content-type", get_ct(environ))]
    with get_sqlalchemy_conn("iem") as conn:
        data = get_data(conn, dt, fmt).encode("utf-8")
    start_response("200 OK", headers)
    return data
