223 lines
7.9 KiB
Go
223 lines
7.9 KiB
Go
package main
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
)
|
|
|
|
func ensureSchema(db *sql.DB) error {
|
|
stmts := []string{
|
|
`CREATE TABLE IF NOT EXISTS settings(
|
|
id TINYINT PRIMARY KEY,
|
|
days_per_month INT NOT NULL DEFAULT 2,
|
|
current_month TINYINT NOT NULL DEFAULT 1,
|
|
current_day TINYINT NOT NULL DEFAULT 1
|
|
)`,
|
|
`ALTER TABLE settings ADD COLUMN IF NOT EXISTS current_month TINYINT NOT NULL DEFAULT 1`,
|
|
`ALTER TABLE settings ADD COLUMN IF NOT EXISTS current_day TINYINT NOT NULL DEFAULT 1`,
|
|
`INSERT INTO settings(id,days_per_month,current_month,current_day) VALUES (1,2,1,1) ON DUPLICATE KEY UPDATE id=id`,
|
|
`UPDATE settings SET current_month=1 WHERE current_month < 1 OR current_month > 12`,
|
|
`UPDATE settings SET current_day=1 WHERE current_day < 1`,
|
|
|
|
`CREATE TABLE IF NOT EXISTS fields(
|
|
id BIGINT AUTO_INCREMENT PRIMARY KEY,
|
|
number INT NOT NULL UNIQUE,
|
|
name VARCHAR(120) NOT NULL DEFAULT '',
|
|
owned TINYINT(1) NOT NULL DEFAULT 0,
|
|
group_key VARCHAR(64) NOT NULL DEFAULT '',
|
|
group_name VARCHAR(120) NOT NULL DEFAULT '',
|
|
created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP
|
|
)`,
|
|
`ALTER TABLE fields ADD COLUMN IF NOT EXISTS owned TINYINT(1) NOT NULL DEFAULT 0`,
|
|
`ALTER TABLE fields ADD COLUMN IF NOT EXISTS group_key VARCHAR(64) NOT NULL DEFAULT ''`,
|
|
`ALTER TABLE fields ADD COLUMN IF NOT EXISTS group_name VARCHAR(120) NOT NULL DEFAULT ''`,
|
|
|
|
`CREATE TABLE IF NOT EXISTS crops(
|
|
id BIGINT AUTO_INCREMENT PRIMARY KEY,
|
|
name VARCHAR(80) NOT NULL UNIQUE,
|
|
sow_start_month TINYINT NOT NULL,
|
|
sow_end_month TINYINT NOT NULL,
|
|
grow_months TINYINT NOT NULL,
|
|
regrow_enabled TINYINT(1) NOT NULL DEFAULT 0,
|
|
regrow_cycles INT NOT NULL DEFAULT 0
|
|
)`,
|
|
`ALTER TABLE crops ADD COLUMN IF NOT EXISTS regrow_enabled TINYINT(1) NOT NULL DEFAULT 0`,
|
|
`ALTER TABLE crops ADD COLUMN IF NOT EXISTS regrow_cycles INT NOT NULL DEFAULT 0`,
|
|
`UPDATE crops SET regrow_cycles=0 WHERE regrow_cycles < 0`,
|
|
`CREATE TABLE IF NOT EXISTS crop_steps(
|
|
id BIGINT AUTO_INCREMENT PRIMARY KEY,
|
|
crop_id BIGINT NOT NULL,
|
|
phase ENUM('pre','post') NOT NULL,
|
|
month_offset INT NOT NULL DEFAULT 0,
|
|
title VARCHAR(140) NOT NULL,
|
|
created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
|
CONSTRAINT fk_crop_steps_crop FOREIGN KEY(crop_id) REFERENCES crops(id) ON DELETE CASCADE
|
|
)`,
|
|
|
|
`CREATE TABLE IF NOT EXISTS plans(
|
|
id BIGINT AUTO_INCREMENT PRIMARY KEY,
|
|
field_id BIGINT NULL,
|
|
target_ref VARCHAR(80) NOT NULL DEFAULT '',
|
|
target_name VARCHAR(140) NOT NULL DEFAULT '',
|
|
crop_id BIGINT NOT NULL,
|
|
start_month TINYINT NOT NULL,
|
|
start_day TINYINT NOT NULL,
|
|
harvest_month TINYINT NOT NULL,
|
|
harvest_day TINYINT NOT NULL,
|
|
notes VARCHAR(255) NOT NULL DEFAULT '',
|
|
created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
|
CONSTRAINT fk_plans_fields FOREIGN KEY(field_id) REFERENCES fields(id) ON DELETE CASCADE,
|
|
CONSTRAINT fk_plans_crops FOREIGN KEY(crop_id) REFERENCES crops(id) ON DELETE RESTRICT
|
|
)`,
|
|
`ALTER TABLE plans MODIFY COLUMN field_id BIGINT NULL`,
|
|
`ALTER TABLE plans ADD COLUMN IF NOT EXISTS target_ref VARCHAR(80) NOT NULL DEFAULT '' AFTER field_id`,
|
|
`ALTER TABLE plans ADD COLUMN IF NOT EXISTS target_name VARCHAR(140) NOT NULL DEFAULT '' AFTER target_ref`,
|
|
}
|
|
for _, stmt := range stmts {
|
|
if _, err := db.Exec(stmt); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return seedCrops(db)
|
|
}
|
|
|
|
func (a *App) getSettings() (Settings, error) {
|
|
var s Settings
|
|
err := a.db.QueryRow(`SELECT days_per_month,current_month,current_day FROM settings WHERE id=1`).
|
|
Scan(&s.DaysPerMonth, &s.CurrentMonth, &s.CurrentDay)
|
|
return s, err
|
|
}
|
|
|
|
func (a *App) listFields() ([]Field, error) {
|
|
rows, err := a.db.Query(`SELECT id,number,name,owned,COALESCE(group_key,''),COALESCE(group_name,'') FROM fields ORDER BY number`)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
var out []Field
|
|
for rows.Next() {
|
|
var f Field
|
|
if err := rows.Scan(&f.ID, &f.Number, &f.Name, &f.Owned, &f.GroupKey, &f.GroupName); err != nil {
|
|
return nil, err
|
|
}
|
|
out = append(out, f)
|
|
}
|
|
return out, rows.Err()
|
|
}
|
|
|
|
func (a *App) getFieldsByIDs(ids []int64) ([]Field, error) {
|
|
args := make([]any, 0, len(ids))
|
|
for _, id := range ids {
|
|
args = append(args, id)
|
|
}
|
|
q := fmt.Sprintf(`SELECT id,number,name,owned,COALESCE(group_key,''),COALESCE(group_name,'') FROM fields WHERE id IN (%s)`, placeholders(len(ids)))
|
|
rows, err := a.db.Query(q, args...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
var out []Field
|
|
for rows.Next() {
|
|
var f Field
|
|
if err := rows.Scan(&f.ID, &f.Number, &f.Name, &f.Owned, &f.GroupKey, &f.GroupName); err != nil {
|
|
return nil, err
|
|
}
|
|
out = append(out, f)
|
|
}
|
|
return out, rows.Err()
|
|
}
|
|
|
|
func (a *App) listCrops() ([]Crop, error) {
|
|
rows, err := a.db.Query(`SELECT id,name,sow_start_month,sow_end_month,grow_months,regrow_enabled,regrow_cycles FROM crops ORDER BY name`)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
var out []Crop
|
|
for rows.Next() {
|
|
var c Crop
|
|
if err := rows.Scan(&c.ID, &c.Name, &c.SowStartMonth, &c.SowEndMonth, &c.GrowMonths, &c.RegrowEnabled, &c.RegrowCycles); err != nil {
|
|
return nil, err
|
|
}
|
|
out = append(out, c)
|
|
}
|
|
return out, rows.Err()
|
|
}
|
|
|
|
func (a *App) listPlans() ([]Plan, error) {
|
|
rows, err := a.db.Query(`
|
|
SELECT p.id,p.field_id,COALESCE(p.target_ref,''),COALESCE(p.target_name,''),p.crop_id,COALESCE(c.name,''),COALESCE(c.grow_months,1),p.start_month,p.start_day,p.harvest_month,p.harvest_day,COALESCE(p.notes,''),COALESCE(c.regrow_enabled,0),COALESCE(c.regrow_cycles,0)
|
|
FROM plans p
|
|
JOIN crops c ON c.id=p.crop_id
|
|
ORDER BY p.start_month,p.start_day,p.id DESC`)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
var out []Plan
|
|
for rows.Next() {
|
|
var p Plan
|
|
if err := rows.Scan(&p.ID, &p.FieldID, &p.TargetRef, &p.TargetName, &p.CropID, &p.CropName, &p.GrowMonths, &p.StartMonth, &p.StartDay, &p.HarvestMonth, &p.HarvestDay, &p.Notes, &p.RegrowEnabled, &p.RegrowCycles); err != nil {
|
|
return nil, err
|
|
}
|
|
out = append(out, p)
|
|
}
|
|
return out, rows.Err()
|
|
}
|
|
|
|
func (a *App) listCropSteps() ([]CropStep, error) {
|
|
rows, err := a.db.Query(`
|
|
SELECT s.id,s.crop_id,COALESCE(c.name,''),s.phase,s.month_offset,s.title
|
|
FROM crop_steps s
|
|
JOIN crops c ON c.id=s.crop_id
|
|
ORDER BY c.name, s.phase, s.month_offset, s.id`)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
|
|
var out []CropStep
|
|
for rows.Next() {
|
|
var s CropStep
|
|
if err := rows.Scan(&s.ID, &s.CropID, &s.CropName, &s.Phase, &s.MonthOffset, &s.Title); err != nil {
|
|
return nil, err
|
|
}
|
|
out = append(out, s)
|
|
}
|
|
return out, rows.Err()
|
|
}
|
|
|
|
func (a *App) listCropStepsMap() (map[int64][]CropStep, error) {
|
|
all, err := a.listCropSteps()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
m := make(map[int64][]CropStep)
|
|
for _, s := range all {
|
|
m[s.CropID] = append(m[s.CropID], s)
|
|
}
|
|
return m, nil
|
|
}
|
|
|
|
func seedCrops(db *sql.DB) error {
|
|
items := []Crop{
|
|
{Name: "Weizen", SowStartMonth: 9, SowEndMonth: 11, GrowMonths: 10},
|
|
{Name: "Gerste", SowStartMonth: 9, SowEndMonth: 10, GrowMonths: 9},
|
|
{Name: "Hafer", SowStartMonth: 3, SowEndMonth: 4, GrowMonths: 4},
|
|
{Name: "Raps", SowStartMonth: 8, SowEndMonth: 9, GrowMonths: 11},
|
|
{Name: "Mais", SowStartMonth: 4, SowEndMonth: 5, GrowMonths: 6},
|
|
{Name: "Sorghum", SowStartMonth: 4, SowEndMonth: 5, GrowMonths: 5},
|
|
{Name: "Sojabohnen", SowStartMonth: 4, SowEndMonth: 5, GrowMonths: 5},
|
|
{Name: "Sonnenblumen", SowStartMonth: 4, SowEndMonth: 5, GrowMonths: 5},
|
|
{Name: "Kartoffeln", SowStartMonth: 3, SowEndMonth: 4, GrowMonths: 7},
|
|
{Name: "Zuckerrueben", SowStartMonth: 3, SowEndMonth: 4, GrowMonths: 8},
|
|
{Name: "Baumwolle", SowStartMonth: 2, SowEndMonth: 3, GrowMonths: 8},
|
|
}
|
|
for _, c := range items {
|
|
if _, err := db.Exec(`INSERT INTO crops(name,sow_start_month,sow_end_month,grow_months,regrow_enabled,regrow_cycles) VALUES (?,?,?,?,0,0) ON DUPLICATE KEY UPDATE sow_start_month=VALUES(sow_start_month),sow_end_month=VALUES(sow_end_month),grow_months=VALUES(grow_months)`, c.Name, c.SowStartMonth, c.SowEndMonth, c.GrowMonths); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|