コード例 #1
0
ファイル: handler.go プロジェクト: Monory/messenger
func chatHandler(w http.ResponseWriter, r *http.Request, db *sql.DB) {
	cookieToken, _ := r.Cookie("token")
	token := auth.NewUserToken()
	token.FromString(cookieToken.Value)

	chatToken, err := auth.MakeChatToken(db, token)
	if err != nil {
		internalServerError(w, err)
		return
	}

	var c http.Cookie
	c.Name = "chat_token"
	c.Value = chatToken.String()
	c.Secure = true
	http.SetCookie(w, &c)

	http.ServeFile(w, r, root+"/index.html")
}
コード例 #2
0
ファイル: handler.go プロジェクト: Monory/messenger
func defaultHandler(w http.ResponseWriter, r *http.Request, db *sql.DB) {
	cookieToken, err := r.Cookie("token")
	if err != nil {
		if err == http.ErrNoCookie {
			http.Redirect(w, r, "/auth", http.StatusSeeOther)
			return
		}
		internalServerError(w, err)
		return
	}

	token := auth.NewUserToken()
	err = token.FromString(cookieToken.Value)
	if err == nil {
		err = auth.CheckUserToken(db, token)
	}
	if err != nil {
		if _, ok := err.(auth.AuthError); ok {
			var c http.Cookie
			c.Name = "token"
			c.MaxAge = -1
			c.Secure = true
			c.HttpOnly = true

			http.SetCookie(w, &c)
			http.Redirect(w, r, "/", http.StatusSeeOther)
			return
		}
		internalServerError(w, err)
		return
	}

	if r.URL.Path != "/" {
		http.Redirect(w, r, "/", http.StatusSeeOther)
		return
	}

	chatHandler(w, r, db)
}
コード例 #3
0
ファイル: server.go プロジェクト: Monory/messenger
func (s *Server) Listen(db *sql.DB) {

	log.Println("Listening server...")
	s.db = db

	// websocket handler
	onConnected := func(ws *websocket.Conn) {
		defer func() {
			err := ws.Close()
			if err != nil {
				s.errCh <- err
			}
		}()

		var encodedToken string
		err := websocket.Message.Receive(ws, &encodedToken)
		if err != nil && err != io.EOF {
			log.Fatal(err)
		}

		token := auth.NewUserToken()
		err = token.FromString(encodedToken)
		if err != nil {
			log.Println(err)
			return
		}

		name, err := auth.CheckChatToken(db, token)
		if err != nil {
			log.Println(err)
			return
		}

		client := NewClient(ws, s, name)
		s.Add(client)
		client.Listen()
	}
	http.Handle(s.pattern, websocket.Handler(onConnected))
	log.Println("Created handler")

	for {
		select {

		// Add new a client
		case c := <-s.addCh:
			log.Println("Added new client")
			s.clients[c.name] = c
			log.Println("Now", len(s.clients), "clients connected.")
			s.sendContacts(c)

		// del a client
		case c := <-s.delCh:
			log.Println("Delete client")
			delete(s.clients, c.name)

		case msg := <-s.msgCh:
			log.Println("Message:", *msg)
			s.handleMessage(msg)

		case cmd := <-s.cmdCh:
			log.Println("Command:", *cmd)
			s.handleCommand(cmd)

		case err := <-s.errCh:
			log.Println("Error:", err.Error())

		case <-s.doneCh:
			return
		}
	}
}