audioaxis/main.go

package main

import (
	"fmt"
	"http"
	"exp/template"
	mysql "github.com/Philio/GoMySQL"
	"os"
	"json"
	"strconv"
	"time"
)

type Song struct {
	Yid string
	Title string
	User string
}
type Playlist struct {
	Id string
}

var templates map[string]*template.Template
const debug = true
var db *mysql.Client

// given an id ('abcd1234'), return the pid (1)
func getpid(id string) int {
	query, err := db.Prepare("SELECT `pid` FROM `playlist` WHERE `id` = ?;")
	if err != nil {
		return -1
	}
	err = query.BindParams(id)
	if err != nil {
		return -1
	}
	err = query.Execute()
	if err != nil {
		return -1
	}
	var pid int
	query.BindResult(&pid)
	_, err = query.Fetch()
	if err != nil {
		return -1
	}
	err = query.FreeResult()
	if err != nil {
		return -1
	}
	return pid
}

func home(w http.ResponseWriter, r *http.Request) {
	fmt.Fprintf(w, "path is %s", r.URL.Path[1:])
}

func playlist(w http.ResponseWriter, r *http.Request) {
	id := r.URL.Path[len("/p/"):]
	if len(id) != 8 {
		http.Redirect(w, r, "/", 303)
		return
	}
	playlist := Playlist{Id: id}

	if debug {
		t, err := template.ParseFile("templates/p.html")
		if err != nil {
			http.Error(w, err.String(), http.StatusInternalServerError)
			return
		}
		err = t.Execute(w, playlist)
		if err != nil {
			w.Write([]byte(err.String()))
			fmt.Fprintln(os.Stderr, err.String())
			return
		}
	} else {
		err := templates["p"].Execute(w, playlist)
		if err != nil {
			fmt.Fprintln(os.Stderr, err.String())
		}
	}
}

func add(w http.ResponseWriter, r *http.Request) {
	q := r.URL.Query()
	pid := getpid(q.Get("pid"))
	if pid == -1 {
		http.Error(w, "invalid pid", http.StatusInternalServerError)
		return
	}

	sql :="INSERT INTO `song` (`pid`,`yid`,`title`,`user`) VALUES(%d,'%s','%s','%s')"
	sql = fmt.Sprintf(sql, pid,
			db.Escape(q.Get("yid")),
			db.Escape(q.Get("title")),
			db.Escape(q.Get("user")))
	err := db.Query(sql)
	if err != nil {
		http.Error(w, err.String(), http.StatusInternalServerError)
		return
	}

	w.Write([]byte("1"))
	addUpdate(pid, addAction,
			&Song{Yid: q.Get("yid"), Title: q.Get("title"), User: q.Get("user")})
}

func remove(w http.ResponseWriter, r *http.Request) {
	q := r.URL.Query()
	pid := getpid(q.Get("pid"))
	if pid == -1 {
		http.Error(w, "invalid pid", http.StatusInternalServerError)
		return
	}

	sql :="DELETE FROM `song` WHERE `pid` = %d AND yid = '%s'"
	sql = fmt.Sprintf(sql, pid, q.Get("yid"))
	err := db.Query(sql)
	if err != nil {
		http.Error(w, err.String(), http.StatusInternalServerError)
		return
	}

	w.Write([]byte("1"))
	addUpdate(pid, removeAction, &Song{Yid: q.Get("yid")})
}

func poll(w http.ResponseWriter, r *http.Request) {
	q := r.URL.Query()
	timestamp := q.Get("timestamp")
	if timestamp == "0" {
		query, err := db.Prepare("SELECT `yid`,`title`,`user` FROM `playlist` JOIN `song` WHERE `id` = ?;")
		if err != nil {
			http.Error(w, err.String(), http.StatusInternalServerError)
			return
		}
		err = query.BindParams(q.Get("pid"))
		if err != nil {
			http.Error(w, err.String(), http.StatusInternalServerError)
			return
		}
		err = query.Execute()
		if err != nil {
			http.Error(w, err.String(), http.StatusInternalServerError)
			return
		}

		updates := make([]Update, 0, 2)
		for {
			song := new(Song)
			query.BindResult(&song.Yid, &song.Title, &song.User)
			eof, err := query.Fetch()
			if err != nil {
				http.Error(w, err.String(), http.StatusInternalServerError)
				return
			}
			if eof {
				break
			}
			updates = append(updates, Update{Song: song, Action: addAction, Timestamp: time.Nanoseconds()})
		}
		err = query.FreeResult()
		if err != nil {
			http.Error(w, err.String(), http.StatusInternalServerError)
			return
		}

		output, err := json.MarshalForHTML(updates)
		if err != nil {
			http.Error(w, err.String(), http.StatusInternalServerError)
			return
		}
		w.Write(output)
	} else {
		timestamp, err := strconv.Atoi64(q.Get("timestamp"))
		if err != nil {
			http.Error(w, err.String(), http.StatusInternalServerError)
			return
		}
		var update *Update
		for i := 0; i < 30; i++ {
			update = getUpdates(getpid(q.Get("pid")), timestamp)
			if update != nil {
				w.Write([]byte("["))
				for update != nil {
					output, err := json.MarshalForHTML(update)
					if err == nil {
						w.Write(output)
					}
					update = update.Next
					if update != nil {
						w.Write([]byte(","))
					}
				}
				w.Write([]byte("]"))
				return
			}
			time.Sleep(1e9) // 1 second
		}
		w.Write([]byte("[]"))
	}
}

func main() {
	templates = make(map[string]*template.Template)
	for _, path := range []string{"p"} {
		t, err := template.ParseFile("templates/" + path + ".html")
		if err != nil {
			fmt.Println(err)
			return
		}
		templates[path] = t
	}

	var err os.Error
	//db, err = mysql.DialTCP("raylu.net", "audio", "audio", "audio")
	db, err = mysql.DialTCP("173.228.31.111", "audio", "audio", "audio")
	if err != nil {
		fmt.Println(err)
		os.Exit(1)
	}
	db.Reconnect = true

	http.HandleFunc("/", home)
	http.HandleFunc("/p/", playlist)
	http.HandleFunc("/add/", add)
	http.HandleFunc("/remove/", remove)
	http.HandleFunc("/poll/", poll)
	err = http.ListenAndServe("localhost:8000", nil)
	if err != nil {
		fmt.Println(err)
		os.Exit(1)
	}
}