package database

import (
	"database/sql"
	"fmt"
)

func InsertOrUpdateEstimatedVehicleJourney(db *sql.DB, values []interface{}) (int, string, error) {
	query := `
	INSERT INTO estimatedvehiclejourney (servicedelivery, recordedattime, lineref, directionref, datasource, datedvehiclejourneyref, vehiclemode, dataframeref, originref, destinationref, operatorref, vehicleref, cancellation, other, firstservicedelivery)
	VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $1)
	ON CONFLICT (lineref, directionref, datasource, datedvehiclejourneyref)
	DO UPDATE SET
		servicedelivery = EXCLUDED.servicedelivery,
		recordedattime = EXCLUDED.recordedattime,
		vehiclemode = COALESCE(EXCLUDED.vehiclemode, estimatedvehiclejourney.vehiclemode),
		dataframeref = COALESCE(EXCLUDED.dataframeref, estimatedvehiclejourney.dataframeref),
		originref = COALESCE(EXCLUDED.originref, estimatedvehiclejourney.originref),
		destinationref = COALESCE(EXCLUDED.destinationref, estimatedvehiclejourney.destinationref),
		operatorref = COALESCE(EXCLUDED.operatorref, estimatedvehiclejourney.operatorref),
		vehicleref = COALESCE(EXCLUDED.vehicleref, estimatedvehiclejourney.vehicleref),
		cancellation = COALESCE(EXCLUDED.cancellation, estimatedvehiclejourney.cancellation),
		other = COALESCE(EXCLUDED.other, estimatedvehiclejourney.other)
	RETURNING CASE WHEN xmax = 0 THEN 'insert' ELSE 'update' END, id;
	`

	stmt, err := db.Prepare(query)
	if err != nil {
		return 0, "", fmt.Errorf("error preparing statement: %v", err)
	}
	defer stmt.Close()

	var action string
	var id int
	err = stmt.QueryRow(values...).Scan(&action, &id)
	if err != nil {
		return 0, "", fmt.Errorf("error executing statement: %v", err)
	}

	return id, action, nil
}