package database import ( "crypto/rsa" "git.xeserv.us/xena/route/routerpc" "git.xeserv.us/xena/route/utils" r "github.com/GoRethink/gorethink" ) // DB is the high level wrapper to the datastore. type DB struct { s *r.Session } // New takes a rethinkdb host and databasea and sets up a connection. func New(host, database string) (*DB, error) { session, err := r.Connect(r.ConnectOpts{ Address: host, Database: database, }) if err != nil { return nil, err } db := &DB{ s: session, } return db, nil } var tables = []string{ "certs", "routes", } // Route is a single route object serialized to rethinkdb. type Route struct { ID string `gorethink:"id,omitempty"` Hostname string `gorethink:"hostname"` OnionHostname string `gorethink:"onionhostname"` Token string `gorethink:"token"` OnionKey []byte `gorethink:"onionKey"` // PEM-encoded } // SaveRoute adds the route to the database. func (db *DB) SaveRoute(resp *routerpc.AddHostResponse) error { bytes := utils.RSAPrivateKeyToPem(resp.PrivKey.(*rsa.PrivateKey)) rt := &Route{ Hostname: resp.Hostname, OnionHostname: resp.OnionHostname, Token: resp.Token, OnionKey: bytes, } // TODO: check if OnionHostname or Hostname actually exists in // the database. RethinkDB doesn't support unique secondary indexes. _, err := r.Table("routes").Insert(rt).RunWrite(db.s) if err != nil { return err } return nil } // GetAllRoutes fetches all of the route entries from the database // and returns them. This is intended for the startup process or // admin tooling. func (db *DB) GetAllRoutes() ([]Route, error) { rows, err := r.Table("routes").Run(db.s) if err != nil { return nil, err } var routes []Route err = rows.All(&routes) if err != nil { return nil, err } return routes, nil }