incidence_at.js

// Variables used by Scriptable.
// These must be at the very top of the file. Do not edit.
// icon-color: light-gray; icon-glyph: microscope;
// Lizenz: AGES Dashboard COVID19 AT
// Use 3-digit GKZ (third row of https://covid19-dashboard.ages.at/data/CovidFaelle_Timeline_GKZ.csv).
// Widget Parameter: "204,KFL;312;706" for Klagenfurt Land (shown as KFL), Korneuburg and Landeck
//
// Based on the german variant:
// - kevinkub https://gist.github.com/kevinkub/46caebfebc7e26be63403a7f0587f664
// - Baumchen https://gist.github.com/Baumchen/6d91df0a4c76c45b15576db0632e4329
//
// No guarantee on correctness and completeness of the information provided.
// Version 0.3.0

// user configuration
const useLogarithmicScale = true
const maxDaysDisplayed = {
  small: 3,
  medium: 7,
  large: 7,
}
// user configuration end

const urlTimelineAustria = "https://docs.google.com/spreadsheets/d/e/2PACX-1vQldP8uIuZsfCQWIBUkgybPnIhTEbXlXk3xF2qW1hEOH3Qrh9F8340mXJj6SCGS_iTcUolVmwhwjphx/pub?gid=344127261&single=true&output=csv"
const urlTimelineGkz = "https://docs.google.com/spreadsheets/d/e/2PACX-1vQldP8uIuZsfCQWIBUkgybPnIhTEbXlXk3xF2qW1hEOH3Qrh9F8340mXJj6SCGS_iTcUolVmwhwjphx/pub?gid=712549080&single=true&output=csv"
const urlTimeline = "https://docs.google.com/spreadsheets/d/e/2PACX-1vQldP8uIuZsfCQWIBUkgybPnIhTEbXlXk3xF2qW1hEOH3Qrh9F8340mXJj6SCGS_iTcUolVmwhwjphx/pub?gid=0&single=true&output=csv"
const urlRSeries = "https://docs.google.com/spreadsheets/d/e/2PACX-1vQldP8uIuZsfCQWIBUkgybPnIhTEbXlXk3xF2qW1hEOH3Qrh9F8340mXJj6SCGS_iTcUolVmwhwjphx/pub?gid=776589868&single=true&output=csv"
const urlVaccinations = "https://info.gesundheitsministerium.gv.at/data/COVID19_vaccination_certificates.csv"

const jsonBKZData = "https://gist.githubusercontent.com/pasrom/8b1c9d5def267f5abee45073a153b36a/raw/districts.json"

const widgetSizes = {
  small: { width: 465, height: 465 },
  medium: { width: 987, height: 465 },
  large: { width: 987, height: 1035 },
}

let mediumWidget = (config.widgetFamily === 'medium') ? true : false

class LineChart {
  // LineChart by https://kevinkub.de/
  constructor(width, height, seriesA, seriesB) {
    this.ctx = new DrawContext();
    this.ctx.size = new Size(width, height);
    this.seriesA = seriesA;
    this.seriesB = seriesB;
  }
  _calculatePath(series, fillPath) {
    let maxValue = Math.max(...series);
    let minValue = Math.min(...series);
    let difference = maxValue - minValue;
    let count = series.length;
    let step = this.ctx.size.width / (count - 1);
    let points = series.map((current, index, all) => {
      let x = step * index;
      let y = this.ctx.size.height - (current - minValue) / difference * this.ctx.size.height;
      return new Point(x, y);
    });
    return this._getSmoothPath(points, fillPath);
  }
  _getSmoothPath(points, fillPath) {
    let path = new Path();
    path.move(new Point(0, this.ctx.size.height));
    path.addLine(points[0]);
    for (let i = 0; i < points.length - 1; i++) {
      let xAvg = (points[i].x + points[i + 1].x) / 2;
      let yAvg = (points[i].y + points[i + 1].y) / 2;
      let avg = new Point(xAvg, yAvg);
      let cp1 = new Point((xAvg + points[i].x) / 2, points[i].y);
      let next = new Point(points[i + 1].x, points[i + 1].y);
      let cp2 = new Point((xAvg + points[i + 1].x) / 2, points[i + 1].y);
      path.addQuadCurve(avg, cp1);
      path.addQuadCurve(next, cp2);
    }
    if (fillPath) {
      path.addLine(new Point(this.ctx.size.width, this.ctx.size.height));
      path.closeSubpath();
    }
    return path;
  }
  configure(fn) {
    let pathA = this._calculatePath(this.seriesA, true);
    let pathB = this._calculatePath(this.seriesB, false);
    if (fn) {
      fn(this.ctx, pathA, pathB);
    } else {
      this.ctx.addPath(pathA);
      this.ctx.fillPath(pathA);
      this.ctx.addPath(pathB);
      this.ctx.fillPath(pathB);
    }
    return this.ctx;
  }
}

class DataResponse {
    constructor(data, status = 200) {
        this.data = data
        this.status = status
    }
}

const fileType = {
  json: ".json",
  csv: ".csv",
};

function parseLocation(location) {
  const components = location.split(",")
  return {
    gkz: components[0],
    name: components.length >= 2 ? components[1] : null,
  }
}

function calc(data, location, nr = 0) {
  ctr = 0
  for (line of data) {
    var components = null
    if(line.search(";") >= 0) {
      components = line.split(";")
    } else {
      components = line.split(/,(?=(?:(?:[^"]*"){2})*[^"]*$)/)
    }
    if (components[2] === location["gkz"]) {
      if (nr === ctr) {
        var day = +components[0].substring(0,2)
        var month = +components[0].substring(3,5)
        var year = +components[0].substring(6,11)
        return {
          date: (new Date(Date.UTC(year, month - 1, day, 0, 0, 0, 0))),
          state_district: location["name"] ? location["name"] : components[1],
          id: parseInt(components[2]),
          residents: parseInt(components[3]),
          cases_daily: parseInt(components[4]),
          cases_sum: parseInt(components[5]),
          cases_7_days: parseInt(components[6]),
          incidence_7_days: parseInt(components[7].split('\"')[1]),
          deaths_daily: parseInt(components[8]),
          deaths_sum: parseInt(components[9]),
          cured_daily: parseInt(components[10]),
          cured_sum: parseInt(components[11]),
          active_cases_sum: parseInt(components[5]) - parseInt(components[11]) - parseInt(components[9]),
          hospitalization: parseInt(components[12]),
          icu: parseInt(components[13]),
        }
      }
      ctr++
    }
  }
  return {
    error: "GKZ unbekannt.",
  }
}

function calcR(data, nr = 0) {
    var components = null
    if(data[nr].search(";") >= 0) {
      components = data[nr].split(";")
    } else {
      components = data[nr].split(/,(?=(?:(?:[^"]*"){2})*[^"]*$)/)
    }
    var day = +components[0].substring(8,10)
    var month = +components[0].substring(5,7)
    var year = +components[0].substring(0,4)
    return {
      date: (new Date(Date.UTC(year, month - 1, day, 0, 0, 0, 0))),
      r_value: parseFloat(components[1].replace(',', '.').replaceAll('"', '')),
    }
}

function getVaccinations(data, location, nr = 0) {
  ctr = 0
  for (line of data) {
    var components = null
    if(line.search(";") >= 0) {
      components = line.split(";")
    } else {
      components = line.split(",")
    }
    if (components[1] === location["gkz"] && components[3] === "All") {
      if (nr === ctr) {
        var day = +components[0].substring(8,10)
        var month = +components[0].substring(5,7)
        var year = +components[0].substring(0,4)
        return {
          date: (new Date(Date.UTC(year, month - 1, day, 0, 0, 0, 0))),
          state_id: components[1],
          state_name: location["name"] ? location["name"] : components[2],
          age_group: parseInt(components[3]),
          gender: parseInt(components[4]),
          population: parseFloat(components[5]),
          valid_certificates: parseInt(components[6]),
          valid_certificates_percent: parseFloat(components[7]),
        }
      }
      ctr++
    }
  }
  return {
    error: "GKZ unbekannt.",
  }
}

function getTimeline(data, location, nr) {
  var timeline = []
  for (line of data) {
    line = line.replaceAll('\r', '')
    const components = line.split(",")
    if (components[2] === location["gkz"]) {
      timeline.push(parseFloat(components[nr]))
    }
  }
  return timeline
}

var fm = getFilemanager()
let fmConfigDirectory = fm.joinPath(fm.documentsDirectory(), '/coronaWidget')
if (!fm.isDirectory(fmConfigDirectory)) fm.createDirectory(fmConfigDirectory)

const widgetSize = widgetSizes[config.widgetFamily] ?? widgetSizes.small
const daysDisplayed = maxDaysDisplayed[config.widgetFamily] ?? maxDaysDisplayed.small

let widget = await createWidget(widgetSize, daysDisplayed)
if (!config.runsInWidget) {
  if (mediumWidget) {
      await widget.presentMedium()
  } else {
      await widget.presentSmall()
  }
}

Script.setWidget(widget)
Script.complete()

async function getLocation() {
  try {
    Location.setAccuracyToThreeKilometers()
    return await Location.current()
  } catch (e) {
    return null;
  }
}

async function getBkzNumber(url, location) {
  const act_location = await getLocation();
  const BKZData = await new Request(url).loadJSON()
  const tmp = await Location.reverseGeocode(act_location.latitude, act_location.longitude)
  const district = tmp[0].subAdministrativeArea
  for (var i = 0; i < BKZData.length; i++) {
    if (BKZData[i].Bezirk === district) {
      return BKZData[i].BKZ + "," + "📍" + BKZData[i].KFZ
    }
  }
  throw "No district <" + district + "> could be matched with the current position!"
}

async function getCsvData(url, fileName) {
  var error = 0
  var request = null
  var request_str = null
  try{
    request = new Request(url);
    request_str = await request.loadString()
  } catch (e) {
    error = -1
    log(e)
  }
  if (error >= 0 && request.response.statusCode != 200) {
    error = -1
    log("error " + request.response.statusCode + " getting " + url)
  }
  var re = /text\/csv|text\/plain/i;
  if (error >= 0 && !request.response.mimeType.match(re)) {
    error = -1
    log("Wrong mimeType " + request.response.mimeType + " from url " + url)
  }
  if (error >= 0){
    saveData(fileType.csv, fileName, request_str)
  } else {
    log("using offline data: " + fileName)
    let tmp = await loadData(fileType.csv, fileName)
    request_str = tmp.data
  }

  return request_str.split(/\r\n|\n/).reverse()
}

async function createWidget(widgetSize, daysDisplayed) {
  const list = new ListWidget()
  let nextRefresh = Date.now() + 1000*30*60 // 30 min refresch rate
  list.refreshAfterDate = new Date(nextRefresh)
  list.setPadding(0, 0, 0, 0)
  list.addSpacer(5)

  if (args.widgetParameter) {
    parameter = args.widgetParameter
  } else {
    parameter = "802,B;8,Vbg."
  }
  /* remove last ';' if one is present */
  parameter = parameter.replace(/;\s*$/, "");

  try {
    BKZNr = await getBkzNumber(jsonBKZData) + ";"
  } catch (e) {
    BKZNr = ""
    log(e)
  }
  parameter = BKZNr  + parameter

  const locations = parameter.split(";").map(parseLocation)
  const states = "10,🇦🇹".split(";").map(parseLocation)

  const timeline_gkz_lines = await getCsvData(urlTimelineGkz, "Timeline_GKZ")
  const timeline_lines = await getCsvData(urlTimeline, "Timeline")
  const timeline_austria_lines = await getCsvData(urlTimelineAustria, "Timeline_Austria")
  const r_series = await getCsvData(urlRSeries, "r-series")
  const vaccinations = await getCsvData(urlVaccinations, "Vaccinations")

  vaccinations_data = getVaccinations(vaccinations, states[0])

  const vaccinatons_stack = list.addStack()
  vaccinatons_stack.layoutHorizontally()
  vaccinatons_stack.setPadding(0, 0, 0, 0)
  vaccinatons_stack.addSpacer()
  vaccinations_label = vaccinatons_stack.addText("🇦🇹 " + vaccinations_data["valid_certificates"] + " | " + vaccinations_data["valid_certificates_percent"].toFixed(2) + " 💉")
  vaccinations_label.font = Font.mediumSystemFont(10)
  vaccinatons_stack.addSpacer()
  list.addSpacer(5)

  var data_timeline = []
  for (var i = 0; i < daysDisplayed + 1; i++) {
    data_timeline.push(calc(timeline_lines, states[0], i))
  }

  const infected_stack = list.addStack()
  infected_stack.layoutHorizontally()
  infected_stack.setPadding(0, 0, 0, 0)
  infected_stack.addSpacer()

  for (var i = daysDisplayed-1; i >= 0 ; i--) {
    const text_cases = data_timeline[i]["cases_daily"]
    const date_cases = ('0' + data_timeline[i]["date"].getDate()).slice(-2) + '.'
             + ('0' + (data_timeline[i]["date"].getMonth() + 1)).slice(-2) + '.'
             + data_timeline[i]["date"].getFullYear();
    const text_date = `${data_timeline[i]["date"].getDate()}/${data_timeline[i]["date"].getMonth() + 1}`
    var text_r = "N/A"
    for (var j = 0; j < r_series.length; j++) {
      r_value = calcR(r_series, j)
      date_cases_r = ('0' + r_value["date"].getDate()).slice(-2) + '.'
        + ('0' + (r_value["date"].getMonth() + 1)).slice(-2) + '.'
        + r_value["date"].getFullYear();
      if (date_cases === date_cases_r) {
        text_r = String(r_value["r_value"].toFixed(2))
        break
      }
    }
    text_deaths = data_timeline[i]["deaths_daily"]
    const date_infected = infected_stack.addText(
      text_date + "\n" +
      text_cases + "\n" +
      text_deaths + "\n" +
      text_r
    )
    date_infected.font = Font.mediumSystemFont(10)
    date_infected.centerAlignText()
    infected_stack.addSpacer()
  }
  list.addSpacer(5)

  const icu_stack = list.addStack()
  icu_stack.layoutHorizontally()
  icu_stack.setPadding(0, 0, 0, 0)
  icu_stack.addSpacer()
  vaccinations_label = icu_stack.addText("🏥 " + data_timeline[0]["hospitalization"] + " | " + data_timeline[0]["icu"] + " | " + data_timeline[0]["deaths_sum"])
  vaccinations_label.font = Font.mediumSystemFont(10)
  icu_stack.addSpacer()
  list.addSpacer(5)

  const stack_values = list.addStack()
  stack_values.setPadding(0, 0, 0, 0)
  stack_values.layoutHorizontally()
  stack_values.addSpacer()
  
  // print incidence and active cases
  printCoronaValues(stack_values, data_timeline[0], data_timeline[1])

  ctr = 0
  for (location of locations) {
    if (locations[0]["gkz"] === location["gkz"] && ctr != 0) {
      continue
    }
    if (location["gkz"] > 10) {
      var data_time_line = timeline_gkz_lines
    } else {
      var data_time_line = timeline_lines
    }
    const data = calc(data_time_line, location)
    const data_yesterday = calc(data_time_line, location, 1)
    printCoronaValues(stack_values, data, data_yesterday)
    ctr++
  }

  let data = getTimeline(timeline_austria_lines, states[0], 4).reverse()
  let sum_cases = getTimeline(timeline_austria_lines, states[0], 6).reverse()
  let sum_cured = getTimeline(timeline_austria_lines, states[0], 12).reverse()
  let infected = sum_cases.filter(x => !sum_cured.includes(x));
  
  if (useLogarithmicScale) {
    data = data.map(Math.log)
    infected = infected.map(Math.log)
  }

  let chart = new LineChart(widgetSize.width, widgetSize.height, data, infected).configure((ctx, pathA, pathB) => {
    ctx.opaque = false;
    ctx.setFillColor(new Color("888888", .5));
    ctx.addPath(pathA);
    ctx.fillPath(pathA);
    ctx.addPath(pathB);
    ctx.setStrokeColor(Color.white());
    ctx.setLineWidth(1)
    ctx.strokePath();
  }).getImage();
  list.backgroundImage = chart

  return list
}

function getColor(value) {
  if (value >= 50) {
    return Color.red()
  } else if (value >= 35) {
    return Color.orange()
  } else {
    return Color.green()
  }
}

function printCoronaValues(stack, data, data_yesterday) {
  const corona_value = [{
    color: null,
    text: data["state_district"],
  },{
    color: getColor(data["incidence_7_days"]),
    text: String(data["incidence_7_days"]) + getTrendArrow(data_yesterday["incidence_7_days"], data["incidence_7_days"]),
  },{
    color: getColor(data["incidence_7_days"]),
    text: String(data["active_cases_sum"]) + getTrendArrow(data_yesterday["active_cases_sum"], data["active_cases_sum"]),
  }]
  const line = stack.addStack()
  line.layoutVertically()
  stack.addSpacer()
  corona_value.forEach(function (value) {
    const text_line = line.addStack()
    line.layoutVertically()
    const label = text_line.addText(value.text)
    if(value.color != null) {
      label.font = Font.boldSystemFont(10)
      label.textColor = value.color;
    } else {
      label.font = Font.systemFont(10)
    }
    label.lineLimit = 1
    label.minimumScaleFactor = 0.3
  });
}

function getTrendArrow(preValue, currentValue) {
  return (currentValue <= preValue) ? '↓' : '↑'
}

function saveData(fType, dataId, newData) {
  var path = fm.joinPath(fmConfigDirectory, 'corona-widget-at-' + dataId + fType)
  switch (fType) {
    case fileType.json:
      fm.writeString(path, JSON.stringify(newData))
      break;
    case fileType.csv:
    default:
      fm.writeString(path, newData)
      break;
  }
}

async function loadData(fType, dataId) {
    let path = fm.joinPath(fmConfigDirectory, 'corona-widget-at-' + dataId + fType)
    if (fm.isFileStoredIniCloud(path) && !fm.isFileDownloaded(path)) {
        await fm.downloadFileFromiCloud(path)
    }
    if (fm.fileExists(path)) {
        try {
            let string = fm.readString(path)
            switch (fType) {
              case fileType.json:
                return new DataResponse(JSON.parse(string))
              case fileType.csv:
              default:
                data = string
                return new DataResponse(string)
            }
         } catch (e) {
            log(e)
            return new DataResponse(null, 500)
         }
    }
    return new DataResponse(null, 404)
}


function getFilemanager() {
    try {
        fm = FileManager.iCloud()
    } catch (e) {
        fm = FileManager.local()
    }
    // check if user logged in iCloud
    try { 
        fm.documentsDirectory()
    } catch(e) {
        fm = FileManager.local()
    }
    return fm
}