package chipmunk import ( "fmt" "github.com/jmoiron/sqlx" _ "github.com/lib/pq" ) type DB struct { db *sqlx.DB } func NewDB(dbhost, dbname string) (*DB, error) { var err error config := fmt.Sprintf( "dbname=%s host=%s sslmode=disable", dbname, dbhost, ) db, err := sqlx.Connect( "postgres", config, ) if err != nil { return nil, err } d := &DB{db} d.initializeDB() return d, nil } func (d *DB) initializeDB() { // XXX ignoring errors d.db.Exec(createdb) d.db.Exec(primeCategories) } func (d *DB) getCategories() ([]category, error) { results := []category{} rows, err := d.db.Queryx("SELECT id, name, budget FROM categories") if err != nil { return nil, err } for rows.Next() { var result category err := rows.StructScan(&result) if err != nil { return nil, err } results = append(results, result) } return results, nil } func (d *DB) getUsers() ([]user, error) { results := []user{} rows, err := d.db.Queryx("SELECT id, email, admin FROM users") if err != nil { return nil, err } for rows.Next() { var result user err := rows.StructScan(&result) if err != nil { return nil, err } results = append(results, result) } return results, nil } func (d *DB) adminUser(e string) bool { result := user{} row := d.db.QueryRow("SELECT admin FROM users WHERE email = $1", e, ) err := row.Scan(&result) if err != nil { return false } return result.Admin } func (d *DB) getTranxs() ([]tranx, error) { results := []tranx{} rows, err := d.db.Queryx("SELECT * FROM tranx") if err != nil { return nil, err } for rows.Next() { var result tranx err := rows.StructScan(&result) if err != nil { return nil, err } results = append(results, result) } return results, nil } //func (d *DB) checkOwner(filename, client string) error { // row := d.db.QueryRowx("SELECT client FROM pics WHERE filename = $1", filename) // var owner string // err := row.Scan(&owner) // if err == sql.ErrNoRows { // return errors.New("file not in DB") // } // if owner != strings.ToLower(client) { // return errors.New("filename did not match owner") // } // return nil //} // //func (d *DB) getPic(filename string) (Pic, error) { // p := Pic{} // err := d.db.QueryRowx( // ` // SELECT // filename, // category_id as category, // client, // categorized, // reported // FROM pics // WHERE // filename = $1 // LIMIT 1 // `, // filename, // ).StructScan(&p) // logs.Debug("%+v", p) // if err == sql.ErrNoRows { // return Pic{}, errors.New("pic not found") // } // return p, nil //} // //func (d *DB) votesForPic(filename string) ([]int, error) { // results := []int{} // rows, err := d.db.Queryx(`SELECT vote FROM votes WHERE filename = $1`, filename) // if err != nil { // return []int{}, err // } // for rows.Next() { // var result int // err := rows.Scan(&result) // if err != nil { // return []int{}, err // } // results = append(results, result) // } // return results, nil //} // //func (d *DB) closeVoting(filename string) error { // _, err := d.db.Exec( // `UPDATE pics SET categorized = true WHERE filename = $1`, // filename, // ) // return err //} // //func (d *DB) votingClosed(filename string) (bool, error) { // row := d.db.QueryRow("SELECT categorized FROM pics WHERE filename = $1", filename) // already := false // err := row.Scan(&already) // return already, err //} // //type Token string // //func (d *DB) tokenForFile(filename string) (Token, error) { // row := d.db.QueryRow(` // SELECT // clients.token // FROM pics, clients // WHERE // pics.client = clients.id // AND pics.filename = $1 // `, // filename, // ) // token := sql.NullString{} // err := row.Scan(&token) // return Token(token.String), err //} // //type Pic struct { // Filename string `json:"filename"` // Category int `json:"category"` // Client string `json:"client"` // Categorized bool `json:"categorized"db:"categorized"` // Reported bool `json:"reported"db:"reported"` //} // //type PicDetails struct { // // did the request work; this goes along with HTTP status codes // Success bool `json:"success"` // // Filename string `json:"filename"` // // Message string `json:"message"` // Works bool `json:"works"` // Categorized bool `json:"categorized"` //} // //func (d *DB) getPicDetails(filename string) (PicDetails, error) { // pd := PicDetails{ // Success: false, // Filename: filename, // Message: "unknown ranking", // } // votes, err := d.votesForPic(filename) // if err != nil { // return pd, err // } // message, works, done, err := categorize(votes) // if err != nil { // return pd, err // } // pd.Success = true // pd.Message = message // pd.Works = works // pd.Categorized = done // return pd, nil //} // //type VoteReq struct { // Filename string `json:"filename"` // Client string `json:"client"` // Value int `json:"value"` //} // //type Category struct { // Id int `json:"id"db:"id"` // Name string `json:"name"db:"name"` //} // // //func (d *DB) flagPic(filename string, reporter string) (int, error) { // logs.Debug("(%s, %s)", filename, reporter) // // var count int // row := d.db.QueryRow( // "SELECT COUNT(*) FROM reports WHERE filename = $1 AND reporter = $2", // filename, // reporter, // ) // err := row.Scan(&count) // if err != nil { // return http.StatusInternalServerError, err // } // if count > 0 { // return http.StatusBadRequest, fmt.Errorf("%s already reported %s", reporter, filename) // } // // _, err = d.db.Exec( // `INSERT INTO reports (filename, reporter) VALUES ($1, $2)`, // filename, // reporter, // ) // if err != nil { // return http.StatusBadRequest, err // } // // var culprit string // err = d.db.Get(&culprit, ` // SELECT // client // FROM pics, clients // WHERE // pics.client = clients.id AND // pics.filename = $1 // `, // filename, // ) // if err != nil { // return http.StatusInternalServerError, err // } // // _, err = d.db.Exec(` // UPDATE // clients // SET // infractions = infractions + 1 // WHERE // clients.id = $1 // `, // culprit, // ) // if err != nil { // return http.StatusInternalServerError, err // } // // _, err = d.db.Exec(` // UPDATE // pics // SET // reported = true // WHERE // filename = $1 // `, // filename, // ) // if err != nil { // return http.StatusInternalServerError, err // } // return http.StatusCreated, nil //} // //type client struct { // Id string // Token sql.NullString // Infractions int // db *sqlx.DB //} // //const reportThreshold = 3 // //func (c *client) isBlocked() bool { // if c.Infractions >= reportThreshold { // return true // } // return false //} // //func (c *client) MarshalJSON() ([]byte, error) { // r := struct { // Id string `json:"id"` // Infractions int `json:"infractions"` // }{ // Id: c.Id, // Infractions: c.Infractions, // } // // return json.Marshal(r) //} // //func (c *client) SetToken(token string) error { // _, err := c.db.Exec( // `UPDATE clients SET token = $1 WHERE id = $2`, // token, // c.Id, // ) // if err != nil { // return err // } // c.Token = sql.NullString{ // String: token, // Valid: true, // } // return nil //} // //// getOrCreateClient performs db operations requisite to fetch current info or //// create a new client. //// //// returns: //// *client, created or not, error //func (d *DB) getOrCreateClient(clientId string) (*client, bool, error) { // var created bool // c := &client{ // db: d.db, // } // err := d.db.QueryRowx( // "SELECT * FROM clients WHERE id = $1", // clientId, // ).StructScan(c) // if err == sql.ErrNoRows { // _, err = d.db.Exec( // `INSERT INTO clients (id) VALUES ($1)`, // clientId, // ) // if err != nil { // return nil, false, fmt.Errorf( // "problem inserting client: %s: %+v", // clientId, // err, // ) // } // created = true // err := d.db.QueryRowx( // "SELECT * FROM clients WHERE id = $1", // clientId, // ).StructScan(c) // if err != nil { // return nil, true, err // } // } // return c, created, nil //} // //type UploadResp struct { // Success bool `json:"success"` // Sha string `json:"sha1"` //}