Update: code cleanup
This commit is contained in:
parent
7e356b34ec
commit
221bd4d4b2
5 changed files with 171 additions and 143 deletions
|
@ -1,4 +1,4 @@
|
||||||
package models
|
package handler
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
|
|
164
handler/template.go
Normal file
164
handler/template.go
Normal file
|
@ -0,0 +1,164 @@
|
||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"html/template"
|
||||||
|
"math/rand"
|
||||||
|
"pixivfe/configs"
|
||||||
|
"regexp"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
func GetRandomColor() string {
|
||||||
|
// Some color shade I generated
|
||||||
|
colors := []string{
|
||||||
|
// Green
|
||||||
|
"#3cc223",
|
||||||
|
"#55dc3d",
|
||||||
|
"#7be468",
|
||||||
|
}
|
||||||
|
|
||||||
|
// Randomly choose one and return
|
||||||
|
return colors[rand.Intn(len(colors))]
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetTemplateFunctions() template.FuncMap {
|
||||||
|
return template.FuncMap{
|
||||||
|
"inc": func(n int) int {
|
||||||
|
// For rankings to increment a number by 1
|
||||||
|
return n + 1
|
||||||
|
},
|
||||||
|
"add": func(a int, b int) int {
|
||||||
|
return a + b
|
||||||
|
},
|
||||||
|
|
||||||
|
"dec": func(n int) int {
|
||||||
|
return n - 1
|
||||||
|
},
|
||||||
|
|
||||||
|
"toInt": func(s string) int {
|
||||||
|
n, _ := strconv.Atoi(s)
|
||||||
|
return n
|
||||||
|
},
|
||||||
|
|
||||||
|
"proxyImage": func(url string) string {
|
||||||
|
if strings.Contains(url, "s.pximg.net") {
|
||||||
|
// This subdomain didn't get proxied
|
||||||
|
return url
|
||||||
|
}
|
||||||
|
|
||||||
|
regex := regexp.MustCompile(`.*?pximg\.net`)
|
||||||
|
proxy := "https://" + configs.ProxyServer
|
||||||
|
|
||||||
|
return regex.ReplaceAllString(url, proxy)
|
||||||
|
},
|
||||||
|
"parseEmojis": func(s string) template.HTML {
|
||||||
|
regex := regexp.MustCompile(`\(([^)]+)\)`)
|
||||||
|
|
||||||
|
parsedString := regex.ReplaceAllStringFunc(s, func(s string) string {
|
||||||
|
s = s[1 : len(s)-1] // Get the string inside
|
||||||
|
var id string
|
||||||
|
|
||||||
|
switch s {
|
||||||
|
case "normal":
|
||||||
|
id = "101"
|
||||||
|
case "surprise":
|
||||||
|
id = "102"
|
||||||
|
case "serious":
|
||||||
|
id = "103"
|
||||||
|
case "heaven":
|
||||||
|
id = "104"
|
||||||
|
case "happy":
|
||||||
|
id = "105"
|
||||||
|
case "excited":
|
||||||
|
id = "106"
|
||||||
|
case "sing":
|
||||||
|
id = "107"
|
||||||
|
case "cry":
|
||||||
|
id = "108"
|
||||||
|
case "normal2":
|
||||||
|
id = "201"
|
||||||
|
case "shame2":
|
||||||
|
id = "202"
|
||||||
|
case "love2":
|
||||||
|
id = "203"
|
||||||
|
case "interesting2":
|
||||||
|
id = "204"
|
||||||
|
case "blush2":
|
||||||
|
id = "205"
|
||||||
|
case "fire2":
|
||||||
|
id = "206"
|
||||||
|
case "angry2":
|
||||||
|
id = "207"
|
||||||
|
case "shine2":
|
||||||
|
id = "208"
|
||||||
|
case "panic2":
|
||||||
|
id = "209"
|
||||||
|
case "normal3":
|
||||||
|
id = "301"
|
||||||
|
case "satisfaction3":
|
||||||
|
id = "302"
|
||||||
|
case "surprise3":
|
||||||
|
id = "303"
|
||||||
|
case "smile3":
|
||||||
|
id = "304"
|
||||||
|
case "shock3":
|
||||||
|
id = "305"
|
||||||
|
case "gaze3":
|
||||||
|
id = "306"
|
||||||
|
case "wink3":
|
||||||
|
id = "307"
|
||||||
|
case "happy3":
|
||||||
|
id = "308"
|
||||||
|
case "excited3":
|
||||||
|
id = "309"
|
||||||
|
case "love3":
|
||||||
|
id = "310"
|
||||||
|
case "normal4":
|
||||||
|
id = "401"
|
||||||
|
case "surprise4":
|
||||||
|
id = "402"
|
||||||
|
case "serious4":
|
||||||
|
id = "403"
|
||||||
|
case "love4":
|
||||||
|
id = "404"
|
||||||
|
case "shine4":
|
||||||
|
id = "405"
|
||||||
|
case "sweat4":
|
||||||
|
id = "406"
|
||||||
|
case "shame4":
|
||||||
|
id = "407"
|
||||||
|
case "sleep4":
|
||||||
|
id = "408"
|
||||||
|
case "heart":
|
||||||
|
id = "501"
|
||||||
|
case "teardrop":
|
||||||
|
id = "502"
|
||||||
|
case "star":
|
||||||
|
id = "503"
|
||||||
|
}
|
||||||
|
return fmt.Sprintf(`<img src="https://s.pximg.net/common/images/emoji/%s.png" alt="(%s)" class="emoji" />`, id, s)
|
||||||
|
})
|
||||||
|
return template.HTML(parsedString)
|
||||||
|
},
|
||||||
|
|
||||||
|
"randomColor": func() string {
|
||||||
|
return GetRandomColor()
|
||||||
|
},
|
||||||
|
|
||||||
|
"isEmpty": func(s string) bool {
|
||||||
|
return len(s) < 1
|
||||||
|
},
|
||||||
|
|
||||||
|
"isEmphasize": func(s string) bool {
|
||||||
|
switch s {
|
||||||
|
case
|
||||||
|
"R-18",
|
||||||
|
"R-18G":
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
140
main.go
140
main.go
|
@ -1,13 +1,9 @@
|
||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
|
||||||
"html/template"
|
|
||||||
"pixivfe/configs"
|
"pixivfe/configs"
|
||||||
|
"pixivfe/handler"
|
||||||
"pixivfe/views"
|
"pixivfe/views"
|
||||||
"regexp"
|
|
||||||
"strconv"
|
|
||||||
"strings"
|
|
||||||
|
|
||||||
"github.com/gin-gonic/gin"
|
"github.com/gin-gonic/gin"
|
||||||
)
|
)
|
||||||
|
@ -15,139 +11,7 @@ import (
|
||||||
func setupRouter() *gin.Engine {
|
func setupRouter() *gin.Engine {
|
||||||
server := gin.Default()
|
server := gin.Default()
|
||||||
|
|
||||||
server.SetFuncMap(template.FuncMap{
|
server.SetFuncMap(handler.GetTemplateFunctions())
|
||||||
"inc": func(n int) int {
|
|
||||||
// For rankings to increment a number by 1
|
|
||||||
return n + 1
|
|
||||||
},
|
|
||||||
"add": func(a int, b int) int {
|
|
||||||
return a + b
|
|
||||||
},
|
|
||||||
|
|
||||||
"dec": func(n int) int {
|
|
||||||
return n - 1
|
|
||||||
},
|
|
||||||
|
|
||||||
"toInt": func(s string) int {
|
|
||||||
n, _ := strconv.Atoi(s)
|
|
||||||
return n
|
|
||||||
},
|
|
||||||
|
|
||||||
"proxyImage": func(url string) string {
|
|
||||||
if strings.Contains(url, "s.pximg.net") {
|
|
||||||
// This subdomain didn't get proxied
|
|
||||||
return url
|
|
||||||
}
|
|
||||||
|
|
||||||
regex := regexp.MustCompile(`.*?pximg\.net`)
|
|
||||||
proxy := "https://" + configs.ProxyServer
|
|
||||||
|
|
||||||
return regex.ReplaceAllString(url, proxy)
|
|
||||||
},
|
|
||||||
"parseEmojis": func(s string) template.HTML {
|
|
||||||
regex := regexp.MustCompile(`\(([^)]+)\)`)
|
|
||||||
|
|
||||||
parsedString := regex.ReplaceAllStringFunc(s, func(s string) string {
|
|
||||||
s = s[1 : len(s)-1] // Get the string inside
|
|
||||||
var id string
|
|
||||||
|
|
||||||
switch s {
|
|
||||||
case "normal":
|
|
||||||
id = "101"
|
|
||||||
case "surprise":
|
|
||||||
id = "102"
|
|
||||||
case "serious":
|
|
||||||
id = "103"
|
|
||||||
case "heaven":
|
|
||||||
id = "104"
|
|
||||||
case "happy":
|
|
||||||
id = "105"
|
|
||||||
case "excited":
|
|
||||||
id = "106"
|
|
||||||
case "sing":
|
|
||||||
id = "107"
|
|
||||||
case "cry":
|
|
||||||
id = "108"
|
|
||||||
case "normal2":
|
|
||||||
id = "201"
|
|
||||||
case "shame2":
|
|
||||||
id = "202"
|
|
||||||
case "love2":
|
|
||||||
id = "203"
|
|
||||||
case "interesting2":
|
|
||||||
id = "204"
|
|
||||||
case "blush2":
|
|
||||||
id = "205"
|
|
||||||
case "fire2":
|
|
||||||
id = "206"
|
|
||||||
case "angry2":
|
|
||||||
id = "207"
|
|
||||||
case "shine2":
|
|
||||||
id = "208"
|
|
||||||
case "panic2":
|
|
||||||
id = "209"
|
|
||||||
case "normal3":
|
|
||||||
id = "301"
|
|
||||||
case "satisfaction3":
|
|
||||||
id = "302"
|
|
||||||
case "surprise3":
|
|
||||||
id = "303"
|
|
||||||
case "smile3":
|
|
||||||
id = "304"
|
|
||||||
case "shock3":
|
|
||||||
id = "305"
|
|
||||||
case "gaze3":
|
|
||||||
id = "306"
|
|
||||||
case "wink3":
|
|
||||||
id = "307"
|
|
||||||
case "happy3":
|
|
||||||
id = "308"
|
|
||||||
case "excited3":
|
|
||||||
id = "309"
|
|
||||||
case "love3":
|
|
||||||
id = "310"
|
|
||||||
case "normal4":
|
|
||||||
id = "401"
|
|
||||||
case "surprise4":
|
|
||||||
id = "402"
|
|
||||||
case "serious4":
|
|
||||||
id = "403"
|
|
||||||
case "love4":
|
|
||||||
id = "404"
|
|
||||||
case "shine4":
|
|
||||||
id = "405"
|
|
||||||
case "sweat4":
|
|
||||||
id = "406"
|
|
||||||
case "shame4":
|
|
||||||
id = "407"
|
|
||||||
case "sleep4":
|
|
||||||
id = "408"
|
|
||||||
case "heart":
|
|
||||||
id = "501"
|
|
||||||
case "teardrop":
|
|
||||||
id = "502"
|
|
||||||
case "star":
|
|
||||||
id = "503"
|
|
||||||
}
|
|
||||||
return fmt.Sprintf(`<img src="https://s.pximg.net/common/images/emoji/%s.png" alt="(%s)" class="emoji" />`, id, s)
|
|
||||||
})
|
|
||||||
return template.HTML(parsedString)
|
|
||||||
},
|
|
||||||
|
|
||||||
"isEmpty": func(s string) bool {
|
|
||||||
return len(s) < 1
|
|
||||||
},
|
|
||||||
|
|
||||||
"isEmphasize": func(s string) bool {
|
|
||||||
switch s {
|
|
||||||
case
|
|
||||||
"R-18",
|
|
||||||
"R-18G":
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
},
|
|
||||||
})
|
|
||||||
|
|
||||||
// Static files
|
// Static files
|
||||||
server.StaticFile("/favicon.ico", "./template/favicon.ico")
|
server.StaticFile("/favicon.ico", "./template/favicon.ico")
|
||||||
|
|
|
@ -136,7 +136,7 @@ type User struct {
|
||||||
BackgroundImage string `json:"background"`
|
BackgroundImage string `json:"background"`
|
||||||
Following int `json:"following"`
|
Following int `json:"following"`
|
||||||
MyPixiv int `json:"mypixivCount"`
|
MyPixiv int `json:"mypixivCount"`
|
||||||
Comment string `json:"comment"`
|
Comment template.HTML `json:"commentHtml"`
|
||||||
Artworks []IllustShort `json:"artworks"`
|
Artworks []IllustShort `json:"artworks"`
|
||||||
ArtworksCount int
|
ArtworksCount int
|
||||||
}
|
}
|
||||||
|
|
|
@ -11,7 +11,7 @@ import (
|
||||||
"github.com/gin-gonic/gin"
|
"github.com/gin-gonic/gin"
|
||||||
)
|
)
|
||||||
|
|
||||||
var PC *models.PixivClient
|
var PC *handler.PixivClient
|
||||||
|
|
||||||
func artwork_page(c *gin.Context) {
|
func artwork_page(c *gin.Context) {
|
||||||
id := c.Param("id")
|
id := c.Param("id")
|
||||||
|
@ -241,14 +241,14 @@ func not_found_page(c *gin.Context) {
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewPixivClient(timeout int) *models.PixivClient {
|
func NewPixivClient(timeout int) *handler.PixivClient {
|
||||||
transport := &http.Transport{Proxy: http.ProxyFromEnvironment}
|
transport := &http.Transport{Proxy: http.ProxyFromEnvironment}
|
||||||
client := &http.Client{
|
client := &http.Client{
|
||||||
Timeout: time.Duration(timeout) * time.Millisecond,
|
Timeout: time.Duration(timeout) * time.Millisecond,
|
||||||
Transport: transport,
|
Transport: transport,
|
||||||
}
|
}
|
||||||
|
|
||||||
pc := &models.PixivClient{
|
pc := &handler.PixivClient{
|
||||||
Client: client,
|
Client: client,
|
||||||
Header: make(map[string]string),
|
Header: make(map[string]string),
|
||||||
Cookie: make(map[string]string),
|
Cookie: make(map[string]string),
|
||||||
|
|
Loading…
Reference in a new issue