src/cypherpunks.ru/nncp/ctx.go

   1 /*
   2 NNCP -- Node to Node copy, utilities for store-and-forward data exchange
   3 Copyright (C) 2016-2017 Sergey Matveev <stargrave@stargrave.org>
   4
   5 This program is free software: you can redistribute it and/or modify
   6 it under the terms of the GNU General Public License as published by
   7 the Free Software Foundation, either version 3 of the License, or
   8 (at your option) any later version.
   9
  10 This program is distributed in the hope that it will be useful,
  11 but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 GNU General Public License for more details.
  14
  15 You should have received a copy of the GNU General Public License
  16 along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17 */
  18
  19 package nncp
  20
  21 import (
  22         "errors"
  23         "os"
  24         "path/filepath"
  25 )
  26
  27 type Ctx struct {
  28         Self  *NodeOur
  29         Neigh map[NodeId]*Node
  30         Alias map[string]*NodeId
  31
  32         Spool      string
  33         LogPath    string
  34         Quiet      bool
  35         Debug      bool
  36         NotifyFile *FromToYAML
  37         NotifyFreq *FromToYAML
  38 }
  39
  40 func (ctx *Ctx) FindNode(id string) (*Node, error) {
  41         nodeId, known := ctx.Alias[id]
  42         if known {
  43                 return ctx.Neigh[*nodeId], nil
  44         }
  45         nodeId, err := NodeIdFromString(id)
  46         if err != nil {
  47                 return nil, err
  48         }
  49         node, known := ctx.Neigh[*nodeId]
  50         if !known {
  51                 return nil, errors.New("Unknown node")
  52         }
  53         return node, nil
  54 }
  55
  56 func (ctx *Ctx) ensureRxDir(nodeId *NodeId) error {
  57         dirPath := filepath.Join(ctx.Spool, nodeId.String(), string(TRx))
  58         if err := os.MkdirAll(dirPath, os.FileMode(0700)); err != nil {
  59                 ctx.LogE("dir-ensure", SDS{"dir": dirPath, "err": err}, "")
  60                 return err
  61         }
  62         fd, err := os.Open(dirPath)
  63         if err != nil {
  64                 ctx.LogE("dir-ensure", SDS{"dir": dirPath, "err": err}, "")
  65                 return err
  66         }
  67         fd.Close()
  68         return nil
  69 }