Пример #1
0
class WallabyClient:
	def __init__(self, host_port_pair, routes, debug=False):
		self.sock = ESock(socket.create_connection(host_port_pair), debug=debug)
		self.connected = True
		self.debug = debug
		self.sync = SyncClient(self.sock, PATH, "w_sync", debug=True)
		routes.update({"w_sync" : self.sync})
		self.routes = routes


	def start(self):
		self.sync.start()
		while 1 and self.connected:
			data = self.sock.recv()
			try:
				if data[1] in self.routes:
					self.routes[data[1]].run(data[0], self)
			except Exception as e:
				if not is_socket_related_error(e):
					capture_trace()
				break


	def stop(self):
		self.sock.close()
Пример #2
0
	def __init__(self, host_port_pair, routes, debug=False):
		self.sock = ESock(socket.create_connection(host_port_pair), debug=debug)
		self.connected = True
		self.debug = debug
		self.sync = SyncClient(self.sock, PATH, "w_sync", debug=True)
		routes.update({"w_sync" : self.sync})
		self.routes = routes
Пример #3
0
	def connect(self, connect_details):
		connect_details_raw = connect_details
		connect_details = connect_details.split(":")
		compression_level = int(sublime.load_settings("fl0w.sublime-settings").get("compression_level"))

		if len(connect_details) == 2:
			try:
				# Establish connection to the server
				self.sock = ESock(socket.create_connection((connect_details[0], int(connect_details[1]))),
				                  disconnect_callback=self.invoke_disconnect, debug=False, compression_level=compression_level)
				sublime.status_message("Connected to %s:%s." % (connect_details[0], connect_details[1]))
				# Initialize all routes
				error_report = Fl0w.ErrorReport()
				info = Fl0w.Info()
				wallaby_control = Fl0w.WallabyControl()
				get_info = Fl0w.GetInfo()
				std_stream = Fl0w.StdStream()
				self.s_sync = SyncClient(self.sock, self.folder, "s_sync")
				compile = Fl0w.Compile()
				_thread.start_new_thread(sock_handler, (self.sock, {"error_report": error_report,
					"info" : info, "wallaby_control" : wallaby_control, "get_info" : get_info,
					"s_sync" : self.s_sync, "compile" : compile, "std_stream" : std_stream}, self))
				self.s_sync.start()
				self.connected = True
				# Saving last server address
				sublime.load_settings("fl0w.sublime-settings").set("server_address", connect_details_raw)
				sublime.save_settings("fl0w.sublime-settings")
			except OSError as e:
				sublime.error_message("Error during connection creation:\n %s" % str(e))
		else:
			sublime.error_message("Invalid input.")
Пример #4
0
class Fl0w:
	def __init__(self, window):
		self.connected = False
		self.window = window
		self.folder = window.folders()[0]
		self.selected_wallaby = None
		self.sock = None
		self.s_sync = None

		self.start_menu = Menu()
		self.start_menu.add(Entry("Connect", "Connect to a fl0w server", action=self.invoke_connect))
		self.start_menu.add(Entry("About", "Information about fl0w", action=self.invoke_about))

		self.main_menu = Menu()
		self.main_menu.add(Entry("Wallaby Control", "Control a connected Wallaby", action=self.invoke_wallaby_control))
		self.main_menu.add(Entry("Info", "Server info", action=self.invoke_info))
		self.main_menu.add(Entry("Debug", "Debug options", action=self.invoke_debug_options))
		self.main_menu.add(Entry("Disconnect", "Disconnect from server", action=self.invoke_disconnect))


	def invoke_about(self):
		if sublime.ok_cancel_dialog("fl0w by @robot0nfire", "robot0nfire.com"):
			webbrowser.open("http://robot0nfire.com")


	def invoke_info(self):
		self.sock.send("", "info")


	class Info(Routing.ClientRoute):
		def run(self, data, handler):
			sublime.message_dialog(data)
			handler.main_menu.invoke(handler.window)


	class GetInfo(Routing.ClientRoute):
		def run(self, data, handler):
			if data == "":
				handler.sock.send({"type" : CHANNEL}, "get_info")


	class ErrorReport(Routing.ClientRoute):
		def run(self, data, handler):
			sublime.error_message(data)


	class Compile(Routing.ClientRoute):
		def run(self, data, handler):
			status = ""
			if data["failed"]:
				status = "Compile failed (%s) (%s)." % (data["relpath"], strftime("%H:%M:%S"))
			else:
				status = "Compile completed successfully. (%s) (%s)." % (data["relpath"], strftime("%H:%M:%S"))
			view = handler.window.create_output_panel('compile_panel')
			view.set_syntax_file("Packages/fl0w/CompileHighlight.sublime-syntax")
			view.settings().set("draw_white_space", "none")
			view.settings().set("draw_indent_guides", False)
			view.settings().set("gutter", False)
			view.settings().set("line_numbers", False)
			view.set_read_only(False)
			view.run_command("append", {"characters": data["returned"] + status, "scroll_to_end" : True})
			view.set_read_only(True)
			handler.window.run_command("show_panel", {"panel": "output.compile_panel"})

	class StdStream(Routing.ClientRoute):
		def __init__(self):
			self.view = None
			self.buffer = ""

		def setup_view(self, handler):
			self.view = handler.window.create_output_panel('std_stream_panel')
			self.view.settings().set("draw_white_space", "none")
			self.view.settings().set("draw_indent_guides", False)
			self.view.settings().set("gutter", False)
			self.view.settings().set("line_numbers", False)

		def run(self, data, handler):
			self.setup_view(handler)
			handler.window.run_command("show_panel", {"panel": "output.std_stream_panel"})
			if type(data) is str:
				self.buffer += data
				self.append_to_view(self.buffer)
			elif type(data) is dict:
				if "return_code" in data:
					self.append_to_view(self.buffer + "\nProgram finished with return code %d" % data["return_code"])
					self.buffer = ""

		def append_to_view(self, text):
			self.view.set_read_only(False)
			self.view.run_command("append", {"characters" : text, "scroll_to_end" : True})
			self.view.set_read_only(True)



	def connect(self, connect_details):
		connect_details_raw = connect_details
		connect_details = connect_details.split(":")
		compression_level = int(sublime.load_settings("fl0w.sublime-settings").get("compression_level"))

		if len(connect_details) == 2:
			try:
				# Establish connection to the server
				self.sock = ESock(socket.create_connection((connect_details[0], int(connect_details[1]))),
				                  disconnect_callback=self.invoke_disconnect, debug=False, compression_level=compression_level)
				sublime.status_message("Connected to %s:%s." % (connect_details[0], connect_details[1]))
				# Initialize all routes
				error_report = Fl0w.ErrorReport()
				info = Fl0w.Info()
				wallaby_control = Fl0w.WallabyControl()
				get_info = Fl0w.GetInfo()
				std_stream = Fl0w.StdStream()
				self.s_sync = SyncClient(self.sock, self.folder, "s_sync")
				compile = Fl0w.Compile()
				_thread.start_new_thread(sock_handler, (self.sock, {"error_report": error_report,
					"info" : info, "wallaby_control" : wallaby_control, "get_info" : get_info,
					"s_sync" : self.s_sync, "compile" : compile, "std_stream" : std_stream}, self))
				self.s_sync.start()
				self.connected = True
				# Saving last server address
				sublime.load_settings("fl0w.sublime-settings").set("server_address", connect_details_raw)
				sublime.save_settings("fl0w.sublime-settings")
			except OSError as e:
				sublime.error_message("Error during connection creation:\n %s" % str(e))
		else:
			sublime.error_message("Invalid input.")


	# Input invokers
	def invoke_connect(self):
		address = sublime.load_settings("fl0w.sublime-settings").get("server_address")
		address = "" if type(address) is not str else address
		Input("Address:Port", initial_text=address, on_done=self.connect).invoke(self.window)


	def invoke_disconnect(self):
		if self.connected:
			self.connected = False
			sublime.message_dialog("Connection closed ('%s')" % self.folder)
			self.s_sync.observer.stop()
			self.sock.close()


	def invoke_wallaby_control(self):
		self.sock.send("list_wallaby_controllers", "wallaby_control")


	def invoke_run_menu(self):
		self.sock.send("list_programs", "wallaby_control")


	class WallabyControl(Routing.ClientRoute):
		def run(self, data, handler):
			if "wallaby_controllers" in data:
				wallaby_menu = Menu()
				entry_count = 0
				for wallaby in data["wallaby_controllers"]:
					wallaby_menu.add(Entry(wallaby, str(data["wallaby_controllers"][wallaby]), action=handler.wallaby_control_submenu, kwargs={"wallaby" : wallaby}))
					entry_count += 1
				if entry_count != 0:
					wallaby_menu.invoke(handler.window, back=handler.main_menu)
				else:
					sublime.error_message("No Wallaby Controllers connected.")
			elif "programs" in data:
				run_menu = Menu(subtitles=False)
				entry_count = 0
				for program in data["programs"]:
					run_menu.add(Entry(program, action=handler.run_program, kwargs={"program" : program}))
					entry_count += 1
				if entry_count != 0:
					run_menu.invoke(handler.window, back=handler.main_menu)
				else:
					sublime.error_message("No programs avaliable.")

	def run_program(self, program):
		self.sock.send({self.selected_wallaby : {"run" : program}}, "wallaby_control")


	def wallaby_control_submenu(self, wallaby):
		self.selected_wallaby = wallaby
		menu = Menu(subtitles=False)

		menu.add(Entry("Run", action=self.invoke_run_menu))
		for action in ("Stop", "Shutdown", "Reboot", "Disconnect"):
			menu.add(Entry(action, action=self.sock.send, kwargs={"data" : {wallaby : [action.lower()]}, "route" : "wallaby_control"}))
		menu.invoke(self.window)


	def invoke_debug_options(self):
		debug_menu = Menu(subtitles=False)
		debug_menu.add(Entry("On", action=self.set_debug, kwargs={"debug" : True}))
		debug_menu.add(Entry("Off", action=self.set_debug, kwargs={"debug" : False}))
		debug_menu.invoke(self.window, back=self.main_menu)


	def set_debug(self, debug):
		sublime.status_message("fl0w: Debug now '%s'" % debug)
		self.sock.debug = debug
		sublime.load_settings("fl0w.sublime-settings").set("debug", debug)
		sublime.save_settings("fl0w.sublime-settings")