git.roojs.org Git - app.webkitpdf/blob - BrowserWindow.vala

   1 static BrowserWindow  _BrowserWindow;
   2
   3 public class BrowserWindow : Object
   4 {
   5     public Gtk.Window el;
   6     private BrowserWindow  _this;
   7
   8     public static BrowserWindow singleton()
   9     {
  10         if (_BrowserWindow == null) {
  11             _BrowserWindow= new BrowserWindow();
  12         }
  13         return _BrowserWindow;
  14     }
  15     public Xcls_scrolled_window scrolled_window;
  16     public Xcls_view view;
  17
  18         // my vars (def)
  19
  20     // ctor
  21     public BrowserWindow()
  22     {
  23         _this = this;
  24         this.el = new Gtk.Window( Gtk.WindowType.TOPLEVEL );
  25
  26         // my vars (dec)
  27
  28         // set gobject values
  29         var child_0 = new Xcls_scrolled_window( _this );
  30         child_0.ref();
  31         this.el.add (  child_0.el  );
  32
  33         //listeners
  34         this.el.destroy.connect( () => {
  35             Gtk.main_quit();
  36         });
  37         this.el.show.connect( () => {
  38
  39                 print("resizing window\n");
  40                 var w = webkitpdf.opt_width > 0 ?  webkitpdf.opt_width  : 1200;
  41                 var h = webkitpdf.opt_height > 0 ?  webkitpdf.opt_height  :  500;
  42
  43
  44                 this.el.resize(w,h);
  45
  46                 // we have to do silly stuff with blogspot!!?!?
  47                 var blog = webkitpdf.opt_url;
  48
  49
  50           /* Do not redirect if the domain is .com already */
  51           if (/\.blogspot\.com/.match(blog)) {
  52                         var r =  new Regex ("\\.blogspot\\..*?/");
  53                         blog = r.replace(blog, blog.length, 0, ".blogspot.com/ncr/");
  54           }
  55
  56                 print("loading url %s\n",blog);
  57                 this.view.el.load_uri(blog);
  58
  59                 // if we have not finished in 25 seconds
  60                 // call printit...
  61                 var cd = webkitpdf.opt_delay > 25 ? webkitpdf.opt_delay : 25;
  62                   GLib.Timeout.add_seconds(1, () => {
  63                         cd--;
  64                         print("timeout %d seconds :%d", (int)( webkitpdf.opt_delay > 25 ? webkitpdf.opt_delay : 25), (int)cd);
  65                         if (cd < 1) {
  66                                 print("calling printit - timed out");
  67                                 _this.view.printit();
  68                                 return false;
  69                         }
  70                         return true;
  71
  72                 });
  73
  74
  75         });
  76     }
  77
  78     // user defined functions
  79     public class Xcls_scrolled_window : Object
  80     {
  81         public Gtk.ScrolledWindow el;
  82         private BrowserWindow  _this;
  83
  84
  85             // my vars (def)
  86
  87         // ctor
  88         public Xcls_scrolled_window(BrowserWindow _owner )
  89         {
  90             _this = _owner;
  91             _this.scrolled_window = this;
  92             this.el = new Gtk.ScrolledWindow( null, null );
  93
  94             // my vars (dec)
  95
  96             // set gobject values
  97             var child_0 = new Xcls_view( _this );
  98             child_0.ref();
  99             this.el.add (  child_0.el  );
 100
 101             // init method
 102
 103             {
 104                 this.el.set_policy(Gtk.PolicyType.AUTOMATIC, Gtk.PolicyType.AUTOMATIC);
 105             }
 106         }
 107
 108         // user defined functions
 109     }
 110     public class Xcls_view : Object
 111     {
 112         public WebKit.WebView el;
 113         private BrowserWindow  _this;
 114
 115
 116             // my vars (def)
 117         public bool print_started;
 118         public string cookie_file;
 119         public int load_try;
 120
 121         // ctor
 122         public Xcls_view(BrowserWindow _owner )
 123         {
 124             _this = _owner;
 125             _this.view = this;
 126             this.el = new WebKit.WebView();
 127
 128             // my vars (dec)
 129             this.print_started = false;
 130             this.load_try = 0;
 131
 132             // set gobject values
 133
 134             // init method
 135
 136             {
 137                         this.load_try = 0;
 138
 139
 140
 141
 142             #if GTK3
 143                          this.el.load_changed.connect( (ev ) => {
 144                                 if (ev != WebKit.LoadEvent.FINISHED) {
 145                                         return;
 146                                   }
 147
 148                            this.printit();
 149
 150
 151                         });
 152
 153                                 this.el.resource_load_started.connect( (resource, request) => {
 154                                 //      print("Adding referrer header %s\n" , request.get_uri());
 155
 156                                         if (request.get_http_headers() != null) {
 157                                                 request.get_http_headers().remove("Referer");
 158                                                 request.get_http_headers().append("Referer", request.get_uri());
 159                                         }
 160                         });
 161                         var settings = this.el.get_settings();
 162                                 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" );
 163                                 settings.hardware_acceleration_policy = WebKit.HardwareAccelerationPolicy.NEVER;
 164                                 settings.enable_accelerated_2d_canvas = false;
 165                                 if (webkitpdf.opt_disable_javascript) {
 166                                         settings.enable_javascript  = false;
 167                                 }
 168
 169                                 WebKit.WebContext.get_default().set_tls_errors_policy(  WebKit.TLSErrorsPolicy.IGNORE );
 170
 171             #else
 172              //listeners
 173                       //  this.el.resource_request_starting.connect( (p0, webres, netreq ) => {
 174                         //  print("resource req. started %s\n", netreq.uri);
 175
 176                       //  });
 177
 178                         this.el.load_finished.connect( ( ) => {
 179
 180                                this.printit();
 181
 182                         });
 183                         WebKit.get_default_session().request_started.connect((message, socket) => {
 184                             Soup.URI uri = message.get_uri();
 185
 186                                         var host = uri.get_host ();
 187                                         if (host.contains("google-analytics") || host.contains("googlesyndication") || host.contains("facebook") ) {
 188                                                 print("DISCONNECTING: %s\n", host);
 189                                                 socket.disconnect();
 190                                                 return;
 191                                         }
 192
 193                                         print("Adding header: %s\n", uri.to_string(false));
 194                                         message.request_headers.remove("Referer");
 195                                         message.request_headers.append("Referer", "http://"+host+"/");
 196
 197
 198                         });
 199                         var sess = WebKit.get_default_session();
 200                                 sess.user_agent= "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/63.0.3239.84 Safari/537.36";
 201                                 sess.ssl_strict = false;
 202
 203                         var settings = this.el.get_settings();
 204
 205                                 settings.enable_webgl = false;
 206                                 settings.enable_plugins  = false;
 207                                 settings.enable_java_applet  = false;
 208                                 if (webkitpdf.opt_disable_javascript) {
 209                                         settings.enable_scripts  = false;
 210                                 }
 211
 212
 213                                 this.el.console_message.connect((msg, line, sid) => {
 214                                         print("CONSOLE: %s:%d %s\n", sid, line, msg);
 215                                         return false;
 216                                 });
 217                                 this.el.create_web_view.connect((frame) => {
 218                                         print("WEBFRAME:%s\n", frame.uri);
 219                                         return new WebKit.WebView();
 220                                 });
 221
 222             #endif
 223             }
 224                         this.cookie_file = "";
 225                         if (webkitpdf.opt_cookies != null) {
 226                                 print("setting cookie %s\n",webkitpdf.opt_cookies);
 227
 228                                 var uri =   GLib.Uri.parse (webkitpdf.opt_url, GLib.UriFlags.NONE);
 229
 230                                 string[] cookies = webkitpdf.opt_cookies.split ("=", 2);
 231
 232                                 if(cookies.length > 1 ) {
 233
 234             #if GTK3
 235                                         var context = WebKit.WebContext.get_default();
 236                                         var cookie_manager = context.get_cookie_manager();
 237                                         cookie_manager.set_accept_policy(WebKit.CookieAcceptPolicy.NEVER);
 238                                         string scheme = uri.get_scheme ();
 239
 240                                         try {
 241                                             FileIOStream iostream;
 242                                             File file = File.new_tmp ("cookie-XXXXXX.txt", out iostream);
 243                                             this.cookie_file = file.get_path ();
 244                                             print("cookie tmp file name: %s\n", file.get_path ());
 245
 246                                             OutputStream ostream = iostream.output_stream;
 247                                             DataOutputStream dostream = new DataOutputStream (ostream);
 248                                             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]));
 249                                             cookie_manager.set_persistent_storage(file.get_path(), WebKit.CookiePersistentStorage.TEXT);
 250                                         } catch (Error e) {
 251                                             stdout.printf ("Error: %s\n", e.message);
 252                                         }
 253             #else
 254                                         var cookie_jar = new Soup.CookieJar();
 255                                         var cookie = new Soup.Cookie (cookies[0], cookies[1], uri.get_host(), "/", -1);
 256                                         if(uri.get_scheme() =="https") {
 257                                             cookie.set_secure(true);
 258                                         }
 259                                         cookie_jar.set_accept_policy(Soup.CookieJarAcceptPolicy.NEVER);
 260                                         cookie_jar.add_cookie(cookie);
 261                                         WebKit.get_default_session().add_feature(cookie_jar);
 262             #endif
 263
 264                                 }
 265
 266                         }
 267                         // init method
 268         }
 269
 270         // user defined functions
 271         public bool printpdf () {
 272
 273
 274
 275                         var list = Gtk.PaperSize.get_paper_sizes(false);
 276                     var psetup = new Gtk.PageSetup();
 277                     for(var i = 0; i < list.length(); i++ ) {
 278                         var entry = list.nth_data(i).copy();
 279                         if (entry.get_name() == "iso_a2") {
 280                             psetup.set_paper_size(entry);
 281                         }
 282                     }
 283
 284
 285         #if GTK3
 286
 287                         var  pe = new  WebKit.PrintOperation(this.el);
 288                         pe.set_page_setup(psetup);
 289                         pe.finished.connect( () => {
 290                                 print("print completed\n");
 291                     this.delete_cookiefile();
 292                                  Gtk.main_quit();
 293
 294                         });
 295
 296                         var ps = new Gtk.PrintSettings();
 297                     ps.set_printer("Print to File");
 298                     ps.set("output-file-format", "pdf");
 299                     ps.set("output-uri", "file://" + webkitpdf.opt_target_pdf);
 300
 301
 302                         pe.set_print_settings(ps);
 303
 304
 305                         pe.print();
 306
 307
 308         #else
 309                         var  pe = new  Gtk.PrintOperation();
 310                         pe.export_filename = webkitpdf.opt_target_pdf;
 311                         print("got callback for print\n");
 312                         pe.ref();
 313                         /*pe.failed.connect(() => {
 314                                 print("print failed\n");
 315                                  Gtk.main_quit();
 316
 317                         })
 318                         */;
 319                          pe.done.connect(() => {
 320                                 print("print completed\n");
 321                     this.delete_cookiefile();
 322                                  Gtk.main_quit();
 323
 324                         });
 325
 326                         //pe.run_dialog(_this.el);
 327
 328
 329
 330
 331
 332                     pe.set_default_page_setup(psetup);
 333                     print("Calling Print?\n");
 334                     //pe.run_dialog(_this.el);
 335
 336
 337
 338
 339
 340                     try {
 341                             this.el.get_main_frame().print_full(pe,Gtk.PrintOperationAction.EXPORT);
 342                     } catch (Error e) {
 343                                 // print failed...
 344                         }
 345         #endif
 346
 347
 348                         return false;
 349         }
 350         public void printpng () {
 351             //var filename = "/tmp/test.pdf";
 352             //print("load_changed %d ?= %d\n", le, LoadEvent.FINISHED);
 353
 354             //if (le != LoadEvent.FINISHED) {
 355             //    return;
 356            // }
 357
 358         // what size is the documet.
 359
 360              print("making screenshot\n");
 361         #if GTK3
 362         GLib.MainLoop mainloop = new GLib.MainLoop();
 363           _this.view.el.get_snapshot.begin(WebKit.SnapshotRegion.FULL_DOCUMENT, WebKit.SnapshotOptions.NONE, null, (obj, res) => {
 364                          var sf = _this.view.el.get_snapshot.end(res);
 365                          sf.write_to_png(webkitpdf.opt_target_png);
 366                 print("Saved to %s\n", webkitpdf.opt_target_png);
 367                          mainloop.quit();
 368
 369           });
 370
 371                 mainloop.run();
 372
 373         #else
 374             // fix vapi - get_snapshot add '?' to all null.
 375             var pixmap = _this.view.el.get_snapshot( null );
 376
 377
 378
 379             int w,h;
 380
 381             // add out to gdkpixmap.get_size
 382                 pixmap.get_size( out  w , out   h);
 383                 var pixbuf  = Gdk.pixbuf_get_from_drawable(null, pixmap, null , 0, 0, 0, 0, w, h);
 384                 print("pixbuf size: %d x %d\n ", w,h);
 385
 386
 387
 388              pixbuf.save(webkitpdf.opt_target_png, "png");
 389              print("Saved to %s\n", webkitpdf.opt_target_png);
 390             //view.get_snapshot.begin(WebKit.SnapshotRegion.FULL_DOCUMENT, WebKit.SnapshotOptions.NONE, null, (obj, res) => {
 391             //    var sf = view.get_snapshot.end(res);
 392
 393             //    sf.write_to_png(Browser.opt_target_png);
 394             //});
 395         #endif
 396            this.delete_cookiefile();
 397
 398             Gtk.main_quit();
 399
 400
 401
 402
 403
 404         }
 405         public void delete_cookiefile () {
 406                 if(this.cookie_file.length < 1 ){
 407                         return;
 408                 }
 409             print("deleting tmp file %s\n",this.cookie_file);
 410             File file = File.new_for_path (this.cookie_file);
 411             try {
 412                 if (file.query_exists () == true) {
 413                    file.delete ();
 414                 }
 415             } catch (Error e) {
 416                 stdout.printf ("Error: %s\n", e.message);
 417             }
 418
 419         }
 420         public void printit () {
 421            print("load_finsihed\n");
 422
 423            if (this.print_started) {
 424                print("load_finsihed - already printing..%d %s\n", this.load_try, this.el.uri);
 425
 426
 427
 428
 429               return;
 430           }
 431
 432
 433             this.print_started = true; // flag to stop double call.
 434
 435
 436                 // inject any javascript if needed..
 437                 if (webkitpdf.opt_inject_js != null) {
 438
 439                         print("injecting javascript files \n");
 440                         foreach(var inject_js in  webkitpdf.opt_inject_js) {
 441                         print("inject? %s\n", inject_js);
 442                                 string str;
 443                                 try {
 444                                         FileUtils.get_contents( inject_js, out str);
 445                                 } catch(FileError e) {
 446                                         print("Failed to inject %s\n",  inject_js);
 447                                         continue;
 448                                 }
 449         #if GTK3
 450                                 GLib.MainLoop mainloop = new GLib.MainLoop();
 451                                 this.el.run_javascript.begin(str, null, (obj,res) => {
 452                                         mainloop.quit();
 453                                 });
 454                                 mainloop.run();
 455
 456         #else
 457                                 this.el.execute_script(str);
 458         #endif
 459                         }
 460                 }
 461
 462                 print("calling print in %d seconds\n", (int)webkitpdf.opt_delay);
 463                 if (webkitpdf.opt_target_pdf != null) {
 464                         GLib.Timeout.add_seconds(webkitpdf.opt_delay, () => {
 465                                 print("delay done URL: %s\n", this.el.uri);
 466
 467
 468                                 this.printpdf();
 469                                  return false;
 470                         }, GLib.Priority.DEFAULT);
 471                         return;
 472                 }
 473
 474
 475         #if GTK3
 476                 GLib.Timeout.add_seconds(webkitpdf.opt_delay, () => {
 477                                 print("delay done URL: %s\n", this.el.uri);
 478                                 this.printpng();
 479                                 return false;
 480                         }, GLib.Priority.DEFAULT);
 481                         return;
 482         #else
 483
 484             GLib.Timeout.add_seconds(webkitpdf.opt_delay, () => {
 485
 486
 487                         // ----------------- JPEG
 488                         // resize the window...
 489                         // window.document.documentElement.scrollHeight
 490                         var scroll_height = (int) _this.view.el.get_dom_document().document_element.scroll_height;
 491                         print("Scroll height %d\n", scroll_height);
 492                         if (scroll_height> 1024 || webkitpdf.opt_width > 0  || webkitpdf.opt_height > 0  ) {
 493                                 var w = webkitpdf.opt_width > 0 ?  webkitpdf.opt_width  : 1200;
 494                                 var h = int.min(scroll_height, 6000);
 495                                 _this.scrolled_window.el.set_size_request( w, h);
 496                                 _this.el.resize (w+50, h+50);
 497                                 print("Resize to %d, %d\n", w,h);
 498                             GLib.Timeout.add_seconds(webkitpdf.opt_delay > 0 ? webkitpdf.opt_delay : 1 , () => {
 499                                         this.printpng();
 500                                         return false;
 501                                   }, GLib.Priority.DEFAULT);
 502                             return false;
 503                         }
 504                         this.printpng();
 505                         return false;
 506                   }, GLib.Priority.DEFAULT);
 507
 508
 509         #endif
 510
 511         }
 512     }
 513
 514
 515 }