main.go

package main

import (
	"database/sql"
	"fmt"
	_ "github.com/mattn/go-sqlite3"
	"io/ioutil"
	"net/http"
	"os"
	"path/filepath"
	"regexp"
	"strings"
)

const AUDITOR_URL = "http://oh-mahoning-auditor.publicaccessnow.com/DesktopModules/OWS/IM.aspx?&mpropertynumber=%s&_OWS_=lxC:1,lxP:0,s:1x,m:427,pm:116,p:89,lxSrc:dnn$ctr427$PropertyInfo,key:Module.Load,file:/DesktopModules/PropertyInfo/App_LocalResources/PropertyInfo.ascx.resx,pp:0"

func main() {
	arg := os.Args

	if len(arg) == 1 {
		sync()
		fmt.Fprintf(os.Stderr, "error: no command given\n")
		os.Exit(1)
	}

	cmd := arg[1]

	input := "" //FIXME
	if len(arg) > 2 {
		input = arg[2]
	}

	if cmd == "add" {
		add(input)
	} else if cmd == "remove" {
		remove(input)
	} else if cmd == "list" {
		list()
	} else if cmd == "sync" {
		sync()
		fmt.Println("Sync and compare done.")
	} else {
		fmt.Fprintf(os.Stderr, "error: invalid command\n")
		os.Exit(1)
	}
}

func sync() {
	db := _get_or_create_db()

	rows, _ := db.Query("SELECT id, parcel_id, owner FROM properties")

	var id int
	var parcel_id string
	var owner string

	updates := make(map[int]string)

	for rows.Next() {
		rows.Scan(&id, &parcel_id, &owner)

		new_owner := _get_owner(parcel_id)

		if owner != new_owner {
			fmt.Printf("New owner for %s: %s\n", parcel_id, new_owner)
			updates[id] = new_owner
		}
	}

	for k, v := range updates {
		statement, _ := db.Prepare("UPDATE properties SET owner=? WHERE id=?")
		statement.Exec(v, k)
	}

	db.Close()
}

func list() {
	db := _get_or_create_db()

	rows, _ := db.Query("SELECT id, parcel_id, owner FROM properties")

	var id int
	var parcel_id string
	var owner string

	for rows.Next() {
		rows.Scan(&id, &parcel_id, &owner)
		fmt.Println(parcel_id + " " + owner)
	}
	db.Close()
}

func add(parcel_id string) {
	parcel_id = strings.TrimSpace(parcel_id)

	db := _get_or_create_db()

	owner := _get_owner(parcel_id)

	var id int
	err := db.QueryRow(`SELECT id FROM properties WHERE parcel_id=?`, parcel_id).Scan(&id)

	if err == sql.ErrNoRows {
		statement, _ := db.Prepare("INSERT INTO properties(parcel_id, owner) values(?,?)")
		statement.Exec(parcel_id, owner)
		fmt.Printf("Parcel added %s: %s\n", parcel_id, owner)
	} else {
		fmt.Fprintf(os.Stderr, "error: This parcel already exists in the database\n")
		os.Exit(1)
	}

	db.Close()
}

func remove(parcel_id string) {
	parcel_id = strings.TrimSpace(parcel_id)

	db := _get_or_create_db()

	statement, err := db.Prepare("DELETE FROM properties WHERE parcel_id=?")
	_checkErr(err)

	statement.Exec(parcel_id)

	fmt.Printf("Parcel removed: %s\n", parcel_id)

	db.Close()
}

func _get_or_create_db() *sql.DB {
	ex, err := os.Executable()
	if err != nil {
		panic(err)
	}
	exPath := filepath.Dir(ex)

	db, _ := sql.Open("sqlite3", exPath+string(os.PathSeparator)+"snipe.db")

	statement, _ := db.Prepare("CREATE TABLE IF NOT EXISTS properties (id INTEGER PRIMARY KEY, parcel_id TEXT, owner TEXT)")
	statement.Exec()

	return db

}

func _checkErr(err error) {
	if err != nil {
		panic(err)
	}
}

func _get_owner(parcel_id string) string {
	url := fmt.Sprintf(AUDITOR_URL, parcel_id)
	res, _ := http.Get(url)
	body, _ := ioutil.ReadAll(res.Body)
	res.Body.Close()

	r, _ := regexp.Compile("Owner Name</td><td >(.+)</td>")

	match := r.FindStringSubmatch(string(body[:]))

	if len(match) < 2 {
		fmt.Fprintf(os.Stderr, "No property found for that parcel_id\n")
		os.Exit(1)
	}

	return strings.TrimSpace(match[1])
}