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         SelfId *NodeId
  30         Neigh  map[NodeId]*Node
  31         Alias  map[string]*NodeId
  32
  33         Spool      string
  34         LogPath    string
  35         Quiet      bool
  36         Debug      bool
  37         NotifyFile *FromToYAML
  38         NotifyFreq *FromToYAML
  39 }
  40
  41 func (ctx *Ctx) FindNode(id string) (*Node, error) {
  42         nodeId, known := ctx.Alias[id]
  43         if known {
  44                 return ctx.Neigh[*nodeId], nil
  45         }
  46         nodeId, err := NodeIdFromString(id)
  47         if err != nil {
  48                 return nil, err
  49         }
  50         node, known := ctx.Neigh[*nodeId]
  51         if !known {
  52                 return nil, errors.New("Unknown node")
  53         }
  54         return node, nil
  55 }
  56
  57 func (ctx *Ctx) ensureRxDir(nodeId *NodeId) error {
  58         dirPath := filepath.Join(ctx.Spool, nodeId.String(), string(TRx))
  59         if err := os.MkdirAll(dirPath, os.FileMode(0700)); err != nil {
  60                 ctx.LogE("dir-ensure", SDS{"dir": dirPath, "err": err}, "")
  61                 return err
  62         }
  63         fd, err := os.Open(dirPath)
  64         if err != nil {
  65                 ctx.LogE("dir-ensure", SDS{"dir": dirPath, "err": err}, "")
  66                 return err
  67         }
  68         fd.Close()
  69         return nil
  70 }