import {InfluxDB} from '@influxdata/influxdb-client';
import {cfg} from './config.mjs';

const MEASUREMENTS = ['accumulation', 'environment', 'light', 'lightning', 'rain', 'seismic', 'wind'];

let client, currentUrl, currentToken;

function getClient() {
	const c = cfg();
	if(c.INFLUX_URL !== currentUrl || c.INFLUX_TOKEN !== currentToken) {
		client = new InfluxDB({url: c.INFLUX_URL, token: c.INFLUX_TOKEN});
		currentUrl = c.INFLUX_URL;
		currentToken = c.INFLUX_TOKEN;
	}
	return {client, c};
}

async function query(flux) {
	const {client, c} = getClient();
	const api = client.getQueryApi(c.INFLUX_ORG);
	const rows = [];
	return new Promise((resolve, reject) => {
		api.queryRows(flux, {
			next(row, meta) { rows.push(meta.toObject(row)); },
			error(err) { reject(err); },
			complete() { resolve(rows); },
		});
	});
}

function truncateToHour(iso) {
	const d = new Date(iso);
	d.setMinutes(0, 0, 0);
	const pad = n => String(n).padStart(2, '0');
	return `${d.getFullYear()}-${pad(d.getMonth() + 1)}-${pad(d.getDate())}T${pad(d.getHours())}:00`;
}

function truncateToDay(iso) {
	return new Date(iso).toISOString().slice(0, 10);
}

function extractRow(row) {
	const out = {};
	for(const [k, v] of Object.entries(row)) {
		if(!k.startsWith('_') && k !== 'result' && k !== 'table') out[k] = v;
	}
	return out;
}

export async function queryCurrent() {
	const {c} = getClient();
	const flux = `
    from(bucket: "${c.INFLUX_BUCKET}")
      |> range(start: -1h)
      |> filter(fn: (r) => ${MEASUREMENTS.map(m => `r._measurement == "${m}"`).join(' or ')})
      |> last()
      |> pivot(rowKey: ["_time"], columnKey: ["_field"], valueColumn: "_value")
  `;
	const rows = await query(flux);
	const result = {time: new Date()};
	for(const row of rows) {
		Object.assign(result, extractRow(row));
	}
	return result;
}

export async function queryHourly(start, end) {
	const {c} = getClient();
	const flux = `
    from(bucket: "${c.INFLUX_BUCKET}")
      |> range(start: ${start.toISOString()}, stop: ${end.toISOString()})
      |> filter(fn: (r) => ${MEASUREMENTS.map(m => `r._measurement == "${m}"`).join(' or ')})
      |> group(columns: ["_field"])
      |> aggregateWindow(every: 1h, fn: mean, createEmpty: false)
      |> pivot(rowKey: ["_time"], columnKey: ["_field"], valueColumn: "_value")
  `;
	const rows = await query(flux);
	return rows.map(row => ({
		time: truncateToHour(row._time),
		...extractRow(row),
	}));
}

export async function queryDaily(start, end) {
	const {c} = getClient();
	const measurements = ['accumulation', 'environment', 'light', 'lightning', 'seismic'];
	const results = {};

	for(const m of measurements) {
		const [means, mins, maxs] = await Promise.all([
			query(`
        from(bucket: "${c.INFLUX_BUCKET}")
          |> range(start: ${start.toISOString()}, stop: ${end.toISOString()})
          |> filter(fn: (r) => r._measurement == "${m}")
          |> group(columns: ["_field"])
          |> aggregateWindow(every: 1d, fn: mean, createEmpty: false)
          |> pivot(rowKey: ["_time"], columnKey: ["_field"], valueColumn: "_value")
      `),
			query(`
        from(bucket: "${c.INFLUX_BUCKET}")
          |> range(start: ${start.toISOString()}, stop: ${end.toISOString()})
          |> filter(fn: (r) => r._measurement == "${m}")
          |> group(columns: ["_field"])
          |> aggregateWindow(every: 1d, fn: min, createEmpty: false)
          |> pivot(rowKey: ["_time"], columnKey: ["_field"], valueColumn: "_value")
      `),
			query(`
        from(bucket: "${c.INFLUX_BUCKET}")
          |> range(start: ${start.toISOString()}, stop: ${end.toISOString()})
          |> filter(fn: (r) => r._measurement == "${m}")
          |> group(columns: ["_field"])
          |> aggregateWindow(every: 1d, fn: max, createEmpty: false)
          |> pivot(rowKey: ["_time"], columnKey: ["_field"], valueColumn: "_value")
      `),
		]);

		for(const row of means) {
			const time = truncateToDay(row._time);
			if(!results[time]) results[time] = {time};
			Object.assign(results[time], extractRow(row));
		}

		for(const row of mins) {
			const time = truncateToDay(row._time);
			if(!results[time]) results[time] = {time};
			if(row.temperature != null) results[time].env_temp_min_c = row.temperature;
		}
		for(const row of maxs) {
			const time = truncateToDay(row._time);
			if(!results[time]) results[time] = {time};
			if(row.temperature != null) results[time].env_temp_max_c = row.temperature;
		}
	}

	return Object.values(results).sort((a, b) => a.time.localeCompare(b.time));
}

export async function getCoords() {
	const {c} = getClient();
	const flux = `
    from(bucket: "${c.INFLUX_BUCKET}")
      |> range(start: -24h)
      |> filter(fn: (r) => r._measurement == "gps")
      |> last()
      |> pivot(rowKey: ["_time"], columnKey: ["_field"], valueColumn: "_value")
  `;
	const rows = await query(flux);
	if(rows.length && rows[0].latitude && rows[0].longitude) {
		return {latitude: rows[0].latitude, longitude: rows[0].longitude, altitude: rows[0].altitude || c.ALTITUDE};
	}
	return {latitude: c.LATITUDE, longitude: c.LONGITUDE, altitude: c.ALTITUDE};
}