mirror of
https://github.com/netfun2000/rttys_zhaojh329.git
synced 2026-02-27 09:53:24 +08:00
Jianhui Zhao
111 lines
3.1 KiB
Go
Executable File
111 lines
3.1 KiB
Go
Executable File
/*
|
|
* Copyright (C) 2017 Jianhui Zhao <jianhuizhao329@gmail.com>
|
|
*
|
|
* This program is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301
|
|
* USA
|
|
*/
|
|
|
|
package main
|
|
|
|
import (
|
|
"flag"
|
|
"log"
|
|
"time"
|
|
"strconv"
|
|
"math/rand"
|
|
"net/http"
|
|
"crypto/md5"
|
|
"encoding/hex"
|
|
"encoding/json"
|
|
_ "github.com/zhaojh329/rttys/statik"
|
|
"github.com/rakyll/statik/fs"
|
|
"github.com/zhaojh329/rttys/version"
|
|
)
|
|
|
|
type DeviceInfo struct {
|
|
ID string `json:"id"`
|
|
Uptime int64 `json:"uptime"`
|
|
Description string `json:"description"`
|
|
}
|
|
|
|
func generateSID(devid string) string {
|
|
md5Ctx := md5.New()
|
|
md5Ctx.Write([]byte(devid + strconv.FormatFloat(rand.Float64(), 'e', 6, 32)))
|
|
cipherStr := md5Ctx.Sum(nil)
|
|
return hex.EncodeToString(cipherStr)
|
|
}
|
|
|
|
func main() {
|
|
port := flag.Int("port", 5912, "http service port")
|
|
cert := flag.String("cert", "", "certFile Path")
|
|
key := flag.String("key", "", "keyFile Path")
|
|
|
|
flag.Parse()
|
|
|
|
rand.Seed(time.Now().Unix())
|
|
|
|
log.Println("BuildDate:", version.BuildDate)
|
|
log.Println("Gitref:", version.Gitref)
|
|
|
|
br := newBridge()
|
|
go br.run()
|
|
|
|
statikFS, err := fs.New()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
return
|
|
}
|
|
|
|
staticfs := http.FileServer(statikFS)
|
|
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
|
if r.URL.Path == "/" {
|
|
if t := r.URL.Query().Get("t"); t == "" {
|
|
http.Redirect(w, r, "/?t=" + strconv.FormatInt(time.Now().Unix(), 10), http.StatusFound)
|
|
return
|
|
}
|
|
}
|
|
staticfs.ServeHTTP(w, r)
|
|
})
|
|
|
|
http.HandleFunc("/devs", func(w http.ResponseWriter, r *http.Request) {
|
|
devs := make([]DeviceInfo, 0)
|
|
for _, c := range br.devices {
|
|
if c.isDev {
|
|
d := DeviceInfo{c.devid, time.Now().Unix() - c.timestamp, c.description}
|
|
devs = append(devs, d)
|
|
}
|
|
}
|
|
|
|
js, _ := json.Marshal(devs)
|
|
w.Write(js)
|
|
})
|
|
|
|
http.HandleFunc("/cmd", func(w http.ResponseWriter, r *http.Request) {
|
|
serveCmd(br, w, r)
|
|
})
|
|
|
|
http.HandleFunc("/ws", func(w http.ResponseWriter, r *http.Request) {
|
|
serveWs(br, w, r)
|
|
})
|
|
|
|
if *cert != "" && *key != "" {
|
|
log.Println("Listen on: ", *port, "SSL on")
|
|
log.Fatal(http.ListenAndServeTLS(":" + strconv.Itoa(*port), *cert, *key, nil))
|
|
} else {
|
|
log.Println("Listen on: ", *port, "SSL off")
|
|
log.Fatal(http.ListenAndServe(":" + strconv.Itoa(*port), nil))
|
|
}
|
|
}
|