main.go

package main

// Import dependencies.
import (
	// internals
	"database/sql"
	"html/template"
	"log"
	"net/http"
	"os"

	// externals
	_ "github.com/go-sql-driver/mysql"
	"github.com/gorilla/mux"
	"github.com/gorilla/websocket"
)

var db *sql.DB
var err error

var clients = make(map[*websocket.Conn]wsSession)

// Configure the upgrader
var upgrader = websocket.Upgrader{
	CheckOrigin: func(r *http.Request) bool {
		return true
	},
}

var templates = template.Must(template.ParseFiles("views/auth/login.html", "views/auth/signup.html", "views/index.html", "views/header.html", "views/footer.html", "views/communities.html", "views/post.html", "views/create_post.html", "views/create_comment.html", "views/comment_preview.html", "views/profile_sidebar.html", "views/user.html"))

// Main function.
func main() {

	// Connect to the database.
	db, err = sql.Open("mysql", "root:@unix(/tmp/mysql.sock)/slaj?parseTime=true&charset=utf8mb4,utf8")
	if err != nil {

		// we were unable to connect to the database
		log.Printf("[err]: unable to connect to the database...\n")
		log.Printf("       %v\n", err)
		os.Exit(1)

	}

	// ping the database once to ensure that it is available
	err = db.Ping()
	if err != nil {

		// we were unable to ping it
		log.Printf("[err]: unable to ping the database...\n")
		log.Printf("       %v\n", err)
		os.Exit(1)

	}

	// close the database connection after this function exits
	defer db.Close()

	// Initialize routes.
	r := mux.NewRouter()

	// Index route.
	r.HandleFunc("/", index)

	// Auth routes.
	r.HandleFunc("/signup", signup)
	r.HandleFunc("/login", login)
	r.HandleFunc("/act/logout", logout)

	// User routes.
	r.HandleFunc("/users/{username}", showUser)

	// Post routes.
	r.HandleFunc("/posts/{id:[0-9]+}", showPost)
	r.HandleFunc("/posts/{id:[0-9]+}/comments", createComment)

	// Yeah routes.
	r.HandleFunc("/posts/{id:[0-9]+}/yeah", createPostYeah)
	r.HandleFunc("/posts/{id:[0-9]+}/yeahu", deletePostYeah)
	r.HandleFunc("/comments/{id:[0-9]+}/yeah", createCommentYeah)
	r.HandleFunc("/comments/{id:[0-9]+}/yeahu", deleteCommentYeah)

	// Follow routes.
	r.HandleFunc("/users/{username}/follow", createFollow)
	r.HandleFunc("/users/{username}/unfollow", deleteFollow)

	// Community routes.
	r.HandleFunc("/communities/{id:[0-9]+}", showCommunity)
	r.HandleFunc("/communities/{id:[0-9]+}/posts", createPost).Methods("POST")

	// Upload image.
	r.HandleFunc("/upload", uploadImage).Methods("POST")

	// Test websocket route.
	r.HandleFunc("/ws", handleConnections)

	// Serve static assets.
	http.Handle("/assets/", http.StripPrefix("/assets/", http.FileServer(http.Dir("assets"))))

	// tell the http server to handle routing with the router we just made
	http.Handle("/", r)

	// tell the person who started this that we have started the server
	log.Printf("listening on :8080")

	// start the server
	http.ListenAndServe(":8080", nil)

}