Promoting branch to master

Co-authored-by: grumbulon <grumbulon@dismail.de>
Reviewed-on: #1
This commit is contained in:
grumbulon 2022-06-26 04:07:58 +00:00
parent 182ec20469
commit 8c61f9f409
4 changed files with 162 additions and 1 deletions

4
awl.go
View file

@ -3,9 +3,13 @@ package main
import (
"fmt"
"os"
"git.froth.zone/sam/awl/logawl"
)
func main() {
logger := logawl.New()
logger.Println("Test PLEASE WORK")
app := prepareCLI()
err := app.Run(os.Args)
if err != nil {

2
cli.go
View file

@ -186,7 +186,7 @@ func parseArgs(args []string) (util.Answers, error) {
// TODO: Actually find where windows stuffs its dns resolvers
resp.Answers.Server = "8.8.4.4"
} else {
resp.Answers.Server = resolv.Servers[rand.Intn(len(resolv.Servers)-1)]
resp.Answers.Server = resolv.Servers[rand.Intn(len(resolv.Servers))]
}
}

44
logawl/logawl.go Normal file
View file

@ -0,0 +1,44 @@
package logawl
import "fmt"
type Lvl int
type Level struct {
lvl int
Prefix string
}
func (l *Level) GetLevel(i int) (string, error) {
switch i {
case int(FatalLevel):
return "FATAL ", nil
case int(ErrorLevel):
return "ERROR ", nil
case int(InfoLevel):
return "INFO ", nil
case int(DebugLevel):
return "DEBUG ", nil
}
return "", fmt.Errorf("Invalid log level choice")
}
var AllLevels = []Lvl{
FatalLevel,
ErrorLevel,
InfoLevel,
DebugLevel,
}
const (
// Fatal logs (will call exit(1))
FatalLevel Lvl = iota
// Error logs
ErrorLevel
// What is going on level
InfoLevel
// Verbose log level.
DebugLevel
)

113
logawl/logger.go Normal file
View file

@ -0,0 +1,113 @@
package logawl
import (
"fmt"
"io"
"os"
"sync"
"sync/atomic"
"time"
)
type Logger struct {
Mu sync.Mutex
Level Lvl
Prefix string
Out io.Writer
buf []byte
isDiscard int32
}
// Calling New instantiates Logger
//
// Level can be changed to one of the other log levels (FatalLevel, ErrorLevel, InfoLevel, DebugLevel)
func New() *Logger {
return &Logger{
Out: os.Stderr,
//TODO: Make cli.go change this value when calling logawl.New()
Level: InfoLevel, //Default value is InfoLevel
}
}
// Takes any and prints it out to Logger -> Out (io.Writer (default is std.Err))
func (l *Logger) Println(v ...any) {
if atomic.LoadInt32(&l.isDiscard) != 0 {
return
}
switch l.Level { //Goes through log levels and does stuff based on them (Fatal os.Exit...etc)
case 0:
l.Printer(0, fmt.Sprintln(v...)) //Fatal level
os.Exit(1)
case 1:
l.Printer(1, fmt.Sprintln(v...)) //Error level
case 2:
l.Printer(2, fmt.Sprintln(v...)) //Info level
case 3:
l.Printer(3, fmt.Sprintln(v...)) //Debug level
}
}
// Formats the log header as such <LogLevel> YYYY/MM/DD HH:MM:SS (local time) <the message to log>
func (l *Logger) formatHeader(buf *[]byte, t time.Time, level int, line int) {
var lv Level
if logLvl, err := lv.GetLevel(level); err != nil {
fmt.Printf("Invalid log level %s. Error: %v", logLvl, err)
os.Exit(1)
} else {
*buf = append(*buf, logLvl...)
year, month, day := t.Date()
formatter(buf, year, 4)
*buf = append(*buf, '/')
formatter(buf, int(month), 2)
*buf = append(*buf, '/')
formatter(buf, day, 2)
*buf = append(*buf, ' ')
hour, min, sec := t.Clock()
formatter(buf, hour, 2)
*buf = append(*buf, ':')
formatter(buf, min, 2)
*buf = append(*buf, ':')
formatter(buf, sec, 2)
*buf = append(*buf, ' ')
}
}
// Printer prints the formatted message directly to stdErr
func (l *Logger) Printer(level int, s string) error {
now := time.Now()
var line int
l.Mu.Lock()
defer l.Mu.Unlock()
l.buf = l.buf[:0]
l.formatHeader(&l.buf, now, level, line)
l.buf = append(l.buf, s...)
if len(s) == 0 || s[len(s)-1] != '\n' {
l.buf = append(l.buf, '\n')
}
_, err := l.Out.Write(l.buf)
return err
}
// Some line formatting stuff from Golang log stdlib file
//
// Please view https://cs.opensource.google/go/go/+/refs/tags/go1.18.3:src/log/log.go;drc=41e1d9075e428c2fc32d966b3752a3029b620e2c;l=96
//
// Cheap integer to fixed-width decimal ASCII. Give a negative width to avoid zero-padding.
func formatter(buf *[]byte, i int, wid int) {
// Assemble decimal in reverse order.
var b [20]byte
bp := len(b) - 1
for i >= 10 || wid > 1 {
wid--
q := i / 10
b[bp] = byte('0' + i - q*10)
bp--
i = q
}
// i < 10
b[bp] = byte('0' + i)
*buf = append(*buf, b[bp:]...)
}