def create_test_instance():
root_path = tempfile.mkdtemp('autofs')
inst_path = os.path.join(root_path, 'inst')
os.mkdir(inst_path)
inst = instance.Instance.create(inst_path)
files1_path = os.path.join(root_path, 'files1')
with file_maker(files1_path) as fm:
fm.make_file('test1', b'hello world')
fm.make_file('test2', b'hello world 2')
fm.make_file('dir_a/test_file', 'abc')
fm.make_file('dir_a/dir_b/test_file', 'abcd')
fm.mkdir_p('dir_a/dir_c')
cmd_bundle.bundle(inst, files1_path)
return root_path, inst
def main():
if len(sys.argv) < 2:
print(
"""Usage:
\tautofs command instance-name
Commands:
\tserve
\tbundle"""
)
sys.exit(1)
# TODO: unicode?
command = sys.argv[1].decode()
instance_path = os.path.realpath(sys.argv[2]).decode()
command_args = [s.decode() for s in sys.argv[3:]]
gevent.signal(signal.SIGQUIT, gevent.shutdown)
if command == "serve":
inst = instance.Instance.load(instance_path)
# ls = local.start_server(ctx, inst)
rs = network.start_server(inst)
# glets = [ls, rs]
# gevent.joinall(glets)
rs.join()
elif command == "connect":
# Client?
inst = instance.Instance.load(instance_path)
ls = local.start_server(inst)
pf = network.find_peers(inst)
glets = [ls, pf]
gevent.joinall(glets)
# "Offline" commands
elif command == "init":
if not os.path.isdir(instance_path):
os.makedirs(instance_path)
instance.Instance.create(instance_path)
else:
print("Instance already exists at {}".format(instance_path))
elif command == "join":
if len(command_args) == 0:
print("Usage: autofs join instance-name remote-host")
if os.path.isdir(instance_path):
print("Cannot join using an existing instance")
inst = cmd_join.join(instance_path, command_args[0])
elif sys.argv[1] == "bundle":
if len(command_args) == 0:
print("Usage: autofs bundle instance-name target-dir")
sys.exit(1)
if not os.path.isdir(instance_path):
os.makedirs(instance_path)
inst = instance.Instance.create(instance_path)
else:
inst = instance.Instance.load(instance_path)
cmd_bundle.bundle(inst, command_args[0])
inst.save()
print("Done")
elif sys.argv[1] == "list":
inst = instance.Instance.load(instance_path)
cmd_list.list_items(inst)
else:
print("Unrecognized command {}".format(command))
sys.exit(1)
