def join_workspace(self, context, host, name, owner, possible_dirs=None):
utils.reload_settings()
# legacy urls in emacs...
if owner and owner[:2] == "r/":
owner = owner[2:]
if not utils.can_auth():
success = yield self.create_or_link_account, context, host, False
if not success:
return
utils.reload_settings()
possible_dirs = possible_dirs or []
for d in possible_dirs:
info = utils.read_floo_file(d)
if not info:
continue
try:
parsed_url = utils.parse_url(info["url"])
except Exception:
parsed_url = None
if (
parsed_url
and parsed_url["host"] == host
and parsed_url["workspace"] == name
and parsed_url["owner"] == owner
):
self.remote_connect(context, host, owner, name, d)
return
try:
d = utils.get_persistent_data()["workspaces"][owner][name]["path"]
except Exception:
d = ""
if d and os.path.isdir(d):
self.remote_connect(context, host, owner, name, d)
return
d = d or os.path.join(G.SHARE_DIR or G.BASE_DIR, owner, name)
join_action = utils.JOIN_ACTION.PROMPT
while True:
d = yield self.user_dir, context, "Save workspace files to: ", d
if not d:
return
d = os.path.realpath(os.path.expanduser(d))
if not os.path.isdir(d):
y_or_n = yield self.user_y_or_n, context, "%s is not a directory. Create it? " % d, "Create Directory"
if not y_or_n:
return
utils.mkdir(d)
if not os.path.isdir(d):
msg.error("Couldn't create directory", d)
continue
join_action = utils.JOIN_ACTION.DOWNLOAD
if os.path.isdir(d):
self.remote_connect(context, host, owner, name, d, join_action)
return
def make_dir(d):
d = os.path.realpath(os.path.expanduser(d))
if not os.path.isdir(d):
make_dir = sublime.ok_cancel_dialog(
'%s is not a directory. Create it?' % d)
if not make_dir:
return self.window.show_input_panel(
'%s is not a directory. Enter an existing path:' % d,
d, None, None, None)
try:
utils.mkdir(d)
except Exception as e:
return sublime.error_message(
'Could not create directory %s: %s' % (d, str_e(e)))
G.PROJECT_PATH = d
if self.upload:
result['upload'] = d
else:
result['upload'] = ""
utils.add_workspace_to_persistent_json(result['owner'],
result['workspace'],
workspace_url, d)
open_workspace_window(lambda: run_agent(**result))
def _scan_dir(self, bufs, ig, read_only):
changed_bufs = []
missing_bufs = []
new_files = set()
if not read_only:
new_files = set([utils.to_rel_path(x) for x in ig.list_paths()])
for buf_id, buf in bufs.items():
buf_id = int(buf_id) # json keys must be strings
buf_path = utils.get_full_path(buf['path'])
new_dir = os.path.dirname(buf_path)
utils.mkdir(new_dir)
self.bufs[buf_id] = buf
self.paths_to_ids[buf['path']] = buf_id
view = self.get_view(buf_id)
if view and not view.is_loading() and buf['encoding'] == 'utf8':
view_text = view.get_text()
view_md5 = hashlib.md5(view_text.encode('utf-8')).hexdigest()
buf['buf'] = view_text
buf['view'] = view
G.VIEW_TO_HASH[view.native_id] = view_md5
if view_md5 == buf['md5']:
msg.debug('md5 sum matches view. not getting buffer ',
buf['path'])
else:
changed_bufs.append(buf)
buf['md5'] = view_md5
continue
try:
if buf['encoding'] == 'utf8':
if io:
buf_fd = io.open(buf_path, 'Urt', encoding='utf8')
buf_buf = buf_fd.read()
else:
buf_fd = open(buf_path, 'rb')
buf_buf = buf_fd.read().decode('utf-8').replace(
'\r\n', '\n')
md5 = hashlib.md5(buf_buf.encode('utf-8')).hexdigest()
else:
buf_fd = open(buf_path, 'rb')
buf_buf = buf_fd.read()
md5 = hashlib.md5(buf_buf).hexdigest()
buf_fd.close()
buf['buf'] = buf_buf
if md5 == buf['md5']:
msg.debug('md5 sum matches. not getting buffer ',
buf['path'])
else:
msg.debug('md5 differs. possibly getting buffer later ',
buf['path'])
changed_bufs.append(buf)
buf['md5'] = md5
except Exception as e:
msg.debug('Error calculating md5 for ', buf['path'], ', ',
str_e(e))
missing_bufs.append(buf)
return changed_bufs, missing_bufs, new_files
def join_workspace(self, context, host, name, owner, possible_dirs=None):
utils.reload_settings()
# legacy urls in emacs...
if owner and owner[:2] == "r/":
owner = owner[2:]
if not utils.can_auth():
success = yield self.create_or_link_account, context, host, False
if not success:
return
utils.reload_settings()
possible_dirs = possible_dirs or []
for d in possible_dirs:
info = utils.read_floo_file(d)
if not info:
continue
try:
parsed_url = utils.parse_url(info['url'])
except Exception:
parsed_url = None
if parsed_url and parsed_url['host'] == host and parsed_url[
'workspace'] == name and parsed_url['owner'] == owner:
self.remote_connect(context, host, owner, name, d)
return
try:
d = utils.get_persistent_data()['workspaces'][owner][name]['path']
except Exception:
d = ''
if d and os.path.isdir(d):
self.remote_connect(context, host, owner, name, d)
return
# TODO: make per-host settings fully general
host_share_dir = G.AUTH.get(host, {}).get('share_dir')
d = d or os.path.join(host_share_dir or G.SHARE_DIR or G.BASE_DIR,
owner, name)
join_action = utils.JOIN_ACTION.PROMPT
while True:
d = yield self.user_dir, context, 'Save workspace files to: ', d
if not d:
return
d = os.path.realpath(os.path.expanduser(d))
if not os.path.isdir(d):
y_or_n = yield self.user_y_or_n, context, '%s is not a directory. Create it? ' % d, "Create Directory"
if not y_or_n:
return
utils.mkdir(d)
if not os.path.isdir(d):
msg.error("Couldn't create directory", d)
continue
join_action = utils.JOIN_ACTION.DOWNLOAD
if os.path.isdir(d):
self.remote_connect(context, host, owner, name, d, join_action)
return
def join_workspace(self, context, host, name, owner, possible_dirs=None):
utils.reload_settings()
# legacy urls in emacs...
if owner and owner[:2] == "r/":
owner = owner[2:]
if not utils.can_auth():
success = yield self.create_or_link_account, context, host, False
if not success:
return
utils.reload_settings()
possible_dirs = possible_dirs or []
for d in possible_dirs:
info = utils.read_floo_file(d)
if not info:
continue
try:
parsed_url = utils.parse_url(info['url'])
except Exception:
parsed_url = None
if parsed_url and parsed_url['host'] == host and parsed_url['workspace'] == name and parsed_url['owner'] == owner:
self.remote_connect(context, host, owner, name, d)
return
try:
d = utils.get_persistent_data()['workspaces'][owner][name]['path']
except Exception:
d = ''
if d and os.path.isdir(d):
self.remote_connect(context, host, owner, name, d)
return
# TODO: make per-host settings fully general
host_share_dir = G.AUTH.get(host, {}).get('share_dir')
d = d or os.path.join(host_share_dir or G.SHARE_DIR or G.BASE_DIR, owner, name)
join_action = utils.JOIN_ACTION.PROMPT
while True:
d = yield self.user_dir, context, 'Save workspace files to: ', d
if not d:
return
d = os.path.realpath(os.path.expanduser(d))
if not os.path.isdir(d):
y_or_n = yield self.user_y_or_n, context, '%s is not a directory. Create it? ' % d, "Create Directory"
if not y_or_n:
return
utils.mkdir(d)
if not os.path.isdir(d):
msg.error("Couldn't create directory", d)
continue
join_action = utils.JOIN_ACTION.DOWNLOAD
if os.path.isdir(d):
self.remote_connect(context, host, owner, name, d, join_action)
return
def _scan_dir(self, bufs, ig, read_only):
changed_bufs = []
missing_bufs = []
new_files = set()
if not read_only:
new_files = set([utils.to_rel_path(x) for x in ig.list_paths()])
for buf_id, buf in bufs.items():
buf_id = int(buf_id) # json keys must be strings
buf_path = utils.get_full_path(buf['path'])
new_dir = os.path.dirname(buf_path)
utils.mkdir(new_dir)
self.bufs[buf_id] = buf
self.paths_to_ids[buf['path']] = buf_id
view = self.get_view(buf_id)
if view and not view.is_loading() and buf['encoding'] == 'utf8':
view_text = view.get_text()
view_md5 = hashlib.md5(view_text.encode('utf-8')).hexdigest()
buf['buf'] = view_text
buf['view'] = view
G.VIEW_TO_HASH[view.native_id] = view_md5
if view_md5 == buf['md5']:
msg.debug('md5 sum matches view. not getting buffer ', buf['path'])
else:
changed_bufs.append(buf)
buf['md5'] = view_md5
continue
try:
if buf['encoding'] == 'utf8':
if io:
buf_fd = io.open(buf_path, 'Urt', encoding='utf8')
buf_buf = buf_fd.read()
else:
buf_fd = open(buf_path, 'rb')
buf_buf = buf_fd.read().decode('utf-8').replace('\r\n', '\n')
md5 = hashlib.md5(buf_buf.encode('utf-8')).hexdigest()
else:
buf_fd = open(buf_path, 'rb')
buf_buf = buf_fd.read()
md5 = hashlib.md5(buf_buf).hexdigest()
buf_fd.close()
buf['buf'] = buf_buf
if md5 == buf['md5']:
msg.debug('md5 sum matches. not getting buffer ', buf['path'])
else:
msg.debug('md5 differs. possibly getting buffer later ', buf['path'])
changed_bufs.append(buf)
buf['md5'] = md5
except Exception as e:
msg.debug('Error calculating md5 for ', buf['path'], ', ', str_e(e))
missing_bufs.append(buf)
return changed_bufs, missing_bufs, new_files
def floobits_join_workspace(workspace_url, d='', upload_path=None):
editor.line_endings = _get_line_endings()
msg.debug('workspace url is %s' % workspace_url)
try:
result = utils.parse_url(workspace_url)
except Exception as e:
return msg.error(str(e))
if d:
utils.mkdir(d)
else:
try:
d = utils.get_persistent_data()['workspaces'][result['owner']][result['workspace']]['path']
except Exception:
d = os.path.realpath(os.path.join(G.COLAB_DIR, result['owner'], result['workspace']))
prompt = 'Save workspace files to: '
if not os.path.isdir(d):
while True:
d = vim_input(prompt, d, 'dir')
if d == '':
continue
d = os.path.realpath(os.path.expanduser(d))
if os.path.isfile(d):
prompt = '%s is not a directory. Enter an existing path or a path I can create: ' % d
continue
if not os.path.isdir(d):
try:
utils.mkdir(d)
except Exception as e:
prompt = 'Couldn\'t make dir %s: %s ' % (d, str(e))
continue
break
d = os.path.realpath(os.path.abspath(d) + os.sep)
try:
utils.add_workspace_to_persistent_json(result['owner'], result['workspace'], workspace_url, d)
except Exception as e:
return msg.error('Error adding workspace to persistent.json: %s' % str(e))
G.PROJECT_PATH = d
vim.command('cd %s' % G.PROJECT_PATH)
msg.debug('Joining workspace %s' % workspace_url)
floobits_stop_everything()
try:
conn = VimHandler(result['owner'], result['workspace'])
if upload_path:
conn.once('room_info', lambda: G.AGENT.upload(upload_path))
reactor.connect(conn, result['host'], result['port'], result['secure'])
except Exception as e:
msg.error(str(e))
tb = traceback.format_exc()
msg.debug(tb)
if not G.TIMERS:
start_event_loop()
def remote_connect(self,
context,
host,
owner,
workspace,
d,
join_action=utils.JOIN_ACTION.PROMPT):
G.PROJECT_PATH = os.path.realpath(d)
try:
utils.mkdir(os.path.dirname(G.PROJECT_PATH))
except Exception as e:
msg.error("Couldn't create directory", G.PROJECT_PATH, str_e(e))
return
auth = G.AUTH.get(host)
if not auth:
success = yield self.link_account, context, host
if not success:
return
auth = G.AUTH.get(host)
if not auth:
msg.error("Something went really wrong.")
return
try:
res = api.get_workspace(host, owner, workspace)
if res.code == 404:
msg.error("The workspace https://%s/%s/%s does not exist" %
(host, owner, workspace))
return
except Exception as e:
message = 'Error getting workspace https://%s/%s/%s: %s' % (
host, owner, workspace, str_e(e))
msg.error(message)
editor.error_message(message)
return
if self.agent:
try:
self.agent.stop()
except Exception:
pass
G.WORKSPACE_WINDOW = yield self.get_a_window, d
self.agent = self._make_agent(context, owner, workspace, auth,
join_action)
self.emit("agent", self.agent)
reactor.reactor.connect(self.agent, host, G.DEFAULT_PORT, True)
url = self.agent.workspace_url
utils.add_workspace_to_persistent_json(owner, workspace, url, d)
utils.update_recent_workspaces(url)
def _on_rename_buf(self, data):
del self.paths_to_ids[data['old_path']]
self.paths_to_ids[data['path']] = data['id']
new = utils.get_full_path(data['path'])
old = utils.get_full_path(data['old_path'])
new_dir = os.path.split(new)[0]
if new_dir:
utils.mkdir(new_dir)
view = self.get_view(data['id'])
if view:
view.rename(new)
else:
os.rename(old, new)
self.bufs[data['id']]['path'] = data['path']
def _on_rename_buf(self, data):
del self.paths_to_ids[data['old_path']]
self.paths_to_ids[data['path']] = data['id']
new = utils.get_full_path(data['path'])
old = utils.get_full_path(data['old_path'])
new_dir = os.path.split(new)[0]
if new_dir:
utils.mkdir(new_dir)
view = self.get_view(data['id'])
if view:
view.rename(new)
else:
os.rename(old, new)
self.bufs[data['id']]['path'] = data['path']
def make_dir(d):
d = os.path.realpath(os.path.expanduser(d))
if not os.path.isdir(d):
make_dir = sublime.ok_cancel_dialog('%s is not a directory. Create it?' % d)
if not make_dir:
return self.window.show_input_panel('%s is not a directory. Enter an existing path:' % d, d, None, None, None)
try:
utils.mkdir(d)
except Exception as e:
return sublime.error_message('Could not create directory %s: %s' % (d, str(e)))
G.PROJECT_PATH = d
add_workspace_to_persistent_json(result['owner'], result['workspace'], workspace_url, d)
open_workspace_window(lambda: run_agent(**result))
def _on_rename_buf(self, data):
del self.paths_to_ids[data['old_path']]
self.paths_to_ids[data['path']] = data['id']
new = utils.get_full_path(data['path'])
old = utils.get_full_path(data['old_path'])
new_dir = os.path.split(new)[0]
if new_dir:
utils.mkdir(new_dir)
view = self.get_view(data['id'])
if view:
view.rename(new)
else:
try:
os.rename(old, new)
except Exception as e:
msg.debug('Error moving ', old, 'to', new, str_e(e))
utils.save_buf(self.bufs[data.id])
self.bufs[data['id']]['path'] = data['path']
def _on_rename_buf(self, data):
del self.paths_to_ids[data['old_path']]
self.paths_to_ids[data['path']] = data['id']
new = utils.get_full_path(data['path'])
old = utils.get_full_path(data['old_path'])
new_dir = os.path.split(new)[0]
if new_dir:
utils.mkdir(new_dir)
view = self.get_view(data['id'])
if view:
view.rename(new)
else:
try:
os.rename(old, new)
except Exception as e:
msg.debug('Error moving ', old, 'to', new, str_e(e))
utils.save_buf(self.bufs[data.id])
self.bufs[data['id']]['path'] = data['path']
def remote_connect(self, context, host, owner, workspace, d, join_action=utils.JOIN_ACTION.PROMPT):
G.PROJECT_PATH = os.path.realpath(d)
try:
utils.mkdir(os.path.dirname(G.PROJECT_PATH))
except Exception as e:
msg.error("Couldn't create directory", G.PROJECT_PATH, str_e(e))
return
auth = G.AUTH.get(host)
if not auth:
success = yield self.link_account, context, host
if not success:
return
auth = G.AUTH.get(host)
if not auth:
msg.error("Something went really wrong.")
return
try:
res = api.get_workspace(host, owner, workspace)
if res.code == 404:
msg.error("The workspace https://%s/%s/%s does not exist" % (host, owner, workspace))
return
except Exception as e:
message = 'Error getting workspace https://%s/%s/%s: %s' % (host, owner, workspace, str_e(e))
msg.error(message)
editor.error_message(message)
return
if self.agent:
try:
self.agent.stop()
except Exception:
pass
G.WORKSPACE_WINDOW = yield self.get_a_window, d
self.agent = self._make_agent(context, owner, workspace, auth, join_action)
self.emit("agent", self.agent)
reactor.reactor.connect(self.agent, host, G.DEFAULT_PORT, True)
url = self.agent.workspace_url
utils.add_workspace_to_persistent_json(owner, workspace, url, d)
utils.update_recent_workspaces(url)
def join_workspace(self, data, owner, workspace, dir_to_make=None):
d = data['response']
workspace_url = utils.to_workspace_url({'secure': True, 'owner': owner, 'workspace': workspace})
if dir_to_make:
if d:
d = dir_to_make
utils.mkdir(d)
else:
d = ''
if d == '':
return self.get_input('Give me a directory to sync data to: ', G.PROJECT_PATH, self.join_workspace, owner, workspace)
d = os.path.realpath(os.path.expanduser(d))
if not os.path.isdir(d):
if dir_to_make:
return msg.error("Couldn't create directory %s" % dir_to_make)
prompt = '%s is not a directory. Create it? ' % d
return self.get_input(prompt, '', self.join_workspace, owner, workspace, dir_to_make=d, y_or_n=True)
try:
G.PROJECT_PATH = d
utils.mkdir(os.path.dirname(G.PROJECT_PATH))
self.remote_connect(workspace_url)
except Exception as e:
return msg.error("Couldn't create directory %s: %s" % (G.PROJECT_PATH, str(e)))
def join_workspace(workspace_url, on_auth=None):
global agent
msg.debug("workspace url is %s" % workspace_url)
try:
result = utils.parse_url(workspace_url)
except Exception as e:
return msg.error(str(e))
G.PROJECT_PATH = os.path.realpath(
os.path.join(G.COLAB_DIR, result['owner'], result['workspace']))
utils.mkdir(os.path.dirname(G.PROJECT_PATH))
d = ''
# TODO: really bad prompt here
prompt = "Give me a directory to sync data to (or just press enter): "
if not os.path.isdir(G.PROJECT_PATH):
while True:
d = vim_input(prompt, d, "dir")
if d == '':
utils.mkdir(G.PROJECT_PATH)
break
d = os.path.realpath(os.path.expanduser(d))
if os.path.isfile(d):
prompt = '%s is not a directory. Enter an existing path (or press enter): ' % d
continue
if not os.path.isdir(d):
try:
utils.mkdir(d)
except Exception as e:
prompt = "Couldn't make dir: %s because %s " % (d, str(e))
continue
try:
os.symlink(d, G.PROJECT_PATH)
break
except Exception as e:
return msg.error("Couldn't create symlink from %s to %s: %s" %
(d, G.PROJECT_PATH, str(e)))
G.PROJECT_PATH = os.path.realpath(G.PROJECT_PATH + os.sep)
vim.command('cd %s' % G.PROJECT_PATH)
msg.debug("joining workspace %s" % workspace_url)
stop_everything()
try:
start_event_loop()
agent = AgentConnection(on_auth=on_auth, Protocol=Protocol, **result)
# owner and workspace name are slugfields so this should be safe
agent.connect()
except Exception as e:
msg.error(str(e))
tb = traceback.format_exc()
msg.debug(tb)
stop_everything()
def floobits_share_dir(dir_to_share, perms):
utils.reload_settings()
workspace_name = os.path.basename(dir_to_share)
G.PROJECT_PATH = os.path.realpath(dir_to_share)
msg.debug('%s %s %s' % (G.USERNAME, workspace_name, G.PROJECT_PATH))
file_to_share = None
dir_to_share = os.path.expanduser(dir_to_share)
dir_to_share = utils.unfuck_path(dir_to_share)
dir_to_share = os.path.abspath(dir_to_share)
dir_to_share = os.path.realpath(dir_to_share)
workspace_name = os.path.basename(dir_to_share)
if os.path.isfile(dir_to_share):
file_to_share = dir_to_share
dir_to_share = os.path.dirname(dir_to_share)
try:
utils.mkdir(dir_to_share)
except Exception:
return msg.error("The directory %s doesn't exist and I can't create it." % dir_to_share)
if not os.path.isdir(dir_to_share):
return msg.error('The directory %s doesn\'t appear to exist' % dir_to_share)
floo_file = os.path.join(dir_to_share, '.floo')
# look for the .floo file for hints about previous behavior
info = {}
try:
floo_info = open(floo_file, 'rb').read().decode('utf-8')
info = json.loads(floo_info)
except (IOError, OSError):
pass
except Exception:
msg.warn('couldn\'t read the floo_info file: %s' % floo_file)
workspace_url = info.get('url')
if workspace_url:
parsed_url = api.prejoin_workspace(workspace_url, dir_to_share, {'perms': perms})
if parsed_url:
return floobits_join_workspace(workspace_url, dir_to_share, upload_path=file_to_share or dir_to_share)
filter_func = lambda workspace_url: api.prejoin_workspace(workspace_url, dir_to_share, {'perms': perms})
parsed_url = utils.get_workspace_by_path(dir_to_share, filter_func)
if parsed_url:
return floobits_join_workspace(workspace_url, dir_to_share, upload_path=file_to_share or dir_to_share)
try:
r = api.get_orgs_can_admin()
except IOError as e:
return editor.error_message('Error getting org list: %s' % str(e))
if r.code >= 400 or len(r.body) == 0:
workspace_name = vim_input('Workspace name:', workspace_name, "file")
return create_workspace(workspace_name, dir_to_share, G.USERNAME, perms, upload_path=file_to_share or dir_to_share)
orgs = r.body
if len(orgs) == 0:
return create_workspace(workspace_name, dir_to_share, G.USERNAME, perms, upload_path=file_to_share or dir_to_share)
choices = []
choices.append(G.USERNAME)
for o in orgs:
choices.append(o['name'])
owner = vim_choice('Create workspace for:', G.USERNAME, choices)
if owner:
return create_workspace(workspace_name, dir_to_share, owner, perms, upload_path=file_to_share or dir_to_share)
def on_input(self, dir_to_share):
file_to_share = None
dir_to_share = os.path.expanduser(dir_to_share)
dir_to_share = os.path.realpath(utils.unfuck_path(dir_to_share))
workspace_name = os.path.basename(dir_to_share)
workspace_url = None
print(G.COLAB_DIR, G.USERNAME, workspace_name)
def find_workspace(workspace_url):
r = api.get_workspace_by_url(workspace_url)
if r.code < 400:
on_room_info_waterfall.add(ignore.create_flooignore, dir_to_share)
on_room_info_waterfall.add(lambda: G.AGENT.upload(dir_to_share, on_room_info_msg))
return r
try:
result = utils.parse_url(workspace_url)
d = utils.get_persistent_data()
del d['workspaces'][result['owner']][result['name']]
utils.update_persistent_data(d)
except Exception as e:
msg.debug(unicode(e))
return
def join_workspace(workspace_url):
try:
w = find_workspace(workspace_url)
except Exception as e:
sublime.error_message('Error: %s' % str(e))
return False
if not w:
return False
msg.debug('workspace: %s', json.dumps(w.body))
# if self.api_args:
anon_perms = w.body.get('perms', {}).get('AnonymousUser', [])
new_anon_perms = self.api_args.get('perms').get('AnonymousUser', [])
if set(anon_perms) != set(new_anon_perms):
msg.debug(str(anon_perms), str(new_anon_perms))
w.body['perms']['AnonymousUser'] = new_anon_perms
response = api.update_workspace(w.body['owner'], w.body['name'], w.body)
msg.debug(str(response.body))
utils.add_workspace_to_persistent_json(w.body['owner'], w.body['name'], workspace_url, dir_to_share)
self.window.run_command('floobits_join_workspace', {
'workspace_url': workspace_url,
'agent_conn_kwargs': {'get_bufs': False}})
return True
if os.path.isfile(dir_to_share):
file_to_share = dir_to_share
dir_to_share = os.path.dirname(dir_to_share)
try:
utils.mkdir(dir_to_share)
except Exception:
return sublime.error_message('The directory %s doesn\'t exist and I can\'t make it.' % dir_to_share)
floo_file = os.path.join(dir_to_share, '.floo')
info = {}
try:
floo_info = open(floo_file, 'r').read()
info = json.loads(floo_info)
except (IOError, OSError):
pass
except Exception:
msg.error('Couldn\'t read the floo_info file: %s' % floo_file)
workspace_url = info.get('url')
try:
utils.parse_url(workspace_url)
except Exception:
workspace_url = None
if workspace_url and join_workspace(workspace_url):
return
for owner, workspaces in utils.get_persistent_data()['workspaces'].items():
for name, workspace in workspaces.items():
if workspace['path'] == dir_to_share:
workspace_url = workspace['url']
if join_workspace(workspace_url):
return
# make & join workspace
on_room_info_waterfall.add(ignore.create_flooignore, dir_to_share)
on_room_info_waterfall.add(lambda: G.AGENT.upload(file_to_share or dir_to_share, on_room_info_msg))
def on_done(owner):
self.window.run_command('floobits_create_workspace', {
'workspace_name': workspace_name,
'dir_to_share': dir_to_share,
'api_args': self.api_args,
'owner': owner[0],
})
try:
r = api.get_orgs_can_admin()
except IOError as e:
return sublime.error_message('Error getting org list: %s' % str(e))
if r.code >= 400 or len(r.body) == 0:
return on_done([G.USERNAME])
orgs = [[org['name'], 'Create workspace under %s' % org['name']] for org in r.body]
orgs.insert(0, [G.USERNAME, 'Create workspace under %s' % G.USERNAME])
self.window.show_quick_panel(orgs, lambda index: index < 0 or on_done(orgs[index]))
def share_dir(dir_to_share):
dir_to_share = os.path.expanduser(dir_to_share)
dir_to_share = utils.unfuck_path(dir_to_share)
dir_to_share = os.path.abspath(dir_to_share)
workspace_name = os.path.basename(dir_to_share)
floo_workspace_dir = os.path.join(G.COLAB_DIR, G.USERNAME, workspace_name)
if os.path.isfile(dir_to_share):
return msg.error('give me a directory please')
if not os.path.isdir(dir_to_share):
return msg.error('The directory %s doesn\'t appear to exist' %
dir_to_share)
floo_file = os.path.join(dir_to_share, '.floo')
# look for the .floo file for hints about previous behavior
info = {}
try:
floo_info = open(floo_file, 'rb').read().decode('utf-8')
info = json.loads(floo_info)
except (IOError, OSError):
pass
except Exception:
msg.warn("couldn't read the floo_info file: %s" % floo_file)
workspace_url = info.get('url')
if workspace_url:
try:
result = utils.parse_url(workspace_url)
except Exception as e:
msg.error(str(e))
else:
workspace_name = result['workspace']
floo_workspace_dir = os.path.join(G.COLAB_DIR, result['owner'],
result['workspace'])
# they have previously joined the workspace
if os.path.realpath(floo_workspace_dir) == os.path.realpath(
dir_to_share):
# it could have been deleted, try to recreate it if possible
# TODO: org or something here?
if result['owner'] == G.USERNAME:
try:
api.create_workspace({'name': workspace_name})
msg.debug('Created workspace %s' % workspace_url)
except Exception as e:
msg.debug('Tried to create workspace' + str(e))
# they wanted to share teh dir, so always share it
return join_workspace(
workspace_url,
lambda x: agent.protocol.create_buf(dir_to_share,
force=True))
# link to what they want to share
try:
utils.mkdir(os.path.dirname(floo_workspace_dir))
os.symlink(dir_to_share, floo_workspace_dir)
except OSError as e:
if e.errno != 17:
raise
except Exception as e:
return msg.error("Couldn't create symlink from %s to %s: %s" %
(dir_to_share, floo_workspace_dir, str(e)))
# make & join workspace
create_workspace(workspace_name, floo_workspace_dir, dir_to_share)
def _on_room_info(self, data):
self.reset()
G.JOINED_WORKSPACE = True
self.workspace_info = data
G.PERMS = data['perms']
if 'patch' not in data['perms']:
msg.log('No patch permission. Setting buffers to read-only')
should_send = yield self.ok_cancel_dialog, 'You don\'t have permission to edit this workspace. All files will be read-only.\n\nDo you want to request edit permission?'
if should_send:
self.send({'name': 'request_perms', 'perms': ['edit_room']})
floo_json = {
'url': utils.to_workspace_url({
'owner': self.owner,
'workspace': self.workspace,
'host': self.proto.host,
'port': self.proto.port,
'secure': self.proto.secure,
})
}
utils.update_floo_file(os.path.join(G.PROJECT_PATH, '.floo'), floo_json)
changed_bufs = []
missing_bufs = []
for buf_id, buf in data['bufs'].items():
buf_id = int(buf_id) # json keys must be strings
buf_path = utils.get_full_path(buf['path'])
new_dir = os.path.dirname(buf_path)
utils.mkdir(new_dir)
self.bufs[buf_id] = buf
self.paths_to_ids[buf['path']] = buf_id
view = self.get_view(buf_id)
if view and not view.is_loading() and buf['encoding'] == 'utf8':
view_text = view.get_text()
view_md5 = hashlib.md5(view_text.encode('utf-8')).hexdigest()
if view_md5 == buf['md5']:
msg.debug('md5 sum matches view. not getting buffer %s' % buf['path'])
buf['buf'] = view_text
G.VIEW_TO_HASH[view.native_id] = view_md5
elif self.should_get_bufs:
changed_bufs.append(buf_id)
else:
try:
buf_fd = open(buf_path, 'rb')
buf_buf = buf_fd.read()
md5 = hashlib.md5(buf_buf).hexdigest()
if md5 == buf['md5']:
msg.debug('md5 sum matches. not getting buffer %s' % buf['path'])
if buf['encoding'] == 'utf8':
buf_buf = buf_buf.decode('utf-8')
buf['buf'] = buf_buf
elif self.should_get_bufs:
changed_bufs.append(buf_id)
except Exception as e:
msg.debug('Error calculating md5:', e)
missing_bufs.append(buf_id)
if changed_bufs and self.should_get_bufs:
if len(changed_bufs) > 4:
prompt = '%s local files are different from the workspace. Overwrite your local files?' % len(changed_bufs)
else:
prompt = 'Overwrite the following local files?\n'
for buf_id in changed_bufs:
prompt += '\n%s' % self.bufs[buf_id]['path']
stomp_local = yield self.ok_cancel_dialog, prompt
for buf_id in changed_bufs:
if stomp_local:
self.get_buf(buf_id)
self.save_on_get_bufs.add(buf_id)
else:
buf = self.bufs[buf_id]
# TODO: this is inefficient. we just read the file 20 lines ago
self.upload(utils.get_full_path(buf['path']))
for buf_id in missing_bufs:
self.get_buf(buf_id)
success_msg = 'Successfully joined workspace %s/%s' % (self.owner, self.workspace)
msg.log(success_msg)
editor.status_message(success_msg)
temp_data = data.get('temp_data', {})
hangout = temp_data.get('hangout', {})
hangout_url = hangout.get('url')
if hangout_url:
self.prompt_join_hangout(hangout_url)
data = utils.get_persistent_data()
data['recent_workspaces'].insert(0, {"url": self.workspace_url})
utils.update_persistent_data(data)
utils.add_workspace_to_persistent_json(self.owner, self.workspace, self.workspace_url, G.PROJECT_PATH)
self.emit("room_info")
def _on_room_info(self, data):
self.reset()
self.joined_workspace = True
self.workspace_info = data
G.PERMS = data['perms']
if 'patch' not in data['perms']:
no_perms_msg = '''You don't have permission to edit this workspace. All files will be read-only.'''
msg.log('No patch permission. Setting buffers to read-only')
if 'request_perm' in data['perms']:
should_send = yield self.ok_cancel_dialog, no_perms_msg + '\nDo you want to request edit permission?'
if should_send:
self.send({'name': 'request_perms', 'perms': ['edit_room']})
else:
editor.error_message(no_perms_msg)
floo_json = {
'url': utils.to_workspace_url({
'owner': self.owner,
'workspace': self.workspace,
'host': self.proto.host,
'port': self.proto.port,
'secure': self.proto.secure,
})
}
utils.update_floo_file(os.path.join(G.PROJECT_PATH, '.floo'), floo_json)
changed_bufs = []
missing_bufs = []
for buf_id, buf in data['bufs'].items():
buf_id = int(buf_id) # json keys must be strings
buf_path = utils.get_full_path(buf['path'])
new_dir = os.path.dirname(buf_path)
utils.mkdir(new_dir)
self.bufs[buf_id] = buf
self.paths_to_ids[buf['path']] = buf_id
view = self.get_view(buf_id)
if view and not view.is_loading() and buf['encoding'] == 'utf8':
view_text = view.get_text()
view_md5 = hashlib.md5(view_text.encode('utf-8')).hexdigest()
buf['buf'] = view_text
buf['view'] = view
G.VIEW_TO_HASH[view.native_id] = view_md5
if view_md5 == buf['md5']:
msg.debug('md5 sum matches view. not getting buffer %s' % buf['path'])
else:
changed_bufs.append(buf_id)
buf['md5'] = view_md5
else:
try:
if buf['encoding'] == "utf8":
if io:
buf_fd = io.open(buf_path, 'Urt', encoding='utf8')
buf_buf = buf_fd.read()
else:
buf_fd = open(buf_path, 'rb')
buf_buf = buf_fd.read().decode('utf-8').replace('\r\n', '\n')
md5 = hashlib.md5(buf_buf.encode('utf-8')).hexdigest()
else:
buf_fd = open(buf_path, 'rb')
buf_buf = buf_fd.read()
md5 = hashlib.md5(buf_buf).hexdigest()
buf_fd.close()
buf['buf'] = buf_buf
if md5 == buf['md5']:
msg.debug('md5 sum matches. not getting buffer %s' % buf['path'])
else:
msg.debug('md5 differs. possibly getting buffer later %s' % buf['path'])
changed_bufs.append(buf_id)
buf['md5'] = md5
except Exception as e:
msg.debug('Error calculating md5 for %s, %s' % (buf['path'], e))
missing_bufs.append(buf_id)
stomp_local = self.should_get_bufs
if stomp_local and (changed_bufs or missing_bufs):
changed = [self.bufs[buf_id] for buf_id in changed_bufs]
missing = [self.bufs[buf_id] for buf_id in missing_bufs]
choice = yield self.stomp_prompt, changed, missing
if choice not in [0, 1]:
self.stop()
return
stomp_local = bool(choice)
for buf_id in changed_bufs:
buf = self.bufs[buf_id]
if stomp_local:
self.get_buf(buf_id, buf.get('view'))
self.save_on_get_bufs.add(buf_id)
else:
self._upload(utils.get_full_path(buf['path']), buf['buf'])
for buf_id in missing_bufs:
buf = self.bufs[buf_id]
if stomp_local:
self.get_buf(buf_id, buf.get('view'))
self.save_on_get_bufs.add(buf_id)
else:
self.send({
'name': 'delete_buf',
'id': buf['id'],
})
success_msg = 'Successfully joined workspace %s/%s' % (self.owner, self.workspace)
msg.log(success_msg)
editor.status_message(success_msg)
temp_data = data.get('temp_data', {})
hangout = temp_data.get('hangout', {})
hangout_url = hangout.get('url')
if hangout_url:
self.prompt_join_hangout(hangout_url)
data = utils.get_persistent_data()
data['recent_workspaces'].insert(0, {"url": self.workspace_url})
utils.update_persistent_data(data)
utils.add_workspace_to_persistent_json(self.owner, self.workspace, self.workspace_url, G.PROJECT_PATH)
self.emit("room_info")
def on_input(self, dir_to_share):
file_to_share = None
dir_to_share = os.path.expanduser(dir_to_share)
dir_to_share = os.path.realpath(utils.unfuck_path(dir_to_share))
workspace_name = os.path.basename(dir_to_share)
workspace_url = None
print(G.COLAB_DIR, G.USERNAME, workspace_name)
def find_workspace(workspace_url):
try:
api.get_workspace_by_url(workspace_url)
except HTTPError:
try:
result = utils.parse_url(workspace_url)
d = utils.get_persistent_data()
del d['workspaces'][result['owner']][result['name']]
utils.update_persistent_data(d)
except Exception as e:
msg.debug(unicode(e))
return False
on_room_info_waterfall.add(on_room_info_msg)
on_room_info_waterfall.add(ignore.create_flooignore, dir_to_share)
on_room_info_waterfall.add(Listener.create_buf, dir_to_share)
return True
if os.path.isfile(dir_to_share):
file_to_share = dir_to_share
dir_to_share = os.path.dirname(dir_to_share)
else:
try:
utils.mkdir(dir_to_share)
except Exception:
return sublime.error_message(
'The directory %s doesn\'t exist and I can\'t make it.' %
dir_to_share)
floo_file = os.path.join(dir_to_share, '.floo')
info = {}
try:
floo_info = open(floo_file, 'rb').read().decode('utf-8')
info = json.loads(floo_info)
except (IOError, OSError):
pass
except Exception:
print('Couldn\'t read the floo_info file: %s' % floo_file)
workspace_url = info.get('url')
try:
result = utils.parse_url(workspace_url)
except Exception:
workspace_url = None
if workspace_url and find_workspace(workspace_url):
add_workspace_to_persistent_json(result['owner'],
result['workspace'],
workspace_url, dir_to_share)
return self.window.run_command(
'floobits_join_workspace', {
'workspace_url': workspace_url,
'agent_conn_kwargs': {
'get_bufs': False
}
})
for owner, workspaces in utils.get_persistent_data(
)['workspaces'].items():
for name, workspace in workspaces.items():
if workspace['path'] == dir_to_share:
workspace_url = workspace['url']
if find_workspace(workspace_url):
return self.window.run_command(
'floobits_join_workspace', {
'workspace_url': workspace_url,
'agent_conn_kwargs': {
'get_bufs': False
}
})
# make & join workspace
on_room_info_waterfall.add(ignore.create_flooignore, dir_to_share)
on_room_info_waterfall.add(Listener.create_buf, file_to_share
or dir_to_share)
def on_done(owner):
self.window.run_command(
'floobits_create_workspace', {
'workspace_name': workspace_name,
'dir_to_share': dir_to_share,
'api_args': self.api_args,
'owner': owner[0],
})
orgs = api.get_orgs_can_admin()
orgs = json.loads(orgs.read().decode('utf-8'))
if len(orgs) == 0:
return on_done([G.USERNAME])
orgs = [[org['name'],
'Create workspace under %s' % org['name']] for org in orgs]
orgs.insert(0, [G.USERNAME, 'Create workspace under %s' % G.USERNAME])
self.window.show_quick_panel(
orgs, lambda index: index < 0 or on_done(orgs[index]))
def share_dir(self, dir_to_share, perms=None):
dir_to_share = os.path.expanduser(dir_to_share)
dir_to_share = utils.unfuck_path(dir_to_share)
workspace_name = os.path.basename(dir_to_share)
G.PROJECT_PATH = os.path.realpath(dir_to_share)
msg.debug('%s %s %s' % (G.USERNAME, workspace_name, G.PROJECT_PATH))
if os.path.isfile(dir_to_share):
return msg.error('%s is a file. Give me a directory please.' % dir_to_share)
try:
utils.mkdir(dir_to_share)
except Exception:
return msg.error("The directory %s doesn't exist and I can't make it." % dir_to_share)
floo_file = os.path.join(dir_to_share, '.floo')
info = {}
try:
floo_info = open(floo_file, 'rb').read().decode('utf-8')
info = json.loads(floo_info)
except (IOError, OSError):
pass
except Exception:
msg.debug("Couldn't read the floo_info file: %s" % floo_file)
workspace_url = info.get('url')
if workspace_url:
try:
result = utils.parse_url(workspace_url)
except Exception as e:
msg.error(str(e))
else:
workspace_name = result['workspace']
try:
# TODO: blocking. beachballs sublime 2 if API is super slow
api.get_workspace_by_url(workspace_url)
except HTTPError:
workspace_url = None
workspace_name = os.path.basename(dir_to_share)
else:
utils.add_workspace_to_persistent_json(result['owner'], result['workspace'], workspace_url, dir_to_share)
workspace_url = utils.get_workspace_by_path(dir_to_share) or workspace_url
if workspace_url:
try:
api.get_workspace_by_url(workspace_url)
except HTTPError:
pass
else:
return self.remote_connect(workspace_url, lambda this: this.protocol.create_buf(dir_to_share))
def on_done(data, choices=None):
self.create_workspace({}, workspace_name, dir_to_share, owner=data.get('response'), perms=perms)
orgs = api.get_orgs_can_admin()
orgs = json.loads(orgs.read().decode('utf-8'))
if len(orgs) == 0:
return on_done({'response': G.USERNAME})
i = 0
choices = []
choices.append([G.USERNAME, i])
for o in orgs:
i += 1
choices.append([o['name'], i])
self.get_input('Create workspace for [press tab for completion]: ', '', on_done, choices=choices)
def on_input(self, dir_to_share):
file_to_share = None
dir_to_share = os.path.expanduser(dir_to_share)
dir_to_share = os.path.realpath(utils.unfuck_path(dir_to_share))
workspace_name = os.path.basename(dir_to_share)
workspace_url = None
# TODO: use prejoin_workspace instead
def find_workspace(workspace_url):
r = api.get_workspace_by_url(workspace_url)
if r.code < 400:
return r
try:
result = utils.parse_url(workspace_url)
d = utils.get_persistent_data()
del d['workspaces'][result['owner']][result['name']]
utils.update_persistent_data(d)
except Exception as e:
msg.debug(str_e(e))
def join_workspace(workspace_url):
try:
w = find_workspace(workspace_url)
except Exception as e:
sublime.error_message('Error: %s' % str_e(e))
return False
if not w:
return False
msg.debug('workspace: %s', json.dumps(w.body))
# if self.api_args:
anon_perms = w.body.get('perms', {}).get('AnonymousUser', [])
new_anon_perms = self.api_args.get('perms').get(
'AnonymousUser', [])
# TODO: warn user about making a private workspace public
if set(anon_perms) != set(new_anon_perms):
msg.debug(str(anon_perms), str(new_anon_perms))
w.body['perms']['AnonymousUser'] = new_anon_perms
response = api.update_workspace(workspace_url, w.body)
msg.debug(str(response.body))
utils.add_workspace_to_persistent_json(w.body['owner'],
w.body['name'],
workspace_url, dir_to_share)
self.window.run_command('floobits_join_workspace',
{'workspace_url': workspace_url})
return True
if os.path.isfile(dir_to_share):
file_to_share = dir_to_share
dir_to_share = os.path.dirname(dir_to_share)
try:
utils.mkdir(dir_to_share)
except Exception:
sublime.error_message(
'The directory %s doesn\'t exist and I can\'t create it.' %
dir_to_share)
return
floo_file = os.path.join(dir_to_share, '.floo')
info = {}
try:
floo_info = open(floo_file, 'r').read()
info = json.loads(floo_info)
except (IOError, OSError):
pass
except Exception:
msg.error('Couldn\'t read the floo_info file: %s' % floo_file)
workspace_url = info.get('url')
try:
utils.parse_url(workspace_url)
except Exception:
workspace_url = None
if workspace_url and join_workspace(workspace_url):
return
for owner, workspaces in utils.get_persistent_data(
)['workspaces'].items():
for name, workspace in workspaces.items():
if workspace['path'] == dir_to_share:
workspace_url = workspace['url']
if join_workspace(workspace_url):
return
auth = yield editor.select_auth, self.window, G.AUTH
if not auth:
return
username = auth.get('username')
host = auth['host']
def on_done(owner):
msg.log('Colab dir: %s, Username: %s, Workspace: %s/%s' %
(G.COLAB_DIR, username, owner[0], workspace_name))
self.window.run_command(
'floobits_create_workspace', {
'workspace_name': workspace_name,
'dir_to_share': dir_to_share,
'upload': file_to_share or dir_to_share,
'api_args': self.api_args,
'owner': owner[0],
'host': host,
})
try:
r = api.get_orgs_can_admin(host)
except IOError as e:
sublime.error_message('Error getting org list: %s' % str_e(e))
return
if r.code >= 400 or len(r.body) == 0:
on_done([username])
return
orgs = [[org['name'],
'Create workspace owned by %s' % org['name']]
for org in r.body]
orgs.insert(0, [username, 'Create workspace owned by %s' % username])
self.window.show_quick_panel(
orgs, lambda index: index < 0 or on_done(orgs[index]))
def _on_room_info(self, data):
self.reset()
self.joined_workspace = True
self.workspace_info = data
G.PERMS = data['perms']
self.proto.reset_retries()
if G.OUTBOUND_FILTERING:
msg.error('Detected outbound port blocking! See https://floobits.com/help/network for more info.')
read_only = False
if 'patch' not in data['perms']:
read_only = True
no_perms_msg = '''You don't have permission to edit this workspace. All files will be read-only.'''
msg.log('No patch permission. Setting buffers to read-only')
if 'request_perm' in data['perms']:
should_send = yield self.ok_cancel_dialog, no_perms_msg + '\nDo you want to request edit permission?'
# TODO: wait for perms to be OK'd/denied before uploading or bailing
if should_send:
self.send({'name': 'request_perms', 'perms': ['edit_room']})
else:
if G.EXPERT_MODE:
editor.status_message(no_perms_msg)
else:
editor.error_message(no_perms_msg)
floo_json = {
'url': utils.to_workspace_url({
'owner': self.owner,
'workspace': self.workspace,
'host': self.proto.host,
'port': self.proto.port,
'secure': self.proto.secure,
})
}
utils.update_floo_file(os.path.join(G.PROJECT_PATH, '.floo'), floo_json)
utils.update_recent_workspaces(self.workspace_url)
changed_bufs = []
missing_bufs = []
new_files = set()
ig = ignore.create_ignore_tree(G.PROJECT_PATH)
G.IGNORE = ig
if not read_only:
new_files = set([utils.to_rel_path(x) for x in ig.list_paths()])
for buf_id, buf in data['bufs'].items():
buf_id = int(buf_id) # json keys must be strings
buf_path = utils.get_full_path(buf['path'])
new_dir = os.path.dirname(buf_path)
utils.mkdir(new_dir)
self.bufs[buf_id] = buf
self.paths_to_ids[buf['path']] = buf_id
view = self.get_view(buf_id)
if view and not view.is_loading() and buf['encoding'] == 'utf8':
view_text = view.get_text()
view_md5 = hashlib.md5(view_text.encode('utf-8')).hexdigest()
buf['buf'] = view_text
buf['view'] = view
G.VIEW_TO_HASH[view.native_id] = view_md5
if view_md5 == buf['md5']:
msg.debug('md5 sum matches view. not getting buffer ', buf['path'])
else:
changed_bufs.append(buf)
buf['md5'] = view_md5
continue
try:
if buf['encoding'] == 'utf8':
if io:
buf_fd = io.open(buf_path, 'Urt', encoding='utf8')
buf_buf = buf_fd.read()
else:
buf_fd = open(buf_path, 'rb')
buf_buf = buf_fd.read().decode('utf-8').replace('\r\n', '\n')
md5 = hashlib.md5(buf_buf.encode('utf-8')).hexdigest()
else:
buf_fd = open(buf_path, 'rb')
buf_buf = buf_fd.read()
md5 = hashlib.md5(buf_buf).hexdigest()
buf_fd.close()
buf['buf'] = buf_buf
if md5 == buf['md5']:
msg.debug('md5 sum matches. not getting buffer ', buf['path'])
else:
msg.debug('md5 differs. possibly getting buffer later ', buf['path'])
changed_bufs.append(buf)
buf['md5'] = md5
except Exception as e:
msg.debug('Error calculating md5 for ', buf['path'], ', ', str_e(e))
missing_bufs.append(buf)
ignored = []
for p, buf_id in self.paths_to_ids.items():
if p not in new_files:
ignored.append(p)
new_files.discard(p)
if self.action == utils.JOIN_ACTION.UPLOAD:
yield self._initial_upload, ig, missing_bufs, changed_bufs
# TODO: maybe use org name here
who = 'Your friends'
anon_perms = G.AGENT.workspace_info.get('anon_perms')
if 'get_buf' in anon_perms:
who = 'Anyone'
_msg = 'You are sharing:\n\n%s\n\n%s can join your workspace at:\n\n%s' % (G.PROJECT_PATH, who, G.AGENT.workspace_url)
# Workaround for horrible Sublime Text bug
utils.set_timeout(editor.message_dialog, 0, _msg)
elif changed_bufs or missing_bufs or new_files:
# TODO: handle readonly here
if self.action == utils.JOIN_ACTION.PROMPT:
stomp_local = yield self.stomp_prompt, changed_bufs, missing_bufs, list(new_files), ignored
if stomp_local not in [0, 1]:
self.stop()
return
elif self.action == utils.JOIN_ACTION.DOWNLOAD:
stomp_local = True
else:
# This should never happen
assert False
return
if stomp_local:
for buf in changed_bufs:
self.get_buf(buf['id'], buf.get('view'))
self.save_on_get_bufs.add(buf['id'])
for buf in missing_bufs:
self.get_buf(buf['id'], buf.get('view'))
self.save_on_get_bufs.add(buf['id'])
else:
yield self._initial_upload, ig, missing_bufs, changed_bufs
success_msg = 'Successfully joined workspace %s/%s' % (self.owner, self.workspace)
msg.log(success_msg)
editor.status_message(success_msg)
data = utils.get_persistent_data()
data['recent_workspaces'].insert(0, {"url": self.workspace_url})
utils.update_persistent_data(data)
utils.add_workspace_to_persistent_json(self.owner, self.workspace, self.workspace_url, G.PROJECT_PATH)
temp_data = data.get('temp_data', {})
hangout = temp_data.get('hangout', {})
hangout_url = hangout.get('url')
if hangout_url:
self.prompt_join_hangout(hangout_url)
self.emit("room_info")
def floobits_join_workspace(workspace_url, d='', upload_path=None):
editor.line_endings = _get_line_endings()
msg.debug('workspace url is %s' % workspace_url)
try:
result = utils.parse_url(workspace_url)
except Exception as e:
return msg.error(str(e))
if d:
utils.mkdir(d)
else:
try:
d = utils.get_persistent_data()['workspaces'][result['owner']][
result['workspace']]['path']
except Exception:
d = os.path.realpath(
os.path.join(G.COLAB_DIR, result['owner'],
result['workspace']))
prompt = 'Save workspace files to: '
if not os.path.isdir(d):
while True:
d = vim_input(prompt, d, 'dir')
if d == '':
continue
d = os.path.realpath(os.path.expanduser(d))
if os.path.isfile(d):
prompt = '%s is not a directory. Enter an existing path or a path I can create: ' % d
continue
if not os.path.isdir(d):
try:
utils.mkdir(d)
except Exception as e:
prompt = 'Couldn\'t make dir %s: %s ' % (d, str(e))
continue
break
d = os.path.realpath(os.path.abspath(d) + os.sep)
try:
utils.add_workspace_to_persistent_json(result['owner'],
result['workspace'],
workspace_url, d)
except Exception as e:
return msg.error('Error adding workspace to persistent.json: %s' %
str(e))
G.PROJECT_PATH = d
vim.command('cd %s' % G.PROJECT_PATH)
msg.debug('Joining workspace %s' % workspace_url)
floobits_stop_everything()
try:
conn = VimHandler(result['owner'], result['workspace'])
if upload_path:
conn.once('room_info', lambda: G.AGENT.upload(upload_path))
reactor.connect(conn, result['host'], result['port'], result['secure'])
except Exception as e:
msg.error(str(e))
tb = traceback.format_exc()
msg.debug(tb)
if not G.TIMERS:
start_event_loop()
def on_input(self, dir_to_share):
file_to_share = None
dir_to_share = os.path.expanduser(dir_to_share)
dir_to_share = os.path.realpath(utils.unfuck_path(dir_to_share))
workspace_name = os.path.basename(dir_to_share)
workspace_url = None
print(G.COLAB_DIR, G.USERNAME, workspace_name)
def find_workspace(workspace_url):
try:
api.get_workspace_by_url(workspace_url)
except HTTPError:
try:
result = utils.parse_url(workspace_url)
d = utils.get_persistent_data()
del d['workspaces'][result['owner']][result['name']]
utils.update_persistent_data(d)
except Exception as e:
msg.debug(unicode(e))
return False
on_room_info_waterfall.add(on_room_info_msg)
on_room_info_waterfall.add(ignore.create_flooignore, dir_to_share)
on_room_info_waterfall.add(Listener.create_buf, dir_to_share)
return True
if os.path.isfile(dir_to_share):
file_to_share = dir_to_share
dir_to_share = os.path.dirname(dir_to_share)
else:
try:
utils.mkdir(dir_to_share)
except Exception:
return sublime.error_message('The directory %s doesn\'t exist and I can\'t make it.' % dir_to_share)
floo_file = os.path.join(dir_to_share, '.floo')
info = {}
try:
floo_info = open(floo_file, 'rb').read().decode('utf-8')
info = json.loads(floo_info)
except (IOError, OSError):
pass
except Exception:
print('Couldn\'t read the floo_info file: %s' % floo_file)
workspace_url = info.get('url')
try:
result = utils.parse_url(workspace_url)
except Exception:
workspace_url = None
if workspace_url and find_workspace(workspace_url):
add_workspace_to_persistent_json(result['owner'], result['workspace'], workspace_url, dir_to_share)
return self.window.run_command('floobits_join_workspace', {
'workspace_url': workspace_url,
'agent_conn_kwargs': {'get_bufs': False}})
for owner, workspaces in utils.get_persistent_data()['workspaces'].items():
for name, workspace in workspaces.items():
if workspace['path'] == dir_to_share:
workspace_url = workspace['url']
if find_workspace(workspace_url):
return self.window.run_command('floobits_join_workspace', {
'workspace_url': workspace_url,
'agent_conn_kwargs': {'get_bufs': False}})
# make & join workspace
on_room_info_waterfall.add(ignore.create_flooignore, dir_to_share)
on_room_info_waterfall.add(Listener.create_buf, file_to_share or dir_to_share)
def on_done(owner):
self.window.run_command('floobits_create_workspace', {
'workspace_name': workspace_name,
'dir_to_share': dir_to_share,
'api_args': self.api_args,
'owner': owner[0],
})
orgs = api.get_orgs_can_admin()
orgs = json.loads(orgs.read().decode('utf-8'))
if len(orgs) == 0:
return on_done([G.USERNAME])
orgs = [[org['name'], 'Create workspace under %s' % org['name']] for org in orgs]
orgs.insert(0, [G.USERNAME, 'Create workspace under %s' % G.USERNAME])
self.window.show_quick_panel(orgs, lambda index: index < 0 or on_done(orgs[index]))
def _on_room_info(self, data):
self.reset()
self.joined_workspace = True
self.workspace_info = data
G.PERMS = data['perms']
if 'patch' not in data['perms']:
no_perms_msg = '''You don't have permission to edit this workspace. All files will be read-only.'''
msg.log('No patch permission. Setting buffers to read-only')
if 'request_perm' in data['perms']:
should_send = yield self.ok_cancel_dialog, no_perms_msg + '\nDo you want to request edit permission?'
if should_send:
self.send({
'name': 'request_perms',
'perms': ['edit_room']
})
else:
editor.error_message(no_perms_msg)
floo_json = {
'url':
utils.to_workspace_url({
'owner': self.owner,
'workspace': self.workspace,
'host': self.proto.host,
'port': self.proto.port,
'secure': self.proto.secure,
})
}
utils.update_floo_file(os.path.join(G.PROJECT_PATH, '.floo'),
floo_json)
changed_bufs = []
missing_bufs = []
for buf_id, buf in data['bufs'].items():
buf_id = int(buf_id) # json keys must be strings
buf_path = utils.get_full_path(buf['path'])
new_dir = os.path.dirname(buf_path)
utils.mkdir(new_dir)
self.bufs[buf_id] = buf
self.paths_to_ids[buf['path']] = buf_id
view = self.get_view(buf_id)
if view and not view.is_loading() and buf['encoding'] == 'utf8':
view_text = view.get_text()
view_md5 = hashlib.md5(view_text.encode('utf-8')).hexdigest()
buf['buf'] = view_text
buf['view'] = view
G.VIEW_TO_HASH[view.native_id] = view_md5
if view_md5 == buf['md5']:
msg.debug('md5 sum matches view. not getting buffer %s' %
buf['path'])
else:
changed_bufs.append(buf_id)
buf['md5'] = view_md5
else:
try:
if buf['encoding'] == "utf8":
if io:
buf_fd = io.open(buf_path, 'Urt', encoding='utf8')
buf_buf = buf_fd.read()
else:
buf_fd = open(buf_path, 'rb')
buf_buf = buf_fd.read().decode('utf-8').replace(
'\r\n', '\n')
md5 = hashlib.md5(buf_buf.encode('utf-8')).hexdigest()
else:
buf_fd = open(buf_path, 'rb')
buf_buf = buf_fd.read()
md5 = hashlib.md5(buf_buf).hexdigest()
buf_fd.close()
buf['buf'] = buf_buf
if md5 == buf['md5']:
msg.debug('md5 sum matches. not getting buffer %s' %
buf['path'])
else:
msg.debug(
'md5 differs. possibly getting buffer later %s' %
buf['path'])
changed_bufs.append(buf_id)
buf['md5'] = md5
except Exception as e:
msg.debug('Error calculating md5 for %s, %s' %
(buf['path'], e))
missing_bufs.append(buf_id)
stomp_local = self.should_get_bufs
if stomp_local and (changed_bufs or missing_bufs):
changed = [self.bufs[buf_id] for buf_id in changed_bufs]
missing = [self.bufs[buf_id] for buf_id in missing_bufs]
choice = yield self.stomp_prompt, changed, missing
if choice not in [0, 1]:
self.stop()
return
stomp_local = bool(choice)
for buf_id in changed_bufs:
buf = self.bufs[buf_id]
if stomp_local:
self.get_buf(buf_id, buf.get('view'))
self.save_on_get_bufs.add(buf_id)
else:
self._upload(utils.get_full_path(buf['path']), buf['buf'])
for buf_id in missing_bufs:
buf = self.bufs[buf_id]
if stomp_local:
self.get_buf(buf_id, buf.get('view'))
self.save_on_get_bufs.add(buf_id)
else:
self.send({
'name': 'delete_buf',
'id': buf['id'],
})
success_msg = 'Successfully joined workspace %s/%s' % (self.owner,
self.workspace)
msg.log(success_msg)
editor.status_message(success_msg)
temp_data = data.get('temp_data', {})
hangout = temp_data.get('hangout', {})
hangout_url = hangout.get('url')
if hangout_url:
self.prompt_join_hangout(hangout_url)
data = utils.get_persistent_data()
data['recent_workspaces'].insert(0, {"url": self.workspace_url})
utils.update_persistent_data(data)
utils.add_workspace_to_persistent_json(self.owner, self.workspace,
self.workspace_url,
G.PROJECT_PATH)
self.emit("room_info")
def share_dir(self, context, dir_to_share, api_args):
utils.reload_settings()
if not utils.can_auth():
success = yield self.create_or_link_account, context, G.DEFAULT_HOST, False
if not success:
return
utils.reload_settings()
dir_to_share = os.path.expanduser(dir_to_share)
dir_to_share = os.path.realpath(dir_to_share)
dir_to_share = utils.unfuck_path(dir_to_share)
if os.path.isfile(dir_to_share):
dir_to_share = os.path.dirname(dir_to_share)
workspace_name = os.path.basename(dir_to_share)
msg.debug('', workspace_name, dir_to_share)
if os.path.isfile(dir_to_share):
dir_to_share = os.path.dirname(dir_to_share)
try:
utils.mkdir(dir_to_share)
except Exception:
msg.error("The directory", dir_to_share, "doesn't exist and I can't create it.")
return
info = utils.read_floo_file(dir_to_share)
def prejoin(workspace_url):
try:
return self.prejoin_workspace(workspace_url, dir_to_share, api_args)
except ValueError:
pass
workspace_url = info.get('url')
if workspace_url:
parsed_url = prejoin(workspace_url)
if parsed_url:
self.remote_connect(context, parsed_url['host'], parsed_url['owner'], parsed_url['workspace'], dir_to_share)
return
parsed_url = utils.get_workspace_by_path(dir_to_share, prejoin)
if parsed_url:
self.remote_connect(context, parsed_url['host'], parsed_url['owner'], parsed_url['workspace'], dir_to_share)
return
host = yield self._get_host, context
if not host:
return
try:
r = api.get_orgs_can_admin(host)
except IOError as e:
editor.error_message('Error getting org list: %s' % str_e(e))
return
choices = [G.AUTH[host]['username']]
if r.code >= 400:
editor.error_message('Error getting org list: %s' % r.body)
elif r.body:
choices += [org['name'] for org in r.body]
if len(choices) == 1:
owner = choices[0]
else:
little = ['Create workspace owned by %s' % s for s in choices]
(owner, index) = yield self.user_select, context, 'Create workspace owned by', choices, little
if not owner:
return
self.create_workspace(context, host, owner, workspace_name, api_args, dir_to_share)
def on_input(self, dir_to_share):
file_to_share = None
dir_to_share = os.path.expanduser(dir_to_share)
dir_to_share = os.path.realpath(utils.unfuck_path(dir_to_share))
workspace_name = os.path.basename(dir_to_share)
workspace_url = None
# TODO: use prejoin_workspace instead
def find_workspace(workspace_url):
r = api.get_workspace_by_url(workspace_url)
if r.code < 400:
return r
try:
result = utils.parse_url(workspace_url)
d = utils.get_persistent_data()
del d['workspaces'][result['owner']][result['name']]
utils.update_persistent_data(d)
except Exception as e:
msg.debug(str_e(e))
def join_workspace(workspace_url):
try:
w = find_workspace(workspace_url)
except Exception as e:
sublime.error_message('Error: %s' % str_e(e))
return False
if not w:
return False
msg.debug('workspace: %s', json.dumps(w.body))
# if self.api_args:
anon_perms = w.body.get('perms', {}).get('AnonymousUser', [])
new_anon_perms = self.api_args.get('perms').get('AnonymousUser', [])
# TODO: warn user about making a private workspace public
if set(anon_perms) != set(new_anon_perms):
msg.debug(str(anon_perms), str(new_anon_perms))
w.body['perms']['AnonymousUser'] = new_anon_perms
response = api.update_workspace(workspace_url, w.body)
msg.debug(str(response.body))
utils.add_workspace_to_persistent_json(w.body['owner'], w.body['name'], workspace_url, dir_to_share)
self.window.run_command('floobits_join_workspace', {'workspace_url': workspace_url})
return True
if os.path.isfile(dir_to_share):
file_to_share = dir_to_share
dir_to_share = os.path.dirname(dir_to_share)
try:
utils.mkdir(dir_to_share)
except Exception:
sublime.error_message('The directory %s doesn\'t exist and I can\'t create it.' % dir_to_share)
return
floo_file = os.path.join(dir_to_share, '.floo')
info = {}
try:
floo_info = open(floo_file, 'r').read()
info = json.loads(floo_info)
except (IOError, OSError):
pass
except Exception:
msg.error('Couldn\'t read the floo_info file: %s' % floo_file)
workspace_url = info.get('url')
try:
utils.parse_url(workspace_url)
except Exception:
workspace_url = None
if workspace_url and join_workspace(workspace_url):
return
for owner, workspaces in utils.get_persistent_data()['workspaces'].items():
for name, workspace in workspaces.items():
if workspace['path'] == dir_to_share:
workspace_url = workspace['url']
if join_workspace(workspace_url):
return
auth = yield editor.select_auth, self.window, G.AUTH
if not auth:
return
username = auth.get('username')
host = auth['host']
def on_done(owner):
msg.log('Colab dir: %s, Username: %s, Workspace: %s/%s' % (G.COLAB_DIR, username, owner[0], workspace_name))
self.window.run_command('floobits_create_workspace', {
'workspace_name': workspace_name,
'dir_to_share': dir_to_share,
'upload': file_to_share or dir_to_share,
'api_args': self.api_args,
'owner': owner[0],
'host': host,
})
try:
r = api.get_orgs_can_admin(host)
except IOError as e:
sublime.error_message('Error getting org list: %s' % str_e(e))
return
if r.code >= 400 or len(r.body) == 0:
on_done([username])
return
orgs = [[org['name'], 'Create workspace owned by %s' % org['name']] for org in r.body]
orgs.insert(0, [username, 'Create workspace owned by %s' % username])
self.window.show_quick_panel(orgs, lambda index: index < 0 or on_done(orgs[index]))
def share_dir(self, context, dir_to_share, api_args):
utils.reload_settings()
if not utils.can_auth():
success = yield self.create_or_link_account, context, G.DEFAULT_HOST, False
if not success:
return
utils.reload_settings()
dir_to_share = os.path.expanduser(dir_to_share)
dir_to_share = os.path.realpath(dir_to_share)
dir_to_share = utils.unfuck_path(dir_to_share)
if os.path.isfile(dir_to_share):
dir_to_share = os.path.dirname(dir_to_share)
workspace_name = os.path.basename(dir_to_share)
msg.debug('', workspace_name, dir_to_share)
if os.path.isfile(dir_to_share):
dir_to_share = os.path.dirname(dir_to_share)
try:
utils.mkdir(dir_to_share)
except Exception:
msg.error("The directory", dir_to_share,
"doesn't exist and I can't create it.")
return
info = utils.read_floo_file(dir_to_share)
def prejoin(workspace_url):
try:
return self.prejoin_workspace(workspace_url, dir_to_share,
api_args)
except ValueError:
pass
workspace_url = info.get('url')
if workspace_url:
parsed_url = prejoin(workspace_url)
if parsed_url:
self.remote_connect(context, parsed_url['host'],
parsed_url['owner'],
parsed_url['workspace'], dir_to_share)
return
parsed_url = utils.get_workspace_by_path(dir_to_share, prejoin)
if parsed_url:
self.remote_connect(context, parsed_url['host'],
parsed_url['owner'], parsed_url['workspace'],
dir_to_share)
return
host = yield self._get_host, context
if not host:
return
try:
r = api.get_orgs_can_admin(host)
except IOError as e:
editor.error_message('Error getting org list: %s' % str_e(e))
return
choices = [G.AUTH[host]['username']]
if r.code >= 400:
editor.error_message('Error getting org list: %s' % r.body)
elif r.body:
choices += [org['name'] for org in r.body]
if len(choices) == 1:
owner = choices[0]
else:
little = ['Create workspace owned by %s' % s for s in choices]
(
owner, index
) = yield self.user_select, context, 'Create workspace owned by', choices, little
if not owner:
return
self.create_workspace(context, host, owner, workspace_name, api_args,
dir_to_share)
def _on_room_info(self, data):
self.joined_workspace = True
self.workspace_info = data
G.PERMS = data['perms']
self.proto.reset_retries()
if G.OUTBOUND_FILTERING:
msg.error(
'Detected outbound port blocking! See https://floobits.com/help/network for more info.'
)
read_only = False
if 'patch' not in data['perms']:
read_only = True
no_perms_msg = '''You don't have permission to edit this workspace. All files will be read-only.'''
msg.log('No patch permission. Setting buffers to read-only')
if 'request_perm' in data['perms']:
should_send = yield self.ok_cancel_dialog, no_perms_msg + '\nDo you want to request edit permission?'
# TODO: wait for perms to be OK'd/denied before uploading or bailing
if should_send:
self.send({
'name': 'request_perms',
'perms': ['edit_room']
})
else:
if G.EXPERT_MODE:
editor.status_message(no_perms_msg)
else:
editor.error_message(no_perms_msg)
floo_json = {
'url':
utils.to_workspace_url({
'owner': self.owner,
'workspace': self.workspace,
'host': self.proto.host,
'port': self.proto.port,
'secure': self.proto.secure,
})
}
utils.update_floo_file(os.path.join(G.PROJECT_PATH, '.floo'),
floo_json)
utils.update_recent_workspaces(self.workspace_url)
changed_bufs = []
missing_bufs = []
new_files = set()
ig = ignore.create_ignore_tree(G.PROJECT_PATH)
G.IGNORE = ig
if not read_only:
new_files = set([utils.to_rel_path(x) for x in ig.list_paths()])
for buf_id, buf in data['bufs'].items():
buf_id = int(buf_id) # json keys must be strings
buf_path = utils.get_full_path(buf['path'])
new_dir = os.path.dirname(buf_path)
utils.mkdir(new_dir)
self.bufs[buf_id] = buf
self.paths_to_ids[buf['path']] = buf_id
view = self.get_view(buf_id)
if view and not view.is_loading() and buf['encoding'] == 'utf8':
view_text = view.get_text()
view_md5 = hashlib.md5(view_text.encode('utf-8')).hexdigest()
buf['buf'] = view_text
buf['view'] = view
G.VIEW_TO_HASH[view.native_id] = view_md5
if view_md5 == buf['md5']:
msg.debug('md5 sum matches view. not getting buffer ',
buf['path'])
else:
changed_bufs.append(buf)
buf['md5'] = view_md5
continue
try:
if buf['encoding'] == 'utf8':
if io:
buf_fd = io.open(buf_path, 'Urt', encoding='utf8')
buf_buf = buf_fd.read()
else:
buf_fd = open(buf_path, 'rb')
buf_buf = buf_fd.read().decode('utf-8').replace(
'\r\n', '\n')
md5 = hashlib.md5(buf_buf.encode('utf-8')).hexdigest()
else:
buf_fd = open(buf_path, 'rb')
buf_buf = buf_fd.read()
md5 = hashlib.md5(buf_buf).hexdigest()
buf_fd.close()
buf['buf'] = buf_buf
if md5 == buf['md5']:
msg.debug('md5 sum matches. not getting buffer ',
buf['path'])
else:
msg.debug('md5 differs. possibly getting buffer later ',
buf['path'])
changed_bufs.append(buf)
buf['md5'] = md5
except Exception as e:
msg.debug('Error calculating md5 for ', buf['path'], ', ',
str_e(e))
missing_bufs.append(buf)
ignored = []
for p, buf_id in self.paths_to_ids.items():
if p not in new_files:
ignored.append(p)
new_files.discard(p)
if self.action == utils.JOIN_ACTION.UPLOAD:
yield self._initial_upload, ig, missing_bufs, changed_bufs
# TODO: maybe use org name here
who = 'Your friends'
anon_perms = G.AGENT.workspace_info.get('anon_perms')
if 'get_buf' in anon_perms:
who = 'Anyone'
_msg = 'You are sharing:\n\n%s\n\n%s can join your workspace at:\n\n%s' % (
G.PROJECT_PATH, who, G.AGENT.workspace_url)
# Workaround for horrible Sublime Text bug
utils.set_timeout(editor.message_dialog, 0, _msg)
elif changed_bufs or missing_bufs or new_files:
# TODO: handle readonly here
if self.action == utils.JOIN_ACTION.PROMPT:
stomp_local = yield self.stomp_prompt, changed_bufs, missing_bufs, list(
new_files), ignored
if stomp_local not in [0, 1]:
self.stop()
return
elif self.action == utils.JOIN_ACTION.DOWNLOAD:
stomp_local = True
else:
# This should never happen
assert False
return
if stomp_local:
for buf in changed_bufs:
self.get_buf(buf['id'], buf.get('view'))
self.save_on_get_bufs.add(buf['id'])
for buf in missing_bufs:
self.get_buf(buf['id'], buf.get('view'))
self.save_on_get_bufs.add(buf['id'])
else:
yield self._initial_upload, ig, missing_bufs, changed_bufs
success_msg = '%s@%s/%s: Joined!' % (self.username, self.owner,
self.workspace)
msg.log(success_msg)
editor.status_message(success_msg)
data = utils.get_persistent_data()
data['recent_workspaces'].insert(0, {"url": self.workspace_url})
utils.update_persistent_data(data)
utils.add_workspace_to_persistent_json(self.owner, self.workspace,
self.workspace_url,
G.PROJECT_PATH)
temp_data = data.get('temp_data', {})
hangout = temp_data.get('hangout', {})
hangout_url = hangout.get('url')
if hangout_url:
self.prompt_join_hangout(hangout_url)
self.emit("room_info")
def _on_room_info(self, data):
self.reset()
G.JOINED_WORKSPACE = True
self.workspace_info = data
G.PERMS = data['perms']
if 'patch' not in data['perms']:
msg.log('No patch permission. Setting buffers to read-only')
should_send = yield self.ok_cancel_dialog, '''You don't have permission to edit this workspace. All files will be read-only.
Do you want to request edit permission?'''
if should_send:
self.send({'name': 'request_perms', 'perms': ['edit_room']})
floo_json = {
'url': utils.to_workspace_url({
'owner': self.owner,
'workspace': self.workspace,
'host': self.proto.host,
'port': self.proto.port,
'secure': self.proto.secure,
})
}
utils.update_floo_file(os.path.join(G.PROJECT_PATH, '.floo'), floo_json)
changed_bufs = []
missing_bufs = []
for buf_id, buf in data['bufs'].items():
buf_id = int(buf_id) # json keys must be strings
buf_path = utils.get_full_path(buf['path'])
new_dir = os.path.dirname(buf_path)
utils.mkdir(new_dir)
self.bufs[buf_id] = buf
self.paths_to_ids[buf['path']] = buf_id
view = self.get_view(buf_id)
if view and not view.is_loading() and buf['encoding'] == 'utf8':
view_text = view.get_text()
view_md5 = hashlib.md5(view_text.encode('utf-8')).hexdigest()
if view_md5 == buf['md5']:
msg.debug('md5 sum matches view. not getting buffer %s' % buf['path'])
buf['buf'] = view_text
G.VIEW_TO_HASH[view.native_id] = view_md5
elif self.should_get_bufs:
changed_bufs.append(buf_id)
else:
try:
buf_fd = open(buf_path, 'rb')
buf_buf = buf_fd.read()
md5 = hashlib.md5(buf_buf).hexdigest()
if md5 == buf['md5']:
msg.debug('md5 sum matches. not getting buffer %s' % buf['path'])
if buf['encoding'] == 'utf8':
buf_buf = buf_buf.decode('utf-8')
buf['buf'] = buf_buf
elif self.should_get_bufs:
changed_bufs.append(buf_id)
except Exception as e:
msg.debug('Error calculating md5:', e)
missing_bufs.append(buf_id)
if changed_bufs and self.should_get_bufs:
if len(changed_bufs) > 4:
prompt = '%s local files are different from the workspace. Overwrite your local files?' % len(changed_bufs)
else:
prompt = 'Overwrite the following local files?\n'
for buf_id in changed_bufs:
prompt += '\n%s' % self.bufs[buf_id]['path']
stomp_local = yield self.ok_cancel_dialog, prompt
for buf_id in changed_bufs:
if stomp_local:
self.get_buf(buf_id)
self.save_on_get_bufs.add(buf_id)
else:
buf = self.bufs[buf_id]
# TODO: this is inefficient. we just read the file 20 lines ago
self.upload(utils.get_full_path(buf['path']))
for buf_id in missing_bufs:
self.get_buf(buf_id)
success_msg = 'Successfully joined workspace %s/%s' % (self.owner, self.workspace)
msg.log(success_msg)
editor.status_message(success_msg)
temp_data = data.get('temp_data', {})
hangout = temp_data.get('hangout', {})
hangout_url = hangout.get('url')
if hangout_url:
self.prompt_join_hangout(hangout_url)
data = utils.get_persistent_data()
data['recent_workspaces'].insert(0, {"url": self.workspace_url})
utils.update_persistent_data(data)
utils.add_workspace_to_persistent_json(self.owner, self.workspace, self.workspace_url, G.PROJECT_PATH)
self.emit("room_info")
def floobits_share_dir(dir_to_share, perms):
utils.reload_settings()
workspace_name = os.path.basename(dir_to_share)
G.PROJECT_PATH = os.path.realpath(dir_to_share)
msg.debug('%s %s %s' % (G.USERNAME, workspace_name, G.PROJECT_PATH))
file_to_share = None
dir_to_share = os.path.expanduser(dir_to_share)
dir_to_share = utils.unfuck_path(dir_to_share)
dir_to_share = os.path.abspath(dir_to_share)
dir_to_share = os.path.realpath(dir_to_share)
workspace_name = os.path.basename(dir_to_share)
if os.path.isfile(dir_to_share):
file_to_share = dir_to_share
dir_to_share = os.path.dirname(dir_to_share)
try:
utils.mkdir(dir_to_share)
except Exception:
return msg.error(
"The directory %s doesn't exist and I can't create it." %
dir_to_share)
if not os.path.isdir(dir_to_share):
return msg.error('The directory %s doesn\'t appear to exist' %
dir_to_share)
floo_file = os.path.join(dir_to_share, '.floo')
# look for the .floo file for hints about previous behavior
info = {}
try:
floo_info = open(floo_file, 'rb').read().decode('utf-8')
info = json.loads(floo_info)
except (IOError, OSError):
pass
except Exception:
msg.warn('couldn\'t read the floo_info file: %s' % floo_file)
workspace_url = info.get('url')
if workspace_url:
parsed_url = api.prejoin_workspace(workspace_url, dir_to_share,
{'perms': perms})
if parsed_url:
return floobits_join_workspace(workspace_url,
dir_to_share,
upload_path=file_to_share
or dir_to_share)
filter_func = lambda workspace_url: api.prejoin_workspace(
workspace_url, dir_to_share, {'perms': perms})
parsed_url = utils.get_workspace_by_path(dir_to_share, filter_func)
if parsed_url:
return floobits_join_workspace(workspace_url,
dir_to_share,
upload_path=file_to_share
or dir_to_share)
try:
r = api.get_orgs_can_admin()
except IOError as e:
return editor.error_message('Error getting org list: %s' % str(e))
if r.code >= 400 or len(r.body) == 0:
workspace_name = vim_input('Workspace name:', workspace_name, "file")
return create_workspace(workspace_name,
dir_to_share,
G.USERNAME,
perms,
upload_path=file_to_share or dir_to_share)
orgs = r.body
if len(orgs) == 0:
return create_workspace(workspace_name,
dir_to_share,
G.USERNAME,
perms,
upload_path=file_to_share or dir_to_share)
choices = []
choices.append(G.USERNAME)
for o in orgs:
choices.append(o['name'])
owner = vim_choice('Create workspace for:', G.USERNAME, choices)
if owner:
return create_workspace(workspace_name,
dir_to_share,
owner,
perms,
upload_path=file_to_share or dir_to_share)
