shrty/main.go

package main

import (
	"net/http"

	"github.com/kreativmonkey/shrt/shrty"
	"log"
	"fmt"
	"path"
)

const version = "0.07"

var (
	short *shrty.DB

	// Global configuration Variable
	config = ReadConfig()
)

func init() {
	var err error
	short, err = shrty.Open(config.DB)
	if err != nil {
		panic(err)
	}
}

type shrt struct {
	Token string `bson:"token" json:"token"`
	URL string `bson:"url" json:"url"`
	Date string `bson:"url_ending" json:"url_ending"`
	Count int64 `bson:"click" json:"click"`
}

func main() {
	router := NewRouter()

	s := http.StripPrefix("/css/", http.FileServer(http.Dir(path.Join(config.RootPath, "/template/css/"))))
	router.PathPrefix("/css/").Handler(s)
	s = http.StripPrefix("/img/", http.FileServer(http.Dir(path.Join(config.RootPath, "/template/img/"))))
	router.PathPrefix("/img/").Handler(s)

	/*
	http.Handle("/css/", http.StripPrefix("/css/", http.FileServer(http.Dir("template/css"))))
	http.Handle("/img/", http.StripPrefix("/img/", http.FileServer(http.Dir("template/img"))))
	*/
	fmt.Printf("Shrty %s is listen on Port: %s \n", version, config.Port)
	fmt.Printf("For the API use following key: %s \n", config.APIkey)
	http.Handle("/", router)

	log.Fatal(http.ListenAndServe(":"+ config.Port, router))
}