1
0
mirror of https://github.com/schollz/cowyo.git synced 2023-08-10 21:13:00 +03:00

Allow certain IPs for certain pages

Former-commit-id: ff23e987f818f840742da0dcca1d9148b54329b9 [formerly 09e7073733503d998128b78be261d48cd91eee68] [formerly 4fff11f4bfd132937354ea5ffd06460de5c6fbc2 [formerly 467eb4d5b6]]
Former-commit-id: 14ceb7f40eb4ed9bdd20a6858ee5edcc23fc3ff8 [formerly 520934eeca459f2d94e7fa8e68ab83904549cb24]
Former-commit-id: c1adb3e8ea81e13d1c0742f493a4517765846191
This commit is contained in:
Zack Scholl 2016-02-07 00:24:17 -05:00
parent 70a8c21821
commit d55bc9e0f9
2 changed files with 12 additions and 2 deletions

12
main.go
View File

@ -9,6 +9,7 @@ import (
"os" "os"
"path" "path"
"runtime" "runtime"
"strings"
"time" "time"
"github.com/boltdb/bolt" "github.com/boltdb/bolt"
@ -21,6 +22,11 @@ import (
var db *bolt.DB var db *bolt.DB
var open bool var open bool
var ExternalIP string var ExternalIP string
var AllowedIPs string
func init() {
AllowedIPs = "192.168.1.13,192.168.1.12"
}
func Open() error { func Open() error {
var err error var err error
@ -128,14 +134,18 @@ func main() {
}) })
r.GET("/:title", func(c *gin.Context) { r.GET("/:title", func(c *gin.Context) {
title := c.Param("title") title := c.Param("title")
fmt.Println("------")
fmt.Println("[" + c.ClientIP() + "]")
fmt.Println("------")
if title == "ws" { if title == "ws" {
wshandler(c.Writer, c.Request) wshandler(c.Writer, c.Request)
} else if title == "about" && strings.Contains(AllowedIPs, c.ClientIP()) != true {
c.Redirect(302, "/about/view")
} else { } else {
c.HTML(http.StatusOK, "index.tmpl", gin.H{ c.HTML(http.StatusOK, "index.tmpl", gin.H{
"Title": title, "Title": title,
"ExternalIP": ExternalIP, "ExternalIP": ExternalIP,
}) })
} }
}) })
r.GET("/:title/*option", func(c *gin.Context) { r.GET("/:title/*option", func(c *gin.Context) {

View File

@ -125,7 +125,7 @@
border: none; border: none;
overflow: auto; overflow: auto;
outline: none; outline: none;
font-size: x-large; font-size: large;
-webkit-box-shadow: none; -webkit-box-shadow: none;
-moz-box-shadow: none; -moz-box-shadow: none;
box-shadow: none; box-shadow: none;