f5eb2690a17203171fc82736a0f842de0c450e0f
[nepi.git] / src / nepi / resources / linux / ccn / ccncat.py
1 #
2 #    NEPI, a framework to manage network experiments
3 #    Copyright (C) 2013 INRIA
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 # Author: Alina Quereilhac <alina.quereilhac@inria.fr>
19
20 from nepi.execution.attribute import Attribute, Flags, Types
21 from nepi.execution.resource import clsinit_copy, ResourceState, \
22     reschedule_delay
23 from nepi.resources.linux.ccn.ccnapplication import LinuxCCNApplication
24
25 import os
26
27 @clsinit_copy
28 class LinuxCCNCat(LinuxCCNApplication):
29     _rtype = "LinuxCCNCat"
30
31     @classmethod
32     def _register_attributes(cls):
33         content_name = Attribute("contentName",
34             "Content name for the content to peek",
35             flags = Flags.Design)
36
37         pipeline = Attribute("pipeline",
38             "CCNCat pipeline",
39             flags = Flags.Design)
40
41         cls._register_attribute(content_name)
42         cls._register_attribute(pipeline)
43
44     def __init__(self, ec, guid):
45         super(LinuxCCNCat, self).__init__(ec, guid)
46         self._home = "ccncat-%s" % self.guid
47
48     def do_deploy(self):
49         if not self.ccnd or self.ccnd.state < ResourceState.READY:
50             self.debug("---- RESCHEDULING DEPLOY ---- node state %s " % self.node.state )
51             self.ec.schedule(reschedule_delay, self.deploy)
52         else:
53             command = self.get("command")
54             if not command:
55                 command = "ccncat %s" % self.get("contentName")
56             if self.get("pipeline"):
57                 command += " -p %s " % self.get("pipeline")
58
59             self.set("command", command) 
60
61             self.info("Deploying command '%s' " % command)
62             
63             if not self.get("env"):
64                 self.set("env", self._environment)
65
66             self.do_discover()
67             self.do_provision()
68
69             self.set_ready()
70
71     def valid_connection(self, guid):
72         # TODO: Validate!
73         return True
74