144 lines
3.3 KiB
Go
144 lines
3.3 KiB
Go
package common
|
|
|
|
import (
|
|
"database/sql"
|
|
"log"
|
|
|
|
"../query_gen/lib"
|
|
)
|
|
|
|
// nolint I don't want to write comments for each of these o.o
|
|
const Hour int = 60 * 60
|
|
const Day int = Hour * 24
|
|
const Week int = Day * 7
|
|
const Month int = Day * 30
|
|
const Year int = Day * 365
|
|
const Kilobyte int = 1024
|
|
const Megabyte int = Kilobyte * 1024
|
|
const Gigabyte int = Megabyte * 1024
|
|
const Terabyte int = Gigabyte * 1024
|
|
const Petabyte int = Terabyte * 1024
|
|
|
|
const SaltLength int = 32
|
|
const SessionLength int = 80
|
|
|
|
var TmplPtrMap = make(map[string]interface{})
|
|
|
|
// ErrNoRows is an alias of sql.ErrNoRows, just in case we end up with non-database/sql datastores
|
|
var ErrNoRows = sql.ErrNoRows
|
|
|
|
// ? - Make this more customisable?
|
|
var ExternalSites = map[string]string{
|
|
"YT": "https://www.youtube.com/",
|
|
}
|
|
|
|
type StringList []string
|
|
|
|
// ? - Should we allow users to upload .php or .go files? It could cause security issues. We could store them with a mangled extension to render them inert
|
|
// TODO: Let admins manage this from the Control Panel
|
|
var AllowedFileExts = StringList{
|
|
"png", "jpg", "jpeg", "svg", "bmp", "gif", "tif", "webp", "apng", // images
|
|
|
|
"txt", "xml", "json", "yaml", "toml", "ini", "md", "html", "rtf", "js", "py", "rb", "css", "scss", "less", "eqcss", "pcss", "java", "ts", "cs", "c", "cc", "cpp", "cxx", "C", "c++", "h", "hh", "hpp", "hxx", "h++", "rs", "rlib", "htaccess", "gitignore", // text
|
|
|
|
"mp3", "mp4", "avi", "wmv", "webm", // video
|
|
|
|
"otf", "woff2", "woff", "ttf", "eot", // fonts
|
|
}
|
|
var ImageFileExts = StringList{
|
|
"png", "jpg", "jpeg", "svg", "bmp", "gif", "tif", "webp", "apng",
|
|
}
|
|
var ArchiveFileExts = StringList{
|
|
"bz2", "zip", "gz", "7z", "tar", "cab",
|
|
}
|
|
var ExecutableFileExts = StringList{
|
|
"exe", "jar", "phar", "shar", "iso",
|
|
}
|
|
|
|
// TODO: Write a test for this
|
|
func (slice StringList) Contains(needle string) bool {
|
|
for _, item := range slice {
|
|
if item == needle {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
type dbInits []func(acc *qgen.Accumulator) error
|
|
|
|
var DbInits dbInits
|
|
|
|
func (inits dbInits) Run() error {
|
|
for _, init := range inits {
|
|
acc := qgen.Builder.Accumulator()
|
|
err := init(acc)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (inits dbInits) Add(init ...func(acc *qgen.Accumulator) error) {
|
|
DbInits = dbInits(append(DbInits, init...))
|
|
}
|
|
|
|
func debugDetail(args ...interface{}) {
|
|
if Dev.SuperDebug {
|
|
log.Print(args...)
|
|
}
|
|
}
|
|
|
|
func debugDetailf(str string, args ...interface{}) {
|
|
if Dev.SuperDebug {
|
|
log.Printf(str, args...)
|
|
}
|
|
}
|
|
|
|
func debugLog(args ...interface{}) {
|
|
if Dev.DebugMode {
|
|
log.Print(args...)
|
|
}
|
|
}
|
|
|
|
func debugLogf(str string, args ...interface{}) {
|
|
if Dev.DebugMode {
|
|
log.Printf(str, args...)
|
|
}
|
|
}
|
|
|
|
// TODO: Make a neater API for this
|
|
var routeMapEnum map[string]int
|
|
var reverseRouteMapEnum map[int]string
|
|
|
|
func SetRouteMapEnum(rme map[string]int) {
|
|
routeMapEnum = rme
|
|
}
|
|
|
|
func SetReverseRouteMapEnum(rrme map[int]string) {
|
|
reverseRouteMapEnum = rrme
|
|
}
|
|
|
|
var agentMapEnum map[string]int
|
|
var reverseAgentMapEnum map[int]string
|
|
|
|
func SetAgentMapEnum(ame map[string]int) {
|
|
agentMapEnum = ame
|
|
}
|
|
|
|
func SetReverseAgentMapEnum(rame map[int]string) {
|
|
reverseAgentMapEnum = rame
|
|
}
|
|
|
|
var osMapEnum map[string]int
|
|
var reverseOSMapEnum map[int]string
|
|
|
|
func SetOSMapEnum(osme map[string]int) {
|
|
osMapEnum = osme
|
|
}
|
|
|
|
func SetReverseOSMapEnum(rosme map[int]string) {
|
|
reverseOSMapEnum = rosme
|
|
}
|