This commit is contained in:
suguo.yao 2022-09-01 06:43:03 +08:00
parent b695a08544
commit 768b22ab03
3 changed files with 70 additions and 1 deletions

3
.gitignore vendored
View File

@ -14,4 +14,5 @@
# Dependency directories (remove the comment below to include it)
# vendor/
.vscode/
version

3
go.mod Normal file
View File

@ -0,0 +1,3 @@
module myschools.me/suguo/version
go 1.19

65
main.go Normal file
View File

@ -0,0 +1,65 @@
package main
import (
"fmt"
"io/ioutil"
"os"
"os/user"
"strconv"
)
func main() {
if len(os.Args) < 2 {
return
}
u, err := user.Current()
if err != nil {
fmt.Print("none")
return
}
dir := fmt.Sprintf("%s/.version", u.HomeDir)
if err := os.MkdirAll(dir, os.ModePerm); err != nil {
fmt.Print("none")
return
}
fname := fmt.Sprintf(`%s/%s`, dir, os.Args[1])
v, err := versionGet(&fname)
if err != nil {
fmt.Print("none")
}
versionSet(&fname, v)
fmt.Print(v)
}
func versionGet(fname *string) (v uint64, err error) {
f, err := os.OpenFile(*fname, os.O_RDONLY|os.O_CREATE, 0600)
if err != nil {
return
}
defer f.Close()
b, err := ioutil.ReadAll(f)
if err != nil {
return
}
v, _ = strconv.ParseUint(string(b), 10, 64)
v++
return
}
func versionSet(fname *string, v uint64) error {
f, err := os.OpenFile(*fname, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0600)
if err != nil {
return err
}
defer f.Close()
buf := strconv.FormatUint(v, 10)
if _, err := f.WriteString(buf); err != nil {
return err
}
return nil
}