package main import ( "net/http" shrty "github.com/kreativmonkey/shrt/shrty" "log" "fmt" ) const version = "0.03" var ( short *shrty.Storage // Global configuration Variable config = ReadConfig() ) func init() { var err error short, err = shrty.Open(config.DB) if err != nil { panic(err) } /* viper.SetDefault("ContentDir", "content") viper.SetDefault("Token", map[string]string{"url": "url", "token": "token"}) viper.SetConfigName("config") // name of config file (without expression) viper.AddConfigPath("/etc/appname/") // path to look for the config file in viper.AddConfigPath("$HOME/.appname") // call multiple times to add many search paths viper.AddConfigPath(".") // optionally look for config in the working directory err = viper.ReadInConfig() // Find and read the config file if err != nil { // Handle errors reading the config file panic(fmt.Errorf("Fatal error config file: %s \n", 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("./template/css/"))) router.PathPrefix("/css/").Handler(s) s = http.StripPrefix("/img/", http.FileServer(http.Dir("./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", version, config.APIkey) http.Handle("/", router) log.Fatal(http.ListenAndServe(":"+ config.Port, router)) }