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) room_name = os.path.basename(dir_to_share) floo_room_dir = os.path.join(G.COLAB_DIR, G.USERNAME, room_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) room_url = info.get('url') if room_url: try: result = utils.parse_url(room_url) except Exception as e: msg.error(str(e)) else: room_name = result['room'] floo_room_dir = os.path.join(G.COLAB_DIR, result['owner'], result['room']) # they have previously joined the room if os.path.realpath(floo_room_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_room(room_name) msg.debug('Created room %s' % room_url) except Exception as e: msg.debug('Tried to create room' + str(e)) # they wanted to share teh dir, so always share it return join_room(room_url, lambda x: agent.protocol.create_buf(dir_to_share)) # link to what they want to share try: utils.mkdir(os.path.dirname(floo_room_dir)) os.symlink(dir_to_share, floo_room_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_room_dir, str(e))) # make & join room create_room(room_name, floo_room_dir, dir_to_share)
def join_room(room_url, on_auth=None): global agent msg.debug("room url is %s" % room_url) try: result = utils.parse_url(room_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['room'])) 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 room %s" % room_url) stop_everything() try: start_event_loop() agent = AgentConnection(on_auth=on_auth, Protocol=Protocol, **result) # owner and room 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 join_room(room_url, on_auth=None): global agent msg.debug("room url is %s" % room_url) try: result = utils.parse_url(room_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['room'])) utils.mkdir(os.path.dirname(G.PROJECT_PATH)) d = '' # TODO: really bad prompt here prompt = "Give me a directory to destructively dump data into (or just press enter): " if not os.path.isdir(G.PROJECT_PATH): while True: d = vim_input(prompt, d) if d == '': utils.mkdir(G.PROJECT_PATH) break d = os.path.realpath(os.path.expanduser(d)) if not os.path.isdir(d): prompt = '%s is not a directory. Enter an existing path (or press enter): ' % d 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))) vim.command('cd %s' % G.PROJECT_PATH) msg.debug("joining room %s" % room_url) stop_everything() try: start_event_loop() agent = AgentConnection(on_auth=on_auth, Protocol=Protocol, **result) # owner and room 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()