Ejemplo n.º 1
0
func New(config *config.Config) *Broker {
	broker := new(Broker)
	broker.cfg = config

	// Connect to the message queue server
	address := fmt.Sprintf("amqp://%s:%s@%s:%d/", config.GetQueueUsername(),
		config.GetQueuePassword(),
		config.GetQueueHost(),
		config.GetQueuePort())

	conn, err := amqp.Dial(address)
	if err != nil {
		fmt.Printf("There was an error connecting to the broker")
		panic(err)
	}
	broker.conn = conn

	// Create a new channel
	channel, err := conn.Channel()
	if err != nil {
		panic(err)
	}
	broker.channel = channel

	broker.initializeQueues()
	return broker
}
Ejemplo n.º 2
0
///////////////////////////////////////////////////////////////
// Database housekeeping
func New(cfg *config.Config) *Db {
	db := new(Db)
	db.cfg = cfg
	connection, err := sql.Open("sqlite3", cfg.GetDbFilename())

	if err != nil {
		fmt.Printf("There was an error opening the database\n")
		return nil
	}
	db.connection = connection
	return db
}
Ejemplo n.º 3
0
func newOAuthClient(ctx context.Context, config *oauth2.Config, cfg *config.Config) *http.Client {
	cacheFile := tokenCacheFile(config)
	token, err := tokenFromFile(cacheFile)
	if err != nil {
		token = tokenFromWeb(ctx, config, cfg)
		token.RefreshToken = cfg.GetYoutubeRefreshToken()
		saveToken(cacheFile, token)
	} else {
		fmt.Printf("Using cached token from %q\n", cacheFile)
	}

	return config.Client(ctx, token)
}
Ejemplo n.º 4
0
func tokenFromWeb(ctx context.Context, config *oauth2.Config, cfg *config.Config) *oauth2.Token {
	ch := make(chan string)
	randState := fmt.Sprintf("st%d", time.Now().UnixNano())

	lis, err := net.Listen("tcp", fmt.Sprintf(":%d", cfg.GetYoutubeOAuthPort()))
	if err != nil {
		panic(err)
	}
	// make our own handler that puts all requests in a wait group.
	h := http.NewServeMux()

	// add a close handlefunc to that handler
	h.HandleFunc("/youtube_callback/", func(rw http.ResponseWriter, req *http.Request) {
		if req.URL.Path == "/favicon.ico" {
			http.Error(rw, "", 404)
			return
		}
		if req.FormValue("state") != randState {
			fmt.Printf("State doesn't match: req = %#v", req)
			http.Error(rw, "", 500)
			return
		}
		if code := req.FormValue("code"); code != "" {
			fmt.Fprintf(rw, "<h1>Success</h1>Authorized.")
			rw.(http.Flusher).Flush()
			ch <- code
			lis.Close()
			return
		}
		fmt.Printf("no code")
		http.Error(rw, "", 500)
		lis.Close()
	})

	//listen and serve until listner is closed
	go http.Serve(lis, h)

	// TODO fix this hard-coded localhost thingy
	config.RedirectURL = fmt.Sprintf("http://127.0.0.1:%d/youtube_callback/", cfg.GetYoutubeOAuthPort())
	authURL := config.AuthCodeURL(randState)
	go openURL(authURL)
	fmt.Printf("Authorize this app at: %s", authURL)
	code := <-ch

	token, err := config.Exchange(ctx, code)
	if err != nil {
		fmt.Printf("Token exchange error: %v", err)
	}
	return token
}
Ejemplo n.º 5
0
func send_email(email string, title string, html string, config *config.Config) {
	gun := mailgun.NewMailgun("munch.utkarshsinha.com",
		config.GetMailPrivateKey(),
		config.GetMailPublicKey())
	msg := mailgun.NewMessage("Broccoli Munch <*****@*****.**>",
		title,
		html,
		"*****@*****.**")
	msg.SetHtml(html)
	response, id, err := gun.Send(msg)

	fmt.Printf("%s\n", response)
	fmt.Printf("%s\n", id)
	fmt.Printf("%s\n", err)
}