Riyyi
5 months ago
commit
eb05e6d588
7 changed files with 292 additions and 0 deletions
@ -0,0 +1,7 @@
|
||||
module worklog |
||||
|
||||
go 1.22.5 |
||||
|
||||
require github.com/alexflint/go-arg v1.5.1 |
||||
|
||||
require github.com/alexflint/go-scalar v1.2.0 // indirect |
@ -0,0 +1,13 @@
|
||||
github.com/alexflint/go-arg v1.5.1 h1:nBuWUCpuRy0snAG+uIJ6N0UvYxpxA0/ghA/AaHxlT8Y= |
||||
github.com/alexflint/go-arg v1.5.1/go.mod h1:A7vTJzvjoaSTypg4biM5uYNTkJ27SkNTArtYXnlqVO8= |
||||
github.com/alexflint/go-scalar v1.2.0 h1:WR7JPKkeNpnYIOfHRa7ivM21aWAdHD0gEWHCx+WQBRw= |
||||
github.com/alexflint/go-scalar v1.2.0/go.mod h1:LoFvNMqS1CPrMVltza4LvnGKhaSpc3oyLEBUZVhhS2o= |
||||
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= |
||||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= |
||||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= |
||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= |
||||
github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs= |
||||
github.com/stretchr/testify v1.7.0 h1:nwc3DEeHmmLAfoZucVR881uASk0Mfjw8xYJ99tb5CcY= |
||||
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= |
||||
gopkg.in/yaml.v3 v3.0.0 h1:hjy8E9ON/egN1tAYqKb61G10WtihqetD4sz2H+8nIeA= |
||||
gopkg.in/yaml.v3 v3.0.0/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= |
@ -0,0 +1,45 @@
|
||||
// go mod init worklog
|
||||
// go build
|
||||
// go run .
|
||||
// go mod tidy
|
||||
|
||||
package main |
||||
|
||||
import "errors" |
||||
import "os" |
||||
|
||||
import "github.com/alexflint/go-arg" |
||||
|
||||
import "worklog/src" |
||||
|
||||
type Args struct { |
||||
Decl string `arg:"-d,--decl" help:"Generate travel declaration table" placeholder:"MONTH"` |
||||
Process bool `arg:"-p,--process" help:"Process specified file and call Jira API"` |
||||
File string `arg:"positional,required" help:"the worklog file to process"` |
||||
} |
||||
|
||||
func (Args) Description() string { |
||||
return "\nworklog - process a worklog file\n" |
||||
} |
||||
|
||||
func main() { |
||||
var args Args |
||||
parser := arg.MustParse(&args) |
||||
|
||||
_, err := os.Stat(args.File); |
||||
if errors.Is(err, os.ErrNotExist) || errors.Is(err, os.ErrPermission) { |
||||
parser.Fail("file was not readable: " + args.File) |
||||
} |
||||
|
||||
if args.Process { |
||||
var api src.Api = src.MakeApi() |
||||
var job = func(line string, line_number int) string { |
||||
return api.Process(line, line_number) |
||||
} |
||||
src.Parse(args.File, job) |
||||
} |
||||
|
||||
if args.Decl != "" { |
||||
// TODO: generate declaration table..
|
||||
} |
||||
} |
@ -0,0 +1,163 @@
|
||||
package src |
||||
|
||||
import ( |
||||
"errors" |
||||
"fmt" |
||||
"regexp" |
||||
"strconv" |
||||
"strings" |
||||
) |
||||
|
||||
type Date struct { |
||||
clock_in string |
||||
location string |
||||
clock_out string |
||||
|
||||
last_time string |
||||
last_item_id string |
||||
last_description string |
||||
} |
||||
|
||||
type Api struct { |
||||
clock_in *regexp.Regexp |
||||
task_line *regexp.Regexp |
||||
clock_out *regexp.Regexp |
||||
dates map[string]Date |
||||
} |
||||
|
||||
// Constructor
|
||||
func MakeApi() Api { |
||||
return Api{ |
||||
clock_in: compileRegex(`^\s*\| [0-9]{4}-[0-9]{2}-[0-9]{2} \|\s+IN\s+\|`), |
||||
task_line: compileRegex(`^\s*\| [0-9]{4}-[0-9]{2}-[0-9]{2} \| [0-9]{2}:[0-9]{2} \|`), |
||||
clock_out: compileRegex(`^\s*\| [0-9]{4}-[0-9]{2}-[0-9]{2} \|\s+OUT\s+\|`), |
||||
dates: make(map[string]Date), |
||||
} |
||||
} |
||||
|
||||
func (self *Api) Process(line string, line_number int) string { |
||||
var err error |
||||
if self.clock_in.MatchString(line) { |
||||
err = self.parseClockIn(line, line_number) |
||||
} else if self.task_line.MatchString(line) { |
||||
err = self.parseTask(line, line_number) |
||||
} else if self.clock_out.MatchString(line) { |
||||
err = self.parseClockOut(line, line_number) |
||||
} |
||||
assert(err) |
||||
|
||||
// fmt.Println(line)
|
||||
return line |
||||
} |
||||
|
||||
// -----------------------------------------
|
||||
|
||||
func compileRegex(pattern string) *regexp.Regexp { |
||||
regex, err := regexp.Compile(pattern) |
||||
assert(err) |
||||
|
||||
return regex |
||||
} |
||||
|
||||
func (self *Api) parseLine(line string, line_number int, size int) ([]string, error) { |
||||
var data []string = strings.Split(line, "|") |
||||
|
||||
if len(data) != size + 2 { |
||||
return nil, errors.New("malformed line " + strconv.Itoa(line_number) + "\n" + line) |
||||
} |
||||
|
||||
data = data[1:size + 1] |
||||
for i, value := range data { |
||||
data[i] = strings.TrimSpace(value) |
||||
|
||||
if len(data[i]) == 0 { |
||||
return nil, errors.New("malformed line " + strconv.Itoa(line_number) + "\n" + line) |
||||
} |
||||
} |
||||
|
||||
return data, nil |
||||
} |
||||
|
||||
func (self *Api) parseClockIn(line string, line_number int) error { |
||||
data, err := self.parseLine(line, line_number, 4) |
||||
if err != nil { return err } |
||||
|
||||
// Set clock_in, location
|
||||
var date Date = self.dates[data[0]] |
||||
date.clock_in = data[2] |
||||
date.location = data[3] |
||||
self.dates[data[0]] = date |
||||
|
||||
return nil |
||||
} |
||||
|
||||
func (self *Api) parseTask(line string, line_number int) error { |
||||
data, err := self.parseLine(line, line_number, 5) |
||||
if err != nil { return err } |
||||
|
||||
var date Date = self.dates[data[0]] |
||||
|
||||
if date.clock_in == "" { |
||||
return errors.New("no clock-in time found") |
||||
} |
||||
|
||||
// Call API for the previous task
|
||||
if date.last_time != "" && date.last_item_id != "" && date.last_description != "" { |
||||
err = self.callApi(data[0], date.last_time, data[1], date.last_item_id, date.last_description) |
||||
} |
||||
|
||||
if err != nil { return err } |
||||
|
||||
// Set last_time, last_item_id, description
|
||||
if data[3] == "X" { |
||||
date.last_time = data[1] |
||||
date.last_item_id = data[2] |
||||
date.last_description = data[4] |
||||
} else { // "V", task is already processed
|
||||
date.last_time = "" |
||||
date.last_item_id = "" |
||||
date.last_description = "" |
||||
} |
||||
self.dates[data[0]] = date |
||||
|
||||
return nil |
||||
} |
||||
|
||||
func (self *Api) parseClockOut(line string, line_number int) error { |
||||
data, err := self.parseLine(line, line_number, 3) |
||||
if err != nil { return err } |
||||
|
||||
// Set clock_out
|
||||
var date Date = self.dates[data[0]] |
||||
date.clock_out = data[2] |
||||
self.dates[data[0]] = date |
||||
|
||||
if date.last_time == "" || date.last_item_id == "" || date.last_description == "" { |
||||
return errors.New("no previous task to use clock-out on") |
||||
} |
||||
|
||||
// Call API for last task of the day
|
||||
self.callApi(data[0], date.last_time, date.clock_out, date.last_item_id, date.last_description) |
||||
|
||||
return nil |
||||
} |
||||
|
||||
func (self *Api) callApi(date string, from_time string, to_time string, item_id string, description string) error { |
||||
fmt.Println("API |" + date + "|" + from_time + "|" + to_time + "|" + item_id + "|" + description) |
||||
|
||||
// parse line
|
||||
// call API
|
||||
// error checking
|
||||
|
||||
return nil |
||||
} |
||||
|
||||
// Example worklog:
|
||||
|
||||
// | 2024-07-06 | IN | 08:30 | Office |
|
||||
|
||||
// | 2024-07-06 | 09:00 | T1-123 | V | I did nothing! |
|
||||
// | 2024-07-06 | 09:30 | T1-456 | X | Blabla |
|
||||
// | 2024-07-06 | 11:00 | T1-789 | X | - |
|
||||
|
||||
// | 2024-07-06 | OUT | 13:00 |
|
@ -0,0 +1,13 @@
|
||||
package src |
||||
|
||||
func assert(err error) { |
||||
if err != nil { |
||||
panic(err) |
||||
} |
||||
} |
||||
|
||||
func verify(condition bool, message string) { |
||||
if !condition { |
||||
panic(message) |
||||
} |
||||
} |
@ -0,0 +1,46 @@
|
||||
package src |
||||
|
||||
import "bufio" |
||||
import "os" |
||||
|
||||
func Parse(path string, job func(line string, line_number int) string) { |
||||
// Input file
|
||||
file, err := os.Open(path) |
||||
assert(err) |
||||
defer file.Close() |
||||
var scanner *bufio.Scanner = bufio.NewScanner(file) |
||||
|
||||
// Output file
|
||||
output_file, err := os.Create(path + ".tmp") |
||||
assert(err) |
||||
defer output_file.Close() |
||||
var writer *bufio.Writer = bufio.NewWriter(output_file) |
||||
defer writer.Flush() |
||||
|
||||
var line string |
||||
var line_number int = 1 |
||||
for scanner.Scan() { |
||||
line = scanner.Text() |
||||
line = job(line, line_number) |
||||
line_number++ |
||||
|
||||
// Write line to output_file
|
||||
_, err := writer.WriteString(line + "\n") |
||||
assert(err) |
||||
} |
||||
|
||||
// Detect table if it was at the end of the file
|
||||
job("", line_number) |
||||
|
||||
err = scanner.Err() |
||||
assert(err) |
||||
|
||||
// move file
|
||||
} |
||||
|
||||
// - [v] while looping, start writing a new file, .tmp, Q: write per line or per chunk? how big are the chunks?
|
||||
// - [v] mark table start with processed mark
|
||||
// - [ ] on table end, call into REST API
|
||||
// - [ ] if true, continue
|
||||
// - [ ] if false, delete .tmp file, panic
|
||||
// - [ ] if no errors, overwrite file with .tmp file
|
Loading…
Reference in new issue