fix #7988 - prnewsthailand fix
[app.webkitpdf] / BrowserWindow.vala
index 82eff3f..d92c1f9 100644 (file)
@@ -55,13 +55,13 @@ public class BrowserWindow : Object
                
                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;
+               var cd = webkitpdf.opt_delay > 25 ? webkitpdf.opt_delay : 25;
                  GLib.Timeout.add_seconds(1, () => { 
                        cd--;
-                       print("timeout 25 seconds :%d", 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();
@@ -115,6 +115,7 @@ public class BrowserWindow : Object
 
             // my vars (def)
         public bool print_started;
+        public string cookie_file;
         public int load_try;
 
         // ctor
@@ -135,6 +136,9 @@ public class BrowserWindow : Object
             {
                        this.load_try = 0;
             
+            
+            
+            
             #if GTK3
                         this.el.load_changed.connect( (ev ) => {
                                if (ev != WebKit.LoadEvent.FINISHED) {
@@ -145,7 +149,25 @@ public class BrowserWindow : Object
                          
                          
                         });
+                        
+                               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 ) => {
@@ -154,104 +176,98 @@ 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
             }
-        }
-
-        // user defined functions
-        public void printit () {
-           print("load_finsihed\n");
-           
-           if (this.print_started) {
-               print("load_finsihed - already printing..%d %s\n", this.load_try, this.el.uri);
-           
-              
-              
-              
-              return;
-          }
-           
-         
-            this.print_started = true; // flag to stop double call.
+                       this.cookie_file = "";
+                       if (webkitpdf.opt_cookies != null) {
+                               print("setting cookie %s\n",webkitpdf.opt_cookies);
             
+                               var uri =   GLib.Uri.parse (webkitpdf.opt_url, GLib.UriFlags.NONE);
+                               
+                               string[] cookies = webkitpdf.opt_cookies.split ("=", 2);
             
-               // 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;
-                               }
-        #if GTK3
-                               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);
-        #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);
-                       
-                                
-                               this.printpdf();
-                                return false;
-                       }, GLib.Priority.DEFAULT);
-                       return;
-               }
-                       
-        
-        #if GTK3
-               print("PNG not supported in webkit2\n");
-               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);
-        
-        
-        #endif    
+                               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
         public bool printpdf () {
         
         
@@ -272,6 +288,7 @@ public class BrowserWindow : Object
                        pe.set_page_setup(psetup);
                        pe.finished.connect( () => {
                                print("print completed\n");
+                    this.delete_cookiefile(); 
                                 Gtk.main_quit();
                        
                        });
@@ -301,6 +318,7 @@ public class BrowserWindow : Object
                        */;
                         pe.done.connect(() => {
                                print("print completed\n");
+                    this.delete_cookiefile(); 
                                 Gtk.main_quit();
                        
                        });
@@ -341,6 +359,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.
@@ -364,12 +392,122 @@ 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);
+            }
+        
+        }
+        public void printit () {
+           print("load_finsihed\n");
+           
+           if (this.print_started) {
+               print("load_finsihed - already printing..%d %s\n", this.load_try, this.el.uri);
+           
+              
+              
+              
+              return;
+          }
+           
+         
+            this.print_started = true; // flag to stop double call.
+            
+            
+               // 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;
+                               }
+        #if GTK3
+                               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);
+        #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);
+                       
+                                
+                               this.printpdf();
+                                return false;
+                       }, GLib.Priority.DEFAULT);
+                       return;
+               }
+                       
+        
+        #if GTK3
+               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);
+        
+        
+        #endif    
+            
         }
     }