Archiver now saves redirects

archiver/archiver-cli.rkt

@@ -35,9 +35,12 @@
           (output-lines? #t)]))
  (define (update-width)
    (when (output-progress?)
-     (with-charterm
+     (case (system-type 'os)
-       (call-with-values (λ () (charterm-screen-size))
+       [(linux)
-                         (λ (cols rows) (set! width cols))))))
+        (with-charterm
+          (call-with-values (λ () (charterm-screen-size))
+                            (λ (cols rows) (set! width cols))))]
+       [else 100])))
  (update-width)
  ;; check
  (when (or (not wikiname) (equal? wikiname ""))
@@ -56,8 +59,8 @@
       (define real-width (min (string-length basename) rest))
       (define spare-width (- rest real-width))
       (define name-display (substring basename 0 real-width))
-      (define whitespace (make-string spare-width #\ ))
+      (printf "\e[2K\r~a~a" prefix name-display)
-      (printf "~a~a~a\r" prefix name-display whitespace)]))
+      (flush-output)]))
  ;; download all stages
  (for ([stage all-stages]
        [i (in-naturals 1)])

archiver/archiver-database.rkt

@@ -48,7 +48,8 @@
         ((query-exec slc* "alter table wiki add column sitename TEXT")
          (query-exec slc* "alter table wiki add column basepage TEXT")
          (query-exec slc* "alter table wiki add column license_text TEXT")
-         (query-exec slc* "alter table wiki add column license_url TEXT"))))
+         (query-exec slc* "alter table wiki add column license_url TEXT"))
+        ((query-exec slc* "alter table page add column redirect"))))
 
      (let do-migrate-step ()
        (when (database-version . < . (length migrations))

archiver/archiver.rkt

@@ -1,5 +1,6 @@
 #lang racket/base
 (require racket/file
+         racket/format
          racket/function
          racket/list
          racket/path
@@ -43,10 +44,18 @@
             wikiname
             (params->query '(("action" . "query")
                              ("meta" . "siteinfo")
-                             ("siprop" . "general|rightsinfo|statistics")
+                             ("siprop" . "general|rightsinfo|statistics|namespaces")
                              ("format" . "json")
                              ("formatversion" . "2")))))
   (define data (response-json (get dest-url)))
+  (define content-nss
+    (sort
+     (for/list ([(k v) (in-hash (jp "/query/namespaces" data))]
+                #:do [(define id (hash-ref v 'id))]
+                #:when (and (id . < . 2900) ; exclude maps namespace
+                            (hash-ref v 'content))) ; exclude non-content and talk namespaces
+       id)
+     <))
   (define exists? (query-maybe-value* "select progress from wiki where wikiname = ?" wikiname))
   (if (and exists? (not (sql-null? exists?)))
       (query-exec* "update wiki set sitename = ?, basepage = ?, license_text = ?, license_url = ? where wikiname = ?"
@@ -61,7 +70,8 @@
                   (second (regexp-match #rx"/wiki/(.*)" (jp "/query/general/base" data)))
                   (jp "/query/rightsinfo/text" data)
                   (jp "/query/rightsinfo/url" data)))
-  (jp "/query/statistics/articles" data))
+  (values (jp "/query/statistics/articles" data)
+          content-nss))
 
 
 (define (check-style-for-images wikiname path)
@@ -131,48 +141,57 @@
   ;; done yet?
   (unless (and (real? wiki-progress) (wiki-progress . >= . 1))
     ;; Count total pages
-    (define num-pages (insert-wiki-entry wikiname))
+    (define-values (num-pages namespaces) (insert-wiki-entry wikiname))
     ;; Download the entire index of pages
-    (define basenames
+    (for*/fold ([total 0])
-      (let loop ([path-with-namefrom "/wiki/Local_Sitemap"]
+               ([namespace namespaces]
-                 [basenames-previous-pages null])
+                [redir-filter '("nonredirects" "redirects")])
-        ;; Download the current index page
+      (let loop ([apcontinue ""]
-        (define url (format "https://~a.fandom.com~a" wikiname path-with-namefrom))
+                 [basenames null])
-        (define r (get url))
+        (cond
-        ;; Metadata from this page (the link to the next page)
+          [apcontinue
-        (define page (html->xexp (bytes->string/utf-8 (response-body r))))
+           (define url (format "https://~a.fandom.com/api.php?~a"
-        (define link-namefrom
+                               wikiname
-          ((query-selector (λ (t a c x) (and (eq? t 'a)
+                               (params->query `(("action" . "query")
-                                             (pair? x)
+                                                ("list" . "allpages")
-                                             (string-contains? (car x) "Next page")
+                                                ("apnamespace" . ,(~a namespace))
-                                             (let ([href (get-attribute 'href a)] )
+                                                ("apfilterredir" . ,redir-filter)
-                                               (and href (string-contains? href "/wiki/Local_Sitemap")))))
+                                                ("aplimit" . "500")
-                           page #:include-text? #t)))
+                                                ("apcontinue" . ,apcontinue)
-        ;; Content from this page
+                                                ("format" . "json")
-        (define basenames-this-page
+                                                ("formatversion" . "2")))))
-          (for/list ([link (in-producer
+           ;; Download the current listing page
-                            (query-selector
+           (define res (get url))
-                             (λ (t a c) (eq? t 'a))
+           (define json (response-json res))
-                             ((query-selector (λ (t a c) (has-class? "mw-allpages-chunk" a)) page)))
+           ;; Content from this page
-                            #f)])
+           (define current-basenames
-            (local-encoded-url->basename (get-attribute 'href (bits->attributes link)))))
+             (for/list ([page (jp "/query/allpages" json)])
-        ;; Call the progress callback
+               (title->basename (jp "/title" page))))
-        (define all-basenames (append basenames-previous-pages basenames-this-page))
+           (when ((length current-basenames) . > . 0)
-        (callback (length all-basenames) num-pages (last all-basenames))
+             ;; Report
-        ;; Recurse to download from the next page
+             (if (equal? redir-filter "nonredirects")
-        (if link-namefrom
+                 (callback (+ (length basenames) (length current-basenames) total) num-pages (last current-basenames))
-            (loop (get-attribute 'href (bits->attributes link-namefrom)) all-basenames)
+                 (callback total num-pages (last current-basenames))))
-            all-basenames)))
+           ;; Loop
-    ;; Save those pages into the database
+           (loop (jp "/continue/apcontinue" json #f) (append basenames current-basenames))]
-    ;; SQLite can have a maximum of 32766 parameters in a single query
+          [else
-    (for ([slice (in-slice 32760 basenames)])
+           ;; All done with this (loop)! Save those pages into the database
-      (define query-template (string-join (make-list (length slice) "(?1, ?, 0)") ", " #:before-first "insert or ignore into page (wikiname, basename, progress) values "))
+           ;; SQLite can have a maximum of 32766 parameters in a single query
-      (call-with-transaction
+           (begin0
-       (get-slc)
+               ;; next for*/fold
-       (λ ()
+               (if (equal? redir-filter "nonredirects")
-         (apply query-exec* query-template wikiname slice)
+                   (+ (length basenames) total)
-         ;; Record that we have the complete list of pages
+                   total) ; redirects don't count for the site statistics total
-         (query-exec* "update wiki set progress = 1 where wikiname = ?" wikiname))))))
+             (call-with-transaction
+              (get-slc)
+              (λ ()
+                (for ([slice (in-slice 32760 basenames)])
+                  (define query-template
+                    (string-join #:before-first "insert or ignore into page (wikiname, redirect, basename, progress) values "
+                                 (make-list (length slice) "(?1, ?2, ?, 0)") ", "))
+                  (apply query-exec* query-template wikiname (if (equal? redir-filter "redirects") 1 sql-null) slice)))))])))
+    ;; Record that we have the complete list of pages
+    (query-exec* "update wiki set progress = 1 where wikiname = ?" wikiname)))
 
 
 ;; 2. Download each page via API and:
@@ -183,7 +202,7 @@
   (define save-dir (build-path archive-root wikiname))
   (make-directory* save-dir)
   ;; gather list of basenames to download (that aren't yet complete)
-  (define basenames (query-list* "select basename from page where wikiname = ? and progress < ?"
+  (define basenames (query-list* "select basename from page where wikiname = ? and progress < ? and redirect is null"
                                 wikiname max-page-progress))
   ;; counter of complete/incomplete basenames
   (define already-done-count
@@ -222,10 +241,41 @@
     (query-exec* "update page set progress = 1 where wikiname = ? and basename = ?"
                 wikiname basename)
     (callback i total-count basename))
+  ;; save redirects as well
+  (save-redirects wikiname callback (+ already-done-count (length basenames)) total-count)
   ;; saved all pages, register that fact in the database
   (query-exec* "update wiki set progress = 2 where wikiname = ?" wikiname))
 
 
+;; 2.5. Download each redirect-target via API and save mapping in database
+(define (save-redirects wikiname callback already-done-count total-count)
+  (define basenames (query-list* "select basename from page where wikiname = ? and progress < ? and redirect = 1"
+                                 wikiname max-page-progress))
+  ;; loop through basenames, in slices of 50 (MediaWiki API max per request), and download
+  (for ([basename basenames]
+        [i (in-naturals (add1 already-done-count))])
+    (define dest-url
+      (format "https://~a.fandom.com/api.php?~a"
+              wikiname
+              (params->query `(("action" . "query")
+                               ("prop" . "links")
+                               ("titles" . ,(basename->name-for-query basename))
+                               ("format" . "json")
+                               ("formatversion" . "2")))))
+    (define res (get dest-url))
+    (define json (response-json res))
+    (define dest-title (jp "/query/pages/0/links/0/title" json #f))
+    (callback i total-count basename)
+    (cond
+      [dest-title
+       ;; store it
+       (define dest-basename (title->basename dest-title))
+       (query-exec* "update page set progress = 1, redirect = ? where wikiname = ? and basename = ?" dest-basename wikiname basename)]
+      [else
+       ;; the page just doesn't exist
+       (query-exec* "delete from page where wikiname = ? and basename = ?" wikiname basename)])))
+
+
 ;; 3. Download CSS and:
 ;; * Save CSS to file
 ;; * Record style images to database

lib/archive-file-mappings.rkt

@@ -7,6 +7,7 @@
  local-encoded-url->segments
  url-segments->basename
  local-encoded-url->basename
+ title->basename
  basename->name-for-query
  url-segments->guess-title)
 
@@ -21,6 +22,12 @@
 (define (local-encoded-url->basename str) ; '("wiki" "Page_title"), no extension or dir prefix
   (url-segments->basename (local-encoded-url->segments str)))
 
+(define (title->basename title) ; "Page title/Strategies" -> "Page_title#Strategies" filename encoded, no extension or dir prefi
+  (define elements (string-split (string-replace title " " "_") "/"))
+  (define extra-encoded (map (λ (s) (bytes->string/latin-1 (percent-encode s filename-set #f))) elements))
+  (define basic-filename (string-join extra-encoded "#"))
+  basic-filename)
+
 (define (basename->name-for-query str)
   (uri-decode (regexp-replace* #rx"#" str "/")))
 

src/extwiki-data.rkt

@@ -377,7 +377,7 @@
     '("runescape") 'default
     'RuneScape
     "RuneScape Wiki"
-    "https://runescape.wiki/"
+    "https://runescape.wiki/w/Main_Page"
     "https://runescape.wiki/images/Wiki.png"
     (λ (props)
       `((p "The RuneScape Wiki was founded on April 8, 2005. In October 2018, the wiki left Fandom (then Wikia), citing their apathy towards the wiki and excessive advertisements."))))
@@ -386,7 +386,7 @@
     '("oldschoolrunescape") 'default
     'RuneScape
     "Old School RuneScape Wiki"
-    "https://oldschool.runescape.wiki/"
+    "https://oldschool.runescape.wiki/w/Main_Page"
     "https://oldschool.runescape.wiki/images/Wiki.png"
     (λ (props)
       `((p "The Old School RuneScape Wiki was founded on February 14, 2013. In October 2018, the RuneScape Wiki left Fandom (then Wikia), citing their apathy towards the wiki and excessive advertisements, with the Old School RuneScape Wiki following suit."))))
@@ -395,7 +395,7 @@
     '("runescapeclassic") 'default
     'RuneScape
     "RuneScape Classic Wiki"
-    "https://classic.runescape.wiki/"
+    "https://classic.runescape.wiki/w/Main_Page"
     "https://classic.runescape.wiki/images/Wiki.png"
     (λ (props)
       `((p "The Old School RuneScape Wiki was founded on April 19, 2009. In October 2018, the RuneScape Wiki left Fandom (then Wikia), citing their apathy towards the wiki and excessive advertisements, with the RuneScape Classic Wiki following suit."))))

src/page-global-search.rkt

@@ -19,7 +19,7 @@
      [(not wikiname)
       (response/output
        #:code 400
-       #:mime-type "text/plain"
+       #:mime-type #"text/plain"
        (λ (out)
          (displayln "Requires wikiname and q parameters." out)))]
      [(or (not q) (equal? q ""))

src/page-search.rkt

@@ -53,7 +53,8 @@
    ;; grab the part after ?q= which is the search terms
    (define query (dict-ref params 'q #f))
    ;; figure out which search provider we're going to use
-   (define search-provider (hash-ref search-providers (config-get 'feature_offline::search)))
+   (define search-provider (hash-ref search-providers (config-get 'feature_offline::search)
+                                     (λ () (error 'search-provider "unknown search provider configured"))))
 
    ;; external special:search url to link at the bottom of the page as the upstream source
    (define external-search-url

src/page-wiki-offline.rkt

@@ -20,6 +20,7 @@
          web-server/dispatchers/dispatch
          ; my libs
          "application-globals.rkt"
+         "../archiver/archiver-database.rkt"
          "config.rkt"
          "data.rkt"
          "log.rkt"
@@ -40,6 +41,9 @@
 
 (define path-archive (anytime-path ".." "storage/archive"))
 
+(when (config-true? 'feature_offline::only)
+  (void (get-slc)))
+
 (define (page-wiki-offline req)
   (response-handler
    (define wikiname (path/param-path (first (url-path (request-uri req)))))
@@ -64,84 +68,94 @@
        [else (error 'archive-format "unknown archive format configured")]))
    (define fs-path (build-path path-archive wikiname (format (car archive-format) maybe-hashed-basename)))
    (define source-url (format "https://~a.fandom.com/wiki/~a" wikiname (basename->name-for-query basename)))
-   (cond
+   (cond/var
-     [(not (file-exists? fs-path))
+
-      (unless (config-true? 'feature_offline::only)
+    [(file-exists? fs-path)
-        (next-dispatcher))
+     (when (config-true? 'debug)
-      (define mirror-path (url->string (request-uri req)))
+       (printf "using offline mode for ~v~n" fs-path))
+     (response-handler
+      (define data (with-input-from-file fs-path (cdr archive-format)))
+      (define article-title (jp "/parse/title" data))
+      (define original-page (html->xexp (preprocess-html-wiki (jp "/parse/text" data))))
+      (define page ((query-selector (λ (t a c) (has-class? "mw-parser-output" a)) original-page)))
+      (define initial-head-data ((head-data-getter wikiname) data))
+      (define head-data
+        (case theme
+          [(light dark)
+           (struct-copy head-data^ initial-head-data
+                        [body-class (regexp-replace #rx"(theme-fandomdesktop-)(light|dark)"
+                                                    (head-data^-body-class initial-head-data)
+                                                    (format "\\1~a" theme))])]
+          [else initial-head-data]))
       (define body
         (generate-wiki-page
-         `(div (@ (class "unsaved-page"))
+         (update-tree-wiki page wikiname)
-               (style ".unsaved-page a { text-decoration: underline !important }")
-               (p "breezewiki.com doesn't have this page saved.")
-               (p "You can see this page by visiting a BreezeWiki mirror:")
-               (ul
-                (li (a (@ (href ,(format "https://antifandom.com~a" mirror-path))) "View on antifandom.com"))
-                (li (a (@ (href ,(format "https://bw.artemislena.eu~a" mirror-path))) "View on artemislena.eu"))
-                (li (a (@ (href ,source-url)) "or, you can see the original page on Fandom (ugh)")))
-               (p "If you'd like " ,wikiname ".fandom.com to be added to breezewiki.com, " (a (@ (href "https://lists.sr.ht/~cadence/breezewiki-requests")) "let me know about it!")))
          #:req req
          #:source-url source-url
          #:wikiname wikiname
-         #:title (url-segments->guess-title segments)
+         #:title article-title
          #:online-styles #f
+         #:head-data head-data
          #:siteinfo (siteinfo-fetch wikiname)
          ))
+      (define redirect-msg ((query-selector (attribute-selector 'class "redirectMsg") body)))
+      (define redirect-query-parameter (dict-ref (url-query (request-uri req)) 'redirect "yes"))
+      (define headers
+        (build-headers
+         always-headers
+         ; redirect-query-parameter: only the string "no" is significant:
+         ; https://github.com/Wikia/app/blob/fe60579a53f16816d65dad1644363160a63206a6/includes/Wiki.php#L367
+         (when (and redirect-msg
+                    (not (equal? redirect-query-parameter "no")))
+           (let* ([dest (get-attribute 'href (bits->attributes ((query-selector (λ (t a c) (eq? t 'a)) redirect-msg))))]
+                  [value (bytes-append #"0;url=" (string->bytes/utf-8 dest))])
+             (header #"Refresh" value)))))
       (when (config-true? 'debug)
         ; used for its side effects
         ; convert to string with error checking, error will be raised if xexp is invalid
         (xexp->html body))
       (response/output
        #:code 200
-       #:headers always-headers
+       #:headers headers
        (λ (out)
-         (write-html body out)))]
+         (write-html body out))))]
-     [#t
+
-      (when (config-true? 'debug)
+    ;; page not found on disk, perhaps it's a redirect? redirects are stored in the database
-        (printf "using offline mode for ~v~n" fs-path))
+    (var target (query-maybe-value* "select redirect from page where wikiname = ? and basename = ?" wikiname basename))
-      (response-handler
+    [target
-       (define data (with-input-from-file fs-path (cdr archive-format)))
+     (generate-redirect (basename->name-for-query target))]
-       (define article-title (jp "/parse/title" data))
+
-       (define original-page (html->xexp (preprocess-html-wiki (jp "/parse/text" data))))
+    ;; breezewiki doesn't have the page archived, see if we can make a network request for it
-       (define page ((query-selector (λ (t a c) (has-class? "mw-parser-output" a)) original-page)))
+    [(not (config-true? 'feature_offline::only))
-       (define initial-head-data ((head-data-getter wikiname) data))
+     (next-dispatcher)]
-       (define head-data
+
-         (case theme
+    ;; no possible way to provide the page
-           [(light dark)
+    [else
-            (struct-copy head-data^ initial-head-data
+     (define mirror-path (url->string (request-uri req)))
-                         [body-class (regexp-replace #rx"(theme-fandomdesktop-)(light|dark)"
+     (define body
-                                                     (head-data^-body-class initial-head-data)
+       (generate-wiki-page
-                                                     (format "\\1~a" theme))])]
+        `(div (@ (class "unsaved-page"))
-           [else initial-head-data]))
+              (style ".unsaved-page a { text-decoration: underline !important }")
-       (define body
+              (p "breezewiki.com doesn't have this page saved.")
-         (generate-wiki-page
+              (p "You can see this page by visiting a BreezeWiki mirror:")
-          (update-tree-wiki page wikiname)
+              (ul
-          #:req req
+               (li (a (@ (href ,(format "https://antifandom.com~a" mirror-path))) "View on antifandom.com"))
-          #:source-url source-url
+               (li (a (@ (href ,(format "https://bw.artemislena.eu~a" mirror-path))) "View on artemislena.eu"))
-          #:wikiname wikiname
+               (li (a (@ (href ,source-url)) "or, you can see the original page on Fandom (ugh)")))
-          #:title article-title
+              (p "If you'd like " ,wikiname ".fandom.com to be added to breezewiki.com, " (a (@ (href "https://lists.sr.ht/~cadence/breezewiki-requests")) "let me know about it!")))
-          #:online-styles #f
+        #:req req
-          #:head-data head-data
+        #:source-url source-url
-          #:siteinfo (siteinfo-fetch wikiname)
+        #:wikiname wikiname
-          ))
+        #:title (url-segments->guess-title segments)
-       (define redirect-msg ((query-selector (attribute-selector 'class "redirectMsg") body)))
+        #:online-styles #f
-       (define redirect-query-parameter (dict-ref (url-query (request-uri req)) 'redirect "yes"))
+        #:siteinfo (siteinfo-fetch wikiname)
-       (define headers
+        ))
-         (build-headers
+     (when (config-true? 'debug)
-          always-headers
+       ; used for its side effects
-          ; redirect-query-parameter: only the string "no" is significant:
+       ; convert to string with error checking, error will be raised if xexp is invalid
-          ; https://github.com/Wikia/app/blob/fe60579a53f16816d65dad1644363160a63206a6/includes/Wiki.php#L367
+       (xexp->html body))
-          (when (and redirect-msg
+     (response/output
-                     (not (equal? redirect-query-parameter "no")))
+      #:code 200
-            (let* ([dest (get-attribute 'href (bits->attributes ((query-selector (λ (t a c) (eq? t 'a)) redirect-msg))))]
+      #:headers always-headers
-                   [value (bytes-append #"0;url=" (string->bytes/utf-8 dest))])
+      (λ (out)
-              (header #"Refresh" value)))))
+        (write-html body out)))])))
-       (when (config-true? 'debug)
-         ; used for its side effects
-         ; convert to string with error checking, error will be raised if xexp is invalid
-         (xexp->html body))
-       (response/output
-        #:code 200
-        #:headers headers
-        (λ (out)
-          (write-html body out))))])))