X-Git-Url: http://www.git.cypherpunks.ru/?a=blobdiff_plain;f=src%2Fctx.go;h=58dba44df6156381216ee90e5043587d9b538039;hb=2b331a082bb710d1e5a529ed32dcbba0a2e843cb;hp=fb26185fbb3af9e4ee6024f8af805d1c2c2e7ee2;hpb=ab288c7da7b7d9896a422bfe9a9fb91e00316c54;p=nncp.git diff --git a/src/ctx.go b/src/ctx.go index fb26185..58dba44 100644 --- a/src/ctx.go +++ b/src/ctx.go @@ -21,13 +21,12 @@ import ( "errors" "fmt" "io/ioutil" - "log" "os" "path/filepath" + "strconv" + "strings" "syscall" - - "golang.org/x/sys/unix" ) type Ctx struct { @@ -36,6 +35,9 @@ type Ctx struct { Neigh map[NodeId]*Node Alias map[string]*NodeId + AreaId2Area map[AreaId]*Area + AreaName2Id map[string]*AreaId + Spool string LogPath string UmaskForce *int @@ -85,9 +87,7 @@ func (ctx *Ctx) ensureRxDir(nodeId *NodeId) error { } func CtxFromCmdline( - cfgPath, - spoolPath, - logPath string, + cfgPath, spoolPath, logPath string, quiet, showPrgrs, omitPrgrs, debug bool, ) (*Ctx, error) { env := os.Getenv(CfgPathEnv) @@ -97,11 +97,27 @@ func CtxFromCmdline( if showPrgrs && omitPrgrs { return nil, errors.New("simultaneous -progress and -noprogress") } - cfgRaw, err := ioutil.ReadFile(cfgPath) + fi, err := os.Stat(cfgPath) if err != nil { return nil, err } - ctx, err := CfgParse(cfgRaw) + var cfg *CfgJSON + if fi.IsDir() { + cfg, err = DirToCfg(cfgPath) + if err != nil { + return nil, err + } + } else { + cfgRaw, err := ioutil.ReadFile(cfgPath) + if err != nil { + return nil, err + } + cfg, err = CfgParse(cfgRaw) + if err != nil { + return nil, err + } + } + ctx, err := Cfg2Ctx(cfg) if err != nil { return nil, err } @@ -121,6 +137,18 @@ func CtxFromCmdline( } else { ctx.LogPath = logPath } + if strings.HasPrefix(ctx.LogPath, LogFdPrefix) { + ptr, err := strconv.ParseUint( + strings.TrimPrefix(ctx.LogPath, LogFdPrefix), 10, 64, + ) + if err != nil { + return nil, err + } + LogFd = os.NewFile(uintptr(ptr), CfgLogEnv) + if LogFd == nil { + return nil, errors.New("can not open:" + ctx.LogPath) + } + } if showPrgrs { ctx.ShowPrgrs = true } @@ -132,14 +160,6 @@ func CtxFromCmdline( return ctx, nil } -func (ctx *Ctx) IsEnoughSpace(want int64) bool { - var s unix.Statfs_t - if err := unix.Statfs(ctx.Spool, &s); err != nil { - log.Fatalln(err) - } - return int64(s.Bavail)*int64(s.Bsize) > want -} - func (ctx *Ctx) Umask() { if ctx.UmaskForce != nil { syscall.Umask(*ctx.UmaskForce)