package sites

import (
	"database/sql"
	"fmt"
)

const createSitesTable = `CREATE TABLE IF NOT EXISTS sites (
id INTEGER NOT NULL PRIMARY KEY,
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
name TEXT NOT NULL,
url TEXT NOT NULL
);`

func CreateTable(db *sql.DB) {
	if _, err := db.Exec(createSitesTable); err != nil {
		fmt.Println(fmt.Errorf(err.Error()))
	}
}

type Site struct {
	id         uint64
	created_at uint64
	Name       string
	Url        string
}

func (s *Site) Save(db *sql.DB) {
	if s.id != 0 {
		query := `UPDATE SITES
		SET
			name=?,
			url=?
		WHERE id =? 
	`
		db.Exec(query, s.Name, s.Url, s.id)
	} else {
		query := `INSERT INTO SITES (name, url) VALUES (?, ?)`
		db.Exec(query, s.Name, s.Url)
	}
}