X-Git-Url: http://git.roojs.org/?a=blobdiff_plain;f=BrowserWindow.vala;h=55f24b6cfca9fb9f9598561c3b0ec92a1c24cb53;hb=6d2f96c1d01f452707f50a7928525745e5c3407b;hp=b88ee7cc87d69faeeebc689e76d2bcd449ff61b4;hpb=8bf410639a783ad29627e17993668d36673aa600;p=app.webkitpdf diff --git a/BrowserWindow.vala b/BrowserWindow.vala index b88ee7c..55f24b6 100644 --- a/BrowserWindow.vala +++ b/BrowserWindow.vala @@ -36,89 +36,40 @@ public class BrowserWindow : Object }); this.el.show.connect( () => { - print("resizing window\n"); - var w = webkitpdf.opt_width > 0 ? webkitpdf.opt_width : 1200; - var h = webkitpdf.opt_height > 0 ? webkitpdf.opt_height : 500; + print("resizing window\n"); + var w = webkitpdf.opt_width > 0 ? webkitpdf.opt_width : 1200; + var h = webkitpdf.opt_height > 0 ? webkitpdf.opt_height : 500; - this.el.resize(w,h); + this.el.resize(w,h); - // we have to do silly stuff with blogspot!!?!? - var blog = webkitpdf.opt_url; + // we have to do silly stuff with blogspot!!?!? + var blog = webkitpdf.opt_url; /* Do not redirect if the domain is .com already */ if (/\.blogspot\.com/.match(blog)) { - var r = new Regex ("\\.blogspot\\..*?/"); - blog = r.replace(blog, blog.length, 0, ".blogspot.com/ncr/"); + var r = new Regex ("\\.blogspot\\..*?/"); + blog = r.replace(blog, blog.length, 0, ".blogspot.com/ncr/"); } - - var iostream = new WebKit.FileIOStream() ; - - var file = File.new_tmp ("cookie-XXXXXX.txt", out iostream); - - print("cookie tmp file name: %s\n", file.get_path ()); - - var ostream = iostream.output_stream; - - if(webkitpdf.opt_cookies != null) { - print("setting cookie %s\n",webkitpdf.opt_cookies); - - var uri = new Soup.URI (blog); - - string[] cookies = webkitpdf.opt_cookies.split ("=", 2); - - if(cookies.length >=2 ) { -#if GTK3 - var context = WebKit.WebContext.get_default(); - var cookie_manager = context.get_cookie_manager(); - cookie_manager.set_accept_policy(WebKit.CookieAcceptPolicy.NEVER); - string scheme = uri.get_scheme (); - var ssl = "FALSE"; - if (scheme=="https") { - ssl="TRUE"; - } - try { - - //File file = File.new_tmp ("cookie-XXXXXX.txt", out iostream); - //print("cookie tmp file name: %s\n", file.get_path ()); - - DataOutputStream dostream = new DataOutputStream (ostream); - - dostream.put_string (uri.get_host() + "\tTRUE\t/\t"+ssl+"\t2147483647\t" + cookies[0] + "\t" + cookies[1] ); - cookie_manager.set_persistent_storage(file.get_path(), WebKit.CookiePersistentStorage.TEXT); - } catch (Error e) { - stdout.printf ("Error: %s\n", e.message); - } -#else - var sess = WebKit.get_default_session(); - var cookie_jar = new Soup.CookieJar(); - cookie_jar.set_accept_policy(Soup.CookieJarAcceptPolicy.NEVER); - cookie_jar.add_cookie(new Soup.Cookie (cookies[0], cookies[1], uri.get_host(), "/", -1)); - sess.add_feature(cookie_jar); -#endif - } - - - } - - print("loading url %s\n",blog); - this.view.el.load_uri(blog); - // if we have not finished in 25 seconds - // call printit... - var cd = 25; - GLib.Timeout.add_seconds(1, () => { - cd--; - print("timeout 25 seconds :%d", cd); - if (cd < 1) { - print("calling printit - timed out"); - _this.view.printit(); - return false; - } - return true; + print("loading url %s\n",blog); + this.view.el.load_uri(blog); + + // if we have not finished in 25 seconds + // call printit... + var cd = webkitpdf.opt_delay > 25 ? webkitpdf.opt_delay : 25; + GLib.Timeout.add_seconds(1, () => { + cd--; + print("timeout %d seconds :%d", (int)( webkitpdf.opt_delay > 25 ? webkitpdf.opt_delay : 25), (int)cd); + if (cd < 1) { + print("calling printit - timed out"); + _this.view.printit(); + return false; + } + return true; - }); + }); }); @@ -150,7 +101,7 @@ public class BrowserWindow : Object // init method { - this.el.set_policy(Gtk.PolicyType.AUTOMATIC, Gtk.PolicyType.AUTOMATIC); + this.el.set_policy(Gtk.PolicyType.AUTOMATIC, Gtk.PolicyType.AUTOMATIC); } } @@ -163,6 +114,7 @@ public class BrowserWindow : Object // my vars (def) + public string cookie_file; public bool print_started; public int load_try; @@ -182,19 +134,37 @@ public class BrowserWindow : Object // init method { - this.load_try = 0; + this.load_try = 0; #if GTK3 - this.el.load_changed.connect( (ev ) => { - if (ev != WebKit.LoadEvent.FINISHED) { - return; - } + this.el.load_changed.connect( (ev ) => { + if (ev != WebKit.LoadEvent.FINISHED) { + return; + } this.printit(); }); + + this.el.resource_load_started.connect( (resource, request) => { + // print("Adding referrer header %s\n" , request.get_uri()); + + if (request.get_http_headers() != null) { + request.get_http_headers().remove("Referer"); + request.get_http_headers().append("Referer", request.get_uri()); + } + }); + var settings = this.el.get_settings(); + settings.set_user_agent( "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/63.0.3239.84 Safari/537.36" ); + settings.hardware_acceleration_policy = WebKit.HardwareAccelerationPolicy.NEVER; + settings.enable_accelerated_2d_canvas = false; + if (webkitpdf.opt_disable_javascript) { + settings.enable_javascript = false; + } + WebKit.WebContext.get_default().set_tls_errors_policy( WebKit.TLSErrorsPolicy.IGNORE ); + #else //listeners // this.el.resource_request_starting.connect( (p0, webres, netreq ) => { @@ -203,13 +173,95 @@ public class BrowserWindow : Object // }); this.el.load_finished.connect( ( ) => { - - + this.printit(); }); + WebKit.get_default_session().request_started.connect((message, socket) => { + Soup.URI uri = message.get_uri(); + + var host = uri.get_host (); + if (host.contains("google-analytics") || host.contains("googlesyndication") || host.contains("facebook") ) { + print("DISCONNECTING: %s\n", host); + socket.disconnect(); + return; + } + + print("Adding header: %s\n", uri.to_string(false)); + message.request_headers.remove("Referer"); + message.request_headers.append("Referer", "http://"+host+"/"); + + + }); + var sess = WebKit.get_default_session(); + sess.user_agent= "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/63.0.3239.84 Safari/537.36"; + sess.ssl_strict = false; + + var settings = this.el.get_settings(); + + settings.enable_webgl = false; + settings.enable_plugins = false; + settings.enable_java_applet = false; + if (webkitpdf.opt_disable_javascript) { + settings.enable_scripts = false; + } + + + this.el.console_message.connect((msg, line, sid) => { + print("CONSOLE: %s:%d %s\n", sid, line, msg); + return false; + }); + this.el.create_web_view.connect((frame) => { + print("WEBFRAME:%s\n", frame.uri); + return new WebKit.WebView(); + }); + #endif } + this.cookie_file = ""; + if (webkitpdf.opt_cookies != null) { + print("setting cookie %s\n",webkitpdf.opt_cookies); + + var uri = new Soup.URI (webkitpdf.opt_url); + + string[] cookies = webkitpdf.opt_cookies.split ("=", 2); + + if(cookies.length > 1 ) { + + #if GTK3 + var context = WebKit.WebContext.get_default(); + var cookie_manager = context.get_cookie_manager(); + cookie_manager.set_accept_policy(WebKit.CookieAcceptPolicy.NEVER); + string scheme = uri.get_scheme (); + + try { + FileIOStream iostream; + File file = File.new_tmp ("cookie-XXXXXX.txt", out iostream); + this.cookie_file = file.get_path (); + print("cookie tmp file name: %s\n", file.get_path ()); + + OutputStream ostream = iostream.output_stream; + DataOutputStream dostream = new DataOutputStream (ostream); + dostream.put_string ("%s\tTRUE\t/\t%s\t2147483647\t%s\t%s".printf(uri.get_host(),scheme == "http" ? "FALSE" : "TRUE",cookies[0],cookies[1])); + cookie_manager.set_persistent_storage(file.get_path(), WebKit.CookiePersistentStorage.TEXT); + } catch (Error e) { + stdout.printf ("Error: %s\n", e.message); + } + #else + var cookie_jar = new Soup.CookieJar(); + var cookie = new Soup.Cookie (cookies[0], cookies[1], uri.get_host(), "/", -1); + if(uri.get_scheme() =="https") { + cookie.set_secure(true); + } + cookie_jar.set_accept_policy(Soup.CookieJarAcceptPolicy.NEVER); + cookie_jar.add_cookie(cookie); + WebKit.get_default_session().add_feature(cookie_jar); + #endif + + } + + } + // init method } // user defined functions @@ -229,73 +281,77 @@ public class BrowserWindow : Object this.print_started = true; // flag to stop double call. - // inject any javascript if needed.. - if (webkitpdf.opt_inject_js != null) { + // inject any javascript if needed.. + if (webkitpdf.opt_inject_js != null) { - print("injecting javascript files \n"); - foreach(var inject_js in webkitpdf.opt_inject_js) { - print("inject? %s\n", inject_js); - string str; - try { - FileUtils.get_contents( inject_js, out str); - } catch(FileError e) { - print("Failed to inject %s\n", inject_js); - continue; - } + print("injecting javascript files \n"); + foreach(var inject_js in webkitpdf.opt_inject_js) { + print("inject? %s\n", inject_js); + string str; + try { + FileUtils.get_contents( inject_js, out str); + } catch(FileError e) { + print("Failed to inject %s\n", inject_js); + continue; + } #if GTK3 - GLib.MainLoop mainloop = new GLib.MainLoop(); - this.el.run_javascript.begin(str, null, (obj,res) => { - mainloop.quit(); - }); - mainloop.run(); + GLib.MainLoop mainloop = new GLib.MainLoop(); + this.el.run_javascript.begin(str, null, (obj,res) => { + mainloop.quit(); + }); + mainloop.run(); #else - this.el.execute_script(str); + this.el.execute_script(str); #endif - } - } + } + } - print("calling print in %d seconds\n", (int)webkitpdf.opt_delay); - if (webkitpdf.opt_target_pdf != null) { - GLib.Timeout.add_seconds(webkitpdf.opt_delay, () => { - print("delay done URL: %s\n", this.el.uri); + print("calling print in %d seconds\n", (int)webkitpdf.opt_delay); + if (webkitpdf.opt_target_pdf != null) { + GLib.Timeout.add_seconds(webkitpdf.opt_delay, () => { + print("delay done URL: %s\n", this.el.uri); - this.printpdf(); - return false; - }, GLib.Priority.DEFAULT); - return; - } + this.printpdf(); + return false; + }, GLib.Priority.DEFAULT); + return; + } #if GTK3 - print("PNG not supported in webkit2\n"); - return; + GLib.Timeout.add_seconds(webkitpdf.opt_delay, () => { + print("delay done URL: %s\n", this.el.uri); + this.printpng(); + return false; + }, GLib.Priority.DEFAULT); + return; #else GLib.Timeout.add_seconds(webkitpdf.opt_delay, () => { - // ----------------- JPEG - // resize the window... - // window.document.documentElement.scrollHeight - var scroll_height = (int) _this.view.el.get_dom_document().document_element.scroll_height; - print("Scroll height %d\n", scroll_height); - if (scroll_height> 1024 || webkitpdf.opt_width > 0 || webkitpdf.opt_height > 0 ) { - var w = webkitpdf.opt_width > 0 ? webkitpdf.opt_width : 1200; - var h = int.min(scroll_height, 6000); - _this.scrolled_window.el.set_size_request( w, h); - _this.el.resize (w+50, h+50); - print("Resize to %d, %d\n", w,h); - GLib.Timeout.add_seconds(webkitpdf.opt_delay > 0 ? webkitpdf.opt_delay : 1 , () => { - this.printpng(); - return false; - }, GLib.Priority.DEFAULT); - return false; - } - this.printpng(); - return false; - }, GLib.Priority.DEFAULT); + // ----------------- JPEG + // resize the window... + // window.document.documentElement.scrollHeight + var scroll_height = (int) _this.view.el.get_dom_document().document_element.scroll_height; + print("Scroll height %d\n", scroll_height); + if (scroll_height> 1024 || webkitpdf.opt_width > 0 || webkitpdf.opt_height > 0 ) { + var w = webkitpdf.opt_width > 0 ? webkitpdf.opt_width : 1200; + var h = int.min(scroll_height, 6000); + _this.scrolled_window.el.set_size_request( w, h); + _this.el.resize (w+50, h+50); + print("Resize to %d, %d\n", w,h); + GLib.Timeout.add_seconds(webkitpdf.opt_delay > 0 ? webkitpdf.opt_delay : 1 , () => { + this.printpng(); + return false; + }, GLib.Priority.DEFAULT); + return false; + } + this.printpng(); + return false; + }, GLib.Priority.DEFAULT); #endif @@ -305,78 +361,80 @@ public class BrowserWindow : Object - var list = Gtk.PaperSize.get_paper_sizes(false); - var psetup = new Gtk.PageSetup(); - for(var i = 0; i < list.length(); i++ ) { - var entry = list.nth_data(i).copy(); - if (entry.get_name() == "iso_a2") { - psetup.set_paper_size(entry); - } - } + var list = Gtk.PaperSize.get_paper_sizes(false); + var psetup = new Gtk.PageSetup(); + for(var i = 0; i < list.length(); i++ ) { + var entry = list.nth_data(i).copy(); + if (entry.get_name() == "iso_a2") { + psetup.set_paper_size(entry); + } + } #if GTK3 - var pe = new WebKit.PrintOperation(this.el); - pe.set_page_setup(psetup); - pe.finished.connect( () => { - print("print completed\n"); - Gtk.main_quit(); + var pe = new WebKit.PrintOperation(this.el); + pe.set_page_setup(psetup); + pe.finished.connect( () => { + print("print completed\n"); + this.delete_cookiefile(); + Gtk.main_quit(); - }); + }); - var ps = new Gtk.PrintSettings(); - ps.set_printer("Print to File"); - ps.set("output-file-format", "pdf"); - ps.set("output-uri", "file://" + webkitpdf.opt_target_pdf); + var ps = new Gtk.PrintSettings(); + ps.set_printer("Print to File"); + ps.set("output-file-format", "pdf"); + ps.set("output-uri", "file://" + webkitpdf.opt_target_pdf); - pe.set_print_settings(ps); + pe.set_print_settings(ps); - pe.print(); + pe.print(); #else - var pe = new Gtk.PrintOperation(); - pe.export_filename = webkitpdf.opt_target_pdf; - print("got callback for print\n"); - pe.ref(); - /*pe.failed.connect(() => { - print("print failed\n"); - Gtk.main_quit(); + var pe = new Gtk.PrintOperation(); + pe.export_filename = webkitpdf.opt_target_pdf; + print("got callback for print\n"); + pe.ref(); + /*pe.failed.connect(() => { + print("print failed\n"); + Gtk.main_quit(); - }) - */; - pe.done.connect(() => { - print("print completed\n"); - Gtk.main_quit(); + }) + */; + pe.done.connect(() => { + print("print completed\n"); + this.delete_cookiefile(); + Gtk.main_quit(); - }); + }); - //pe.run_dialog(_this.el); + //pe.run_dialog(_this.el); - pe.set_default_page_setup(psetup); - print("Calling Print?\n"); - //pe.run_dialog(_this.el); + pe.set_default_page_setup(psetup); + print("Calling Print?\n"); + //pe.run_dialog(_this.el); - try { - this.el.get_main_frame().print_full(pe,Gtk.PrintOperationAction.EXPORT); - } catch (Error e) { - // print failed... - } + try { + this.el.get_main_frame().print_full(pe,Gtk.PrintOperationAction.EXPORT); + } catch (Error e) { + // print failed... + } #endif - return false; + return false; } public void printpng () { //var filename = "/tmp/test.pdf"; @@ -390,6 +448,16 @@ public class BrowserWindow : Object print("making screenshot\n"); #if GTK3 + GLib.MainLoop mainloop = new GLib.MainLoop(); + _this.view.el.get_snapshot.begin(WebKit.SnapshotRegion.FULL_DOCUMENT, WebKit.SnapshotOptions.NONE, null, (obj, res) => { + var sf = _this.view.el.get_snapshot.end(res); + sf.write_to_png(webkitpdf.opt_target_png); + print("Saved to %s\n", webkitpdf.opt_target_png); + mainloop.quit(); + + }); + + mainloop.run(); #else // fix vapi - get_snapshot add '?' to all null. @@ -400,9 +468,9 @@ public class BrowserWindow : Object int w,h; // add out to gdkpixmap.get_size - pixmap.get_size( out w , out h); - var pixbuf = Gdk.pixbuf_get_from_drawable(null, pixmap, null , 0, 0, 0, 0, w, h); - print("pixbuf size: %d x %d\n ", w,h); + pixmap.get_size( out w , out h); + var pixbuf = Gdk.pixbuf_get_from_drawable(null, pixmap, null , 0, 0, 0, 0, w, h); + print("pixbuf size: %d x %d\n ", w,h); @@ -413,12 +481,30 @@ public class BrowserWindow : Object // sf.write_to_png(Browser.opt_target_png); //}); + #endif + this.delete_cookiefile(); + Gtk.main_quit(); - #endif + + } + public void delete_cookiefile () { + if(this.cookie_file.length < 1 ){ + return; + } + print("deleting tmp file %s\n",this.cookie_file); + File file = File.new_for_path (this.cookie_file); + try { + if (file.query_exists () == true) { + file.delete (); + } + } catch (Error e) { + stdout.printf ("Error: %s\n", e.message); + } + } }