package filicop

import ()

type Vehicle struct {
	Id      int64  `json:"id"`
	IdFleet int64  `json:"id_fleet"`
	Label   string `json:"label"`
}

func GetVehicles() (vehicles []Vehicle, err error) {
	if rows, errr := DBQuery("SELECT id_vehicle, id_fleet, label FROM vehicles"); errr != nil {
		return nil, errr
	} else {
		defer rows.Close()

		for rows.Next() {
			var v Vehicle
			if err = rows.Scan(&v.Id, &v.IdFleet, &v.Label); err != nil {
				return
			}
			vehicles = append(vehicles, v)
		}
		if err = rows.Err(); err != nil {
			return
		}

		return
	}
}

func GetVehicle(id int64) (v Vehicle, err error) {
	err = DBQueryRow("SELECT id_vehicle, id_fleet, label FROM vehicles WHERE id_vehicle = ?", id).Scan(&v.Id, &v.IdFleet, &v.Label)
	return
}

func (f Fleet) NewVehicle(label string) (Vehicle, error) {
	if res, err := DBExec("INSERT INTO vehicles (id_fleet, label) VALUES (?, ?)", f.Id, label); err != nil {
		return Vehicle{}, err
	} else if vid, err := res.LastInsertId(); err != nil {
		return Vehicle{}, err
	} else {
		return Vehicle{vid, f.Id, label}, nil
	}
}

func (v Vehicle) Update() (int64, error) {
	if res, err := DBExec("UPDATE vehicles SET id_fleet = ?, label = ? WHERE id_vehicle = ?", v.IdFleet, v.Label, v.Id); err != nil {
		return 0, err
	} else if nb, err := res.RowsAffected(); err != nil {
		return 0, err
	} else {
		return nb, err
	}
}

func (v Vehicle) Delete() (int64, error) {
	if res, err := DBExec("DELETE FROM vehicles WHERE id_vehicle = ?", v.Id); err != nil {
		return 0, err
	} else if nb, err := res.RowsAffected(); err != nil {
		return 0, err
	} else {
		return nb, err
	}
}

func ClearVehicles() (int64, error) {
	if res, err := DBExec("DELETE FROM vehicles"); err != nil {
		return 0, err
	} else if nb, err := res.RowsAffected(); err != nil {
		return 0, err
	} else {
		return nb, err
	}
}

func (f Fleet) ClearVehicles() (int64, error) {
	if res, err := DBExec("DELETE FROM vehicles WHERE id_fleet = ?", f.Id); err != nil {
		return 0, err
	} else if nb, err := res.RowsAffected(); err != nil {
		return 0, err
	} else {
		return nb, err
	}
}