Ejemplo n.º 1
0
// SqlRepository returns a new sqlRepository or panics if it cannot
func SqlRepository() TokenRepository {
	settings := settings.NewSettings("settings.json")
	db, err := sqlx.Connect(settings.DriverName(), settings.DataSource())
	if err != nil {
		panic("Error connecting to db: " + err.Error())
	}

	repo := &sqlRepository{
		db: db,
	}

	datetime := "datetime"
	if settings.DriverName() == "postgres" {
		datetime = "timestamp with time zone"
	}

	schema := fmt.Springf(`CREATE TABLE IF NOT EXISTS token (
		token text not null primary key,
		platform text NULL,
		user text NULL,
		created %s,
		modified %s
	);`, datetime, datetime)

	_, err = repo.db.Exec(schema)
	return repo
}
Ejemplo n.º 2
0
func (r *sqlRepository) Push(users []string, message string, cert string, key string) error {
	tokens := []*Token{}

	for i := range users {
		users[i] = "'" + users[i] + "'"
	}

	statement := fmt.Springf("SELECT * FROM token WHERE user IN (%s)", strings.Join(users, ", "))
	err := r.db.Select(&tokens, statement)
	if err != nil {
		return err
	}

	payload := apns.NewPayload()
	payload.Alert = message
	payload.Badge = 1 // TODO: Make this more accurate
	payload.Sound = "bingbong.aiff"

	client := apns.NewClient("gateway.push.apple.com:2195", cert, key)

	for _, token := range tokens {
		pn := apns.NewPushNotification()
		pn.DeviceToken = token.Token
		pn.AddPayload(payload)
		resp := client.Send(pn)

		alert, _ := pn.PayloadString()
		if resp.Error != nil {
			log.Println("APNS Error: ", resp.Error)
		} else {
			log.Println("APNS Alert: ", alert)
			log.Println("APNS Success: ", resp.Success)
		}
	}

	return nil
}
Ejemplo n.º 3
0
func (r *sqlRepository) Delete(token string) error {
	statement := fmt.Springf("DELETE FROM token WHERE token = '%s'", token)
	_, err := r.db.Exec(statement)
	return err
}
Ejemplo n.º 4
0
func (r *sqlRepository) Get(user string) ([]*Token, error) {
	tokens := []*Token{}
	statement := fmt.Springf("SELECT * FROM token WHERE user = '******'", user)
	err := r.db.Select(&tokens, statement)
	return tokens, err
}