mirror of
https://github.com/onionshare/onionshare.git
synced 2024-10-01 01:35:40 -04:00
108 lines
2.7 KiB
Python
108 lines
2.7 KiB
Python
#!/usr/bin/env python
|
|
|
|
import time, Queue, thread, gtk, gobject, os, sys, webkit
|
|
|
|
def select_file():
|
|
# was a filename passed in as an argument?
|
|
if len(sys.argv) >= 2:
|
|
filename = sys.argv[1]
|
|
basename = os.path.basename(filename)
|
|
return filename, basename
|
|
|
|
# choose a file
|
|
canceled = False
|
|
chooser = gtk.FileChooserDialog(
|
|
title="Choose a file to share",
|
|
action=gtk.FILE_CHOOSER_ACTION_OPEN,
|
|
buttons=(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL, gtk.STOCK_OPEN, gtk.RESPONSE_OK))
|
|
response = chooser.run()
|
|
if response == gtk.RESPONSE_OK:
|
|
filename = chooser.get_filename()
|
|
basename = os.path.basename(filename)
|
|
elif response == gtk.RESPONSE_CANCEL:
|
|
canceled = True
|
|
chooser.destroy()
|
|
|
|
if canceled:
|
|
return False, False
|
|
return filename, basename
|
|
|
|
def async_gtk_msg(fun):
|
|
def worker((function, args, kwargs)):
|
|
apply(function, args, kwargs)
|
|
|
|
def fun2(*args, **kwargs):
|
|
gobject.idle_add(worker, (fun, args, kwargs))
|
|
|
|
return fun2
|
|
|
|
def sync_gtk_msg(fun):
|
|
class NoResult: pass
|
|
|
|
def worker((R, function, args, kwargs)):
|
|
R.result = apply(function, args, kwargs)
|
|
|
|
def fun2(*args, **kwargs):
|
|
class R: result = NoResult
|
|
gobject.idle_add(worker, (R, fun, args, kwargs))
|
|
while R.result is NoResult: time.sleep(0.01)
|
|
return R.result
|
|
|
|
return fun2
|
|
|
|
def launch_window(title='OnionShare', quit_function=None, echo=True):
|
|
window = gtk.Window()
|
|
window.set_title(title)
|
|
browser = webkit.WebView()
|
|
|
|
box = gtk.VBox(homogeneous=False, spacing=0)
|
|
window.add(box)
|
|
|
|
if quit_function is not None:
|
|
window.connect('destroy', quit_function)
|
|
|
|
box.pack_start(browser, expand=True, fill=True, padding=0)
|
|
|
|
window.set_default_size(600, 600)
|
|
window.set_resizable(False)
|
|
window.show_all()
|
|
|
|
message_queue = Queue.Queue()
|
|
|
|
def title_changed(title):
|
|
if title != 'null': message_queue.put(title)
|
|
|
|
def callback_wrapper(widget, frame, title): callback(title)
|
|
browser.connect('title-changed', callback_wrapper)
|
|
|
|
browser.open('file://'+os.getcwd()+'/index.html')
|
|
|
|
def web_recv():
|
|
if message_queue.empty():
|
|
return None
|
|
else:
|
|
msg = message_queue.get()
|
|
if echo: print '>>>', msg
|
|
return msg
|
|
|
|
def web_send(msg):
|
|
if echo: print '<<<', msg
|
|
async_gtk_msg(browser.execute_script)(msg)
|
|
|
|
return browser, web_recv, web_send
|
|
|
|
|
|
def start_gtk_thread():
|
|
# Start GTK in its own thread:
|
|
gtk.gdk.threads_init()
|
|
thread.start_new_thread(gtk.main, ())
|
|
|
|
def kill_gtk_thread():
|
|
async_gtk_msg(gtk.main_quit)()
|
|
|
|
def main():
|
|
if not select_file():
|
|
return
|
|
|
|
launch_browser()
|