BrowserWindow.vala
[app.webkitpdf] / BrowserWindow.vala
index e698229..73ac9ce 100644 (file)
@@ -2,9 +2,9 @@ static BrowserWindow  _BrowserWindow;
 
 public class BrowserWindow : Object
 {
-    public Gtk.OffscreenWindow el;
+    public Gtk.Window el;
     private BrowserWindow  _this;
-
+    public string cookie_file;
     public static BrowserWindow singleton()
     {
         if (_BrowserWindow == null) {
@@ -12,6 +12,7 @@ public class BrowserWindow : Object
         }
         return _BrowserWindow;
     }
+    public Xcls_scrolled_window scrolled_window;
     public Xcls_view view;
 
         // my vars (def)
@@ -20,31 +21,107 @@ public class BrowserWindow : Object
     public BrowserWindow()
     {
         _this = this;
-        this.el = new Gtk.OffscreenWindow();
+        this.el = new Gtk.Window( Gtk.WindowType.TOPLEVEL );
 
         // my vars (dec)
 
         // set gobject values
-        var child_0 = new Xcls_ScrolledWindow2( _this );
+        var child_0 = new Xcls_scrolled_window( _this );
         child_0.ref();
         this.el.add (  child_0.el  );
 
         //listeners
         this.el.destroy.connect( () => {
+                  
             Gtk.main_quit();
         });
         this.el.show.connect( () => {
                
-               print("resizing window\n");
-               this.el.resize(1200,500);
-               print("loading url\n");
-               this.view.el.load_uri(webkitpdf.opt_url);
+         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);
+        
+         // 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/");
+          }
+          
+         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 {            
+                    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 (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;
+        
+         });
+               
         
         });
     }
 
     // user defined functions
-    public class Xcls_ScrolledWindow2 : Object
+    public class Xcls_scrolled_window : Object
     {
         public Gtk.ScrolledWindow el;
         private BrowserWindow  _this;
@@ -53,9 +130,10 @@ public class BrowserWindow : Object
             // my vars (def)
 
         // ctor
-        public Xcls_ScrolledWindow2(BrowserWindow _owner )
+        public Xcls_scrolled_window(BrowserWindow _owner )
         {
             _this = _owner;
+            _this.scrolled_window = this;
             this.el = new Gtk.ScrolledWindow( null, null );
 
             // my vars (dec)
@@ -68,7 +146,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);
             }
         }
 
@@ -81,6 +159,8 @@ public class BrowserWindow : Object
 
 
             // my vars (def)
+        public bool print_started;
+        public int load_try;
 
         // ctor
         public Xcls_view(BrowserWindow _owner )
@@ -90,77 +170,252 @@ public class BrowserWindow : Object
             this.el = new WebKit.WebView();
 
             // my vars (dec)
+            this.print_started = false;
+            this.load_try = 0;
 
             // set gobject values
 
-            //listeners
-            this.el.resource_request_starting.connect( (p0, webres, netreq, netres ) => {
-              print("resource req. started %s\n", netreq.uri);
-              netres.
-              print("resource req. started %s\n", pnetreq.uri);
-            });
-            this.el.load_finished.connect( ( ) => {
-                        
-                   this.printit();
-             
-            });
+            // init method
+
+            {
+                  this.load_try = 0;
+            
+            #if GTK3
+                   this.el.load_changed.connect( (ev ) => {
+                     if (ev != WebKit.LoadEvent.FINISHED) {
+                           return;      
+                       }
+                                 
+                           this.printit(); 
+                         
+                         
+                        });
+            
+            #else
+             //listeners
+                      //  this.el.resource_request_starting.connect( (p0, webres, netreq ) => {
+                        //  print("resource req. started %s\n", netreq.uri);
+                             
+                      //  });
+                        
+                        this.el.load_finished.connect( ( ) => {
+                               
+                               
+                               this.printit(); 
+                         
+                        });
+            #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;
+          }
+           
          
-                       // inject any javascript if needed..
-                       if (webkitpdf.opt_inject_js != null) {
-                               for(var i =0; i < webkitpdf.opt_inject_js.length;i++) {
-                                       string str;
-                                       FileUtils.get_contents( webkitpdf.opt_inject_js[i], out str);
-                                       this.el.execute_script(str);
-                               }
-                       }
-         
-                       print("calling print in %d seconds\n", (int)webkitpdf.opt_delay);
-                GLib.Timeout.add_seconds(webkitpdf.opt_delay, () => { 
-                        
-                       var  pe = new  Gtk.PrintOperation();
-                       pe.export_filename = webkitpdf.opt_target_pdf;
-                               print("got callback for print");
-                       pe.ref();
-                               /*pe.failed.connect(() => {
-                                       print("print failed\n");
-                                        Gtk.main_quit();
-                               
-                       })
-                       */;
-                        pe.done.connect(() => {
-                                       print("print completed\n");
-                                        Gtk.main_quit();
-                               
-                       });
+            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
+         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    
+            
+        }
+        public bool printpdf () {
+        
+        
+        
+            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 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.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();
                        
-                       //pe.run_dialog(_this.el);
-                
+            })
+            */;
+             pe.done.connect(() => {
+               print("print completed\n");
+                Gtk.main_quit();
                        
-                           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);
-                               }
-                           }
-                               
+            });
+                       
+            //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...
+            }
+        #endif     
+                   
+                    
+            return false;
+        }
+        public void printpng () {
+            //var filename = "/tmp/test.pdf";
+            //print("load_changed %d ?= %d\n", le, LoadEvent.FINISHED);
+        
+            //if (le != LoadEvent.FINISHED) {
+            //    return;
+           // }
+           
+        // what size is the documet.
+        
+             print("making screenshot\n");
+        #if GTK3
+        
+        #else     
+            // fix vapi - get_snapshot add '?' to all null.
+            var pixmap = _this.view.el.get_snapshot( null );
+        
+            
+            
+            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);
+                
+            
+            
+             pixbuf.save(webkitpdf.opt_target_png, "png");
+             print("Saved to %s\n", webkitpdf.opt_target_png);
+            //view.get_snapshot.begin(WebKit.SnapshotRegion.FULL_DOCUMENT, WebKit.SnapshotOptions.NONE, null, (obj, res) => {
+            //    var sf = view.get_snapshot.end(res);
+        
+            //    sf.write_to_png(Browser.opt_target_png);
+            //});
+            Gtk.main_quit();
+             
+        #endif   
+            
+            
         
-         
-                           pe.set_default_page_setup(psetup);
-                           print("Calling Print?\n");
-                           //pe.run_dialog(_this.el);
-                           this.el.get_main_frame().print_full(pe,Gtk.PrintOperationAction.EXPORT);
-                           
-                           
-                            
-                               return false;
-                }, GLib.Priority.DEFAULT);
         }
     }