audioaxis/main.go

package main

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

type Song struct {
	Yid string
	Title string
	User string
}
type Playlist struct {
	Id string
	Songs []*Song
}

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
	}
	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(id)
	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
	}

	playlist := Playlist{Id: id, Songs: make([]*Song, 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
		}
		playlist.Songs = append(playlist.Songs, song)
	}
	err = query.FreeResult()
	if err != nil {
		http.Error(w, err.String(), http.StatusInternalServerError)
		return
	}

	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"))
}

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"))
}

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")
	if err != nil {
		fmt.Println(err)
		os.Exit(1)
	}

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