4 from twisted.python import usage
5 from allmydata.util.stringutils import unicode_to_url
6 from allmydata.util.assertutil import precondition
9 # unit tests can override these to point at StringIO instances
15 ["quiet", "q", "Operate silently."],
16 ["version", "V", "Display version numbers and exit."],
17 ["version-and-path", None, "Display version numbers and paths to their locations and exit."],
20 def opt_version(self):
22 print allmydata.get_package_versions_string()
25 def opt_version_and_path(self):
27 print allmydata.get_package_versions_string(show_paths=True)
33 ["multiple", "m", "allow multiple basedirs to be specified at once"],
36 def postOptions(self):
38 raise usage.UsageError("<basedir> parameter is required")
41 self['basedirs'] = [os.path.abspath(os.path.expanduser(b)) for b in self.basedirs]
43 def parseArgs(self, *args):
46 precondition(isinstance(self['basedir'], (str, unicode)), self['basedir'])
47 self.basedirs.append(self['basedir'])
49 precondition(not [x for x in args if not isinstance(x, (str, unicode))], args)
50 self.basedirs.extend(args)
52 if len(args) == 0 and not self.basedirs:
53 if sys.platform == 'win32':
54 from allmydata.windows import registry
55 rbdp = registry.get_base_dir_path()
57 precondition(isinstance(registry.get_base_dir_path(), (str, unicode)), registry.get_base_dir_path())
58 self.basedirs.append(rbdp)
60 precondition(isinstance(os.path.expanduser("~/.tahoe"), (str, unicode)), os.path.expanduser("~/.tahoe"))
61 self.basedirs.append(os.path.expanduser("~/.tahoe"))
63 precondition(isinstance(args[0], (str, unicode)), args[0])
64 self.basedirs.append(args[0])
66 raise usage.UsageError("I wasn't expecting so many arguments")
68 class NoDefaultBasedirMixin(BasedirMixin):
69 def parseArgs(self, *args):
70 # create-client won't default to --basedir=~/.tahoe
73 precondition(isinstance(self['basedir'], (str, unicode)), self['basedir'])
74 self.basedirs.append(self['basedir'])
76 precondition(not [x for x in args if not isinstance(x, (str, unicode))], args)
77 self.basedirs.extend(args)
80 precondition(isinstance(args[0], (str, unicode)), args[0])
81 self.basedirs.append(args[0])
83 raise usage.UsageError("I wasn't expecting so many arguments")
85 raise usage.UsageError("--basedir must be provided")
87 DEFAULT_ALIAS = "tahoe"
90 def get_aliases(nodedir):
91 from allmydata import uri
93 aliasfile = os.path.join(nodedir, "private", "aliases")
94 rootfile = os.path.join(nodedir, "private", "root_dir.cap")
96 f = open(rootfile, "r")
97 rootcap = f.read().strip()
99 aliases["tahoe"] = uri.from_string_dirnode(rootcap).to_string()
100 except EnvironmentError:
103 f = codecs.open(aliasfile, "r", "utf-8")
104 for line in f.readlines():
106 if line.startswith("#") or not line:
108 name, cap = line.split(":", 1)
109 # normalize it: remove http: prefix, urldecode
110 cap = cap.strip().encode('utf-8')
111 aliases[name] = uri.from_string_dirnode(cap).to_string()
112 except EnvironmentError:
116 class DefaultAliasMarker:
119 pretend_platform_uses_lettercolon = False # for tests
120 def platform_uses_lettercolon_drivename():
121 if ("win32" in sys.platform.lower()
122 or "cygwin" in sys.platform.lower()
123 or pretend_platform_uses_lettercolon):
127 class UnknownAliasError(Exception):
130 def get_alias(aliases, path, default):
131 from allmydata import uri
132 # transform "work:path/filename" into (aliases["work"], "path/filename").
133 # If default=None, then an empty alias is indicated by returning
134 # DefaultAliasMarker. We special-case strings with a recognized cap URI
135 # prefix, to make it easy to access specific files/directories by their
137 # If the transformed alias is either not found in aliases, or is blank
138 # and default is not found in aliases, an UnknownAliasError is
141 if uri.has_uri_prefix(path.encode('utf-8')):
142 # We used to require "URI:blah:./foo" in order to get a subpath,
143 # stripping out the ":./" sequence. We still allow that for compatibility,
144 # but now also allow just "URI:blah/foo".
145 sep = path.find(":./")
147 return path[:sep], path[sep+3:]
150 return path[:sep], path[sep+1:]
152 colon = path.find(":")
156 return DefaultAliasMarker, path
157 if default not in aliases:
158 raise UnknownAliasError("No alias specified, and the default "
159 "'tahoe' alias doesn't exist. To create "
160 "it, use 'tahoe create-alias tahoe'.")
161 return aliases[default], path
162 if colon == 1 and default == None and platform_uses_lettercolon_drivename():
163 # treat C:\why\must\windows\be\so\weird as a local path, not a tahoe
164 # file in the "C:" alias
165 return DefaultAliasMarker, path
168 # no alias, but there's a colon in a dirname/filename, like
171 return DefaultAliasMarker, path
172 if default not in aliases:
173 raise UnknownAliasError("No alias specified, and the default "
174 "'tahoe' alias doesn't exist. To create "
175 "it, use 'tahoe create-alias tahoe'.")
176 return aliases[default], path
177 if alias not in aliases:
178 raise UnknownAliasError("Unknown alias '%s', please create it with 'tahoe add-alias' or 'tahoe create-alias'." % alias)
179 return aliases[alias], path[colon+1:]
181 def escape_path(path):
182 segments = path.split("/")
183 return "/".join([urllib.quote(unicode_to_url(s)) for s in segments])