3 from cStringIO import StringIO
4 from twisted.python import usage
6 from .common import BaseOptions, BasedirOptions, get_aliases
7 from .cli import MakeDirectoryOptions, LnOptions, CreateAliasOptions
9 from allmydata.util.encodingutil import argv_to_abspath
10 from allmydata.util import fileutil
11 from allmydata import uri
13 INVITE_SEPARATOR = "+"
15 class CreateOptions(BasedirOptions):
18 synopsis = "MAGIC_ALIAS: [NICKNAME LOCALDIR]"
19 def parseArgs(self, alias, nickname=None, localdir=None):
20 BasedirOptions.parseArgs(self)
21 if not alias.endswith(':'):
22 raise usage.UsageError("An alias must end with a ':' character.")
23 self.alias = alias[:-1]
24 self.nickname = nickname
25 self.localdir = argv_to_abspath(str(localdir))
26 if self.nickname and not self.localdir:
27 raise usage.UsageError("If NICKNAME is specified then LOCALDIR must also be specified.")
28 node_url_file = os.path.join(self['node-directory'], "node.url")
29 self['node-url'] = fileutil.read(node_url_file).strip()
31 def _delegate_options(source_options, target_options):
32 target_options.aliases = get_aliases(source_options['node-directory'])
33 target_options["node-url"] = source_options["node-url"]
34 target_options["node-directory"] = source_options["node-directory"]
35 target_options.stdin = StringIO("")
36 target_options.stdout = StringIO()
37 target_options.stderr = StringIO()
41 from allmydata.scripts import tahoe_add_alias
42 create_alias_options = _delegate_options(options, CreateAliasOptions())
43 create_alias_options.alias = options.alias
45 rc = tahoe_add_alias.create_alias(create_alias_options)
47 print >>options.stderr, create_alias_options.stderr.getvalue()
49 print >>options.stdout, create_alias_options.stdout.getvalue()
51 if options.nickname is not None:
52 invite_options = _delegate_options(options, InviteOptions())
53 invite_options.alias = options.alias
54 invite_options.nickname = options.nickname
55 rc = invite(invite_options)
57 print >>options.stderr, "magic-folder: failed to invite after create\n"
58 print >>options.stderr, invite_options.stderr.getvalue()
60 invite_code = invite_options.stdout.getvalue().strip()
61 join_options = _delegate_options(options, JoinOptions())
62 join_options.invite_code = invite_code
63 fields = invite_code.split(INVITE_SEPARATOR)
65 raise usage.UsageError("Invalid invite code.")
66 join_options.magic_readonly_cap, join_options.dmd_write_cap = fields
67 join_options.local_dir = options.localdir
68 rc = join(join_options)
70 print >>options.stderr, "magic-folder: failed to join after create\n"
71 print >>options.stderr, join_options.stderr.getvalue()
75 class InviteOptions(BasedirOptions):
77 synopsis = "MAGIC_ALIAS: NICKNAME"
79 def parseArgs(self, alias, nickname=None):
80 BasedirOptions.parseArgs(self)
81 if not alias.endswith(':'):
82 raise usage.UsageError("An alias must end with a ':' character.")
83 self.alias = alias[:-1]
84 self.nickname = nickname
85 node_url_file = os.path.join(self['node-directory'], "node.url")
86 self['node-url'] = open(node_url_file, "r").read().strip()
87 aliases = get_aliases(self['node-directory'])
88 self.aliases = aliases
91 from allmydata.scripts import tahoe_mkdir
92 mkdir_options = _delegate_options(options, MakeDirectoryOptions())
93 mkdir_options.where = None
95 rc = tahoe_mkdir.mkdir(mkdir_options)
97 print >>options.stderr, "magic-folder: failed to mkdir\n"
99 dmd_write_cap = mkdir_options.stdout.getvalue().strip()
100 dmd_readonly_cap = unicode(uri.from_string(dmd_write_cap).get_readonly().to_string(), 'utf-8')
101 if dmd_readonly_cap is None:
102 print >>options.stderr, "magic-folder: failed to diminish dmd write cap\n"
105 magic_write_cap = get_aliases(options["node-directory"])[options.alias]
106 magic_readonly_cap = unicode(uri.from_string(magic_write_cap).get_readonly().to_string(), 'utf-8')
107 # tahoe ln CLIENT_READCAP COLLECTIVE_WRITECAP/NICKNAME
108 ln_options = _delegate_options(options, LnOptions())
109 ln_options.from_file = dmd_readonly_cap
110 ln_options.to_file = u"%s/%s" % (magic_write_cap, options.nickname)
111 rc = tahoe_mv.mv(ln_options, mode="link")
113 print >>options.stderr, "magic-folder: failed to create link\n"
114 print >>options.stderr, ln_options.stderr.getvalue()
117 print >>options.stdout, "%s%s%s" % (magic_readonly_cap, INVITE_SEPARATOR, dmd_write_cap)
120 class JoinOptions(BasedirOptions):
121 synopsis = "INVITE_CODE LOCAL_DIR"
123 magic_readonly_cap = ""
124 def parseInvite(self, invite_code, local_dir):
125 self.local_dir = argv_to_abspath(str(local_dir))
126 fields = invite_code.split(INVITE_SEPARATOR)
128 raise usage.UsageError("Invalid invite code.")
129 self.magic_readonly_cap, self.dmd_write_cap = fields
130 def parseArgs(self, invite_code, local_dir):
131 BasedirOptions.parseArgs(self)
132 self.parseInvite(invite_code, local_dir)
135 dmd_cap_file = os.path.join(options["node-directory"], "private/magic_folder_dircap")
136 collective_readcap_file = os.path.join(options["node-directory"], "private/collective_dircap")
138 fileutil.write(dmd_cap_file, options.dmd_write_cap)
139 fileutil.write(collective_readcap_file, options.magic_readonly_cap)
140 fileutil.write(os.path.join(options["node-directory"], "tahoe.cfg"),
141 "[magic_folder]\nenabled = True\nlocal.directory = %s\n"
142 % (options.local_dir.encode('utf-8'),), mode="ab")
145 class MagicFolderCommand(BaseOptions):
147 ["create", None, CreateOptions, "Create a Magic Folder."],
148 ["invite", None, InviteOptions, "Invite someone to a Magic Folder."],
149 ["join", None, JoinOptions, "Join a Magic Folder."],
151 def postOptions(self):
152 if not hasattr(self, 'subOptions'):
153 raise usage.UsageError("must specify a subcommand")
154 def getSynopsis(self):
155 return "Usage: tahoe [global-options] magic SUBCOMMAND"
156 def getUsage(self, width=None):
157 t = BaseOptions.getUsage(self, width)
159 Please run e.g. 'tahoe magic-folder create --help' for more details on each
170 def do_magic_folder(options):
171 so = options.subOptions
172 so.stdout = options.stdout
173 so.stderr = options.stderr
174 f = subDispatch[options.subCommand]
178 ["magic-folder", None, MagicFolderCommand,
179 "Magic Folder subcommands: use 'tahoe magic-folder' for a list."],
183 "magic-folder": do_magic_folder,