mediawiki-butt 0.2.1 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 6ae51034d623d7cf169caf9f5047b7c045657efc
-  data.tar.gz: ff71d9958a84604227b5d97bb57f4dfe23fb28b6
+  metadata.gz: a8d1d049adaacdee763a3b3a741ae4d8c7641813
+  data.tar.gz: 36994491428f1eb43fc994530267b16a3bcf58a2
 SHA512:
-  metadata.gz: 96bb531f1dec544ad84bba39dfa73a13c77b6efbaa7628fb8733972909814e834a700f400490445953f14eb0f2ac2bd3379dece7c87b9294060733bd7fcd1b0d
-  data.tar.gz: c7d7aaa8f0c2cbef183ac3cc79ba3ba68605e6735a59eb60584fcd24745b88c629ab2f1917dcf20123641db7fe3580ce5b1b7db97df4330b2360447b26084a12
+  metadata.gz: 167e862fd9eef9e7212e861c1bad91bca497eaf3c96a3fcd0f16684ea017dbe7bb6d858ac7b29e3760c3a3a2e69062c4dfa22342c54535e2e54197fba2545f20
+  data.tar.gz: dc16fd0e84563e8ca0e2db74ac7604e38ff1dda67f94f242a2dedb511ed884ab940a3aad216ade4a931a9b1b8b82b2f9a50116f9b83a8901e87d2c2ad33d76d7

data/CHANGELOG.md

@@ -1,5 +1,16 @@
 # Changelog
 ## Version 0
+### Version 0.3.0
+* New upload method to upload by URL.
+* New create_page method.
+* New get_userlists method.
+* New get_usergroups method.
+* New get_contrib_count method.
+* Refactor get_usergroups and is_current_user_bot? to use new get_userinfo method.
+* Minor refactors to make optional arguments more Ruby-like with splat arguments.
+* #edit no longer prints the edit token, because that's stupid.
+* #edit no longer sets the summary if it is nil.
+
 ### Version 0.2.1
 * Fix gemspec. You should actually have the new stuff now.
 

data/lib/mediawiki/butt.rb

@@ -40,7 +40,11 @@ module MediaWiki
     # @param header [Hash] The header hash. Optional.
     # @return [JSON/HTTPMessage] Parsed JSON if autoparse is true, or raw response if not.
     def post(params, autoparse = true, header = nil)
-      if header == nil
+      # Note that defining the header argument as a splat argument (*header) causes errors in HTTPClient.
+      # We must use header.nil? rather than a splat argument and defined? header due to this error.
+      # For those interested, the error is: undefined method `downcase' for {"Set-Cookie"=>"cookie"}:Hash (NoMethodError)
+      # This is obvisouly an error in HTTPClient, but we must work around it until there is a fix in the gem.
+      if header.nil?
         response = @client.post(@uri, params)
       else
         response = @client.post(@uri, params, header)
@@ -54,10 +58,16 @@ module MediaWiki
     end
 
     # Returns true if the currently logged in user is in the "bot" group. This can be helpful to some developers, but it is mostly for use internally in MediaWiki::Butt.
+    # @param username [String] The username to check. Optional. Defaults to the currently logged in user if nil.
     # @return [Boolean] true if logged in as a bot, false if not logged in or logged in as a non-bot
-    def is_current_user_bot?
-      if @logged_in == true
+    def is_user_bot?(*username)
+      if defined? username
+        groups = get_usergroups(username)
+      else
         groups = get_usergroups
+      end
+
+      if groups != false
         if groups.include? "bot"
           return true
         else

data/lib/mediawiki/edit.rb

@@ -3,23 +3,22 @@ module MediaWiki
     # Performs a standard edit. It cannot create pages. Please use create_page for that.
     # @param title [String] The page title.
     # @param text [String] The new content.
-    # @param summary [String] The edit summary. Optional.
     # @param minor [Boolean] Will mark the edit as minor if true. Defaults to false.
     # @param bot [Boolean] Will mark the edit as bot edit if true. Defualts to true, for your convenience, bot developers.
+    # @param summary [String] The edit summary. Optional.
     # @return [String] The new revision ID, or if it failed, the error code.
-    def edit(title, text, summary = nil, minor = false, bot = true)
+    def edit(title, text, minor = false, bot = true, *summary)
       params = {
         action: 'edit',
         title: title,
-        summary: summary,
         text: text,
         nocreate: 1,
         format: 'json'
       }
 
       token = get_edit_token(title)
-      puts token
 
+      params[:summary] = summary if defined? summary
       params[:minor] = '1' if minor == true
       params[:bot] = '1' if bot == true
       params[:token] = token
@@ -32,5 +31,66 @@ module MediaWiki
         return response["error"]["code"]
       end
     end
+
+    # Creates a new page.
+    # @param title [String] The new page's title.
+    # @param text [String] The new page's content.
+    # @param summary [String] The edit summary. Defaults to "New page".
+    # @param bot [Boolean] Will mark the edit as a bot edit if true. Defaults to true, for your convenience, bot developers.
+    # @return [String] The new page ID, or if it failed, the error code.
+    def create_page(title, text, summary = "New page", bot = true)
+      params = {
+        action: 'edit',
+        title: title,
+        summary: summary,
+        text: text,
+        createonly: 1,
+        format: 'json'
+      }
+
+      token = get_edit_token(title)
+
+      params[:bot] = '1' if bot == true
+      params[:token] = token
+
+      response = post(params)
+
+      if response["error"]["code"].nil?
+        return response["edit"]["pageid"]
+      else
+        return response["error"]["code"]
+      end
+    end
+
+    # Uploads a file from a URL.
+    # @param url [String] The URL to the file.
+    # @param filename [String] The preferred filename. This can include File: at the beginning, but it will be removed through regex. Optional. If ommitted, it will be everything after the last slash in the URL.
+    # @return [Boolean/String] true if the upload was successful, else the warning key.
+    def upload(url, *filename)
+      params = {
+        action: 'upload',
+        url: url,
+        format: 'json'
+      }
+
+      if defined? filename
+        filename = filename.sub(/$File:/, "")
+      else
+        filename = url.split('/')[-1]
+      end
+
+      token = get_edit_token(filename)
+
+      params[:filename] = filename
+      params[:token] = token
+
+      response = post(params)
+      if response["upload"]["result"] == "Success"
+        return true
+      elsif response["upload"]["result"] == "Warning"
+        warnings = response["upload"]["warnings"]
+        return warnings.keys[0]
+      end
+    end
   end
 end

data/lib/mediawiki/query.rb

@@ -1,3 +1,5 @@
+require 'string-utility'
+
 module MediaWiki
   module Query
     #TODO: Actually decide on a good way to deal with meta information queries.
@@ -21,28 +23,6 @@ module MediaWiki
         end
         return ret
       end
-
-      # Gets an array of all the currently logged in user's groups.
-      # @return [Array/Boolean] All of the user's groups, or false if not logged in.
-      def get_usergroups
-        if @logged_in == true
-          params = {
-            action: 'query',
-            meta: 'userinfo',
-            uiprop: 'groups',
-            format: 'json'
-          }
-
-          result = post(params)
-          ret = Array.new
-          result["query"]["userinfo"]["groups"].each do |g|
-            ret.push(g)
-          end
-          return ret
-        else
-          return false
-        end
-      end
     end
 
     module Properties
@@ -133,7 +113,7 @@ module MediaWiki
         }
 
         if limit > 500
-          if is_current_user_bot? == true
+          if is_user_bot? == true
             if limit > 5000
               params[:bllimit] = 5000
             else
@@ -162,7 +142,7 @@ module MediaWiki
         params = {
           action: 'query',
           list: 'categorymembers',
-          #cmprop: 'title',
+          cmprop: 'title',
           format: 'json'
         }
 
@@ -173,7 +153,7 @@ module MediaWiki
         end
 
         if limit > 500
-          if is_current_user_bot? == true
+          if is_user_bot? == true
             if limit > 5000
               params[:cmlimit] = 5000
             else
@@ -208,7 +188,7 @@ module MediaWiki
         params[:rnnamespace] = namespace if namespace != nil
 
         if namespace > 10
-          if is_current_user_bot? == true
+          if is_user_bot? == true
             if limit > 20
               params[:rnlimit] = 20
             else
@@ -228,6 +208,82 @@ module MediaWiki
         end
         return ret
       end
+
+      # Gets user information. This method should rarely be used by normal users.
+      # @param prop [String] The usprop parameter.
+      # @param username [String] The username to get info for. Optional. Defaults to the currently logged in user if ommitted.
+      # @return [String/Nil] Parsed full response if successful, nil if the username is nil and the Butt is not logged in.
+      def get_userlists(prop, username = nil)
+        params = {
+          action: 'query',
+          list: 'users',
+          usprop: prop,
+          format: 'json'
+        }
+
+        if username.nil?
+          if @logged_in == true
+            response = post(params)
+          else
+            return nil
+          end
+        else
+          params[:ususers] = username
+          response = post(params)
+        end
+
+        return response
+      end
+
+      # Gets an array of all the currently logged in user's groups.
+      # @param username [String] The username to get groups of. Optional. Defaults to the currently logged in user.
+      # @return [Array/Boolean] All of the user's groups, or false if username is nil and Butt is not logged in.
+      def get_usergroups(username = nil)
+        if username.nil?
+          if @logged_in == true
+            info = get_userlists('groups')
+          else
+            return false
+          end
+        else
+          info = get_userlists('groups', username)
+        end
+
+        ret = Array.new
+        info["query"]["users"].each do |i|
+          i["groups"].each do |g|
+            ret.push(g)
+          end
+        end
+        return ret
+      end
+
+      # Gets contribution count for the user.
+      # @param username [String] The username to get the contribution count of. Optional. Defaults to the currently logged in user.
+      # @param autoparse [Boolean] Whether to automatically format the string with commas. Defaults to true.
+      # @return [Boolean/Int/String] False if username is nil and Butt is not logged in. An integer value of the contribution count if autoparse is false. A formatted string version of the contribution count if autoparse is true.
+      def get_contrib_count(username = nil, autoparse = true)
+        if username.nil?
+          if @logged_in == true
+            info = get_userlists('editcount')
+          else
+            return false
+          end
+        else
+          info = get_userlists('editcount', username)
+        end
+
+        count = nil
+        info["query"]["users"].each do |i|
+          count = i["editcount"]
+        end
+
+        if autoparse == true
+          countstring = StringUtility.separate_with(count.to_s)
+          return countstring
+        end
+        return count
+      end
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mediawiki-butt
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.3.0
 platform: ruby
 authors:
 - Eli Foster
@@ -9,10 +9,24 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-09-30 00:00:00.000000000 Z
+date: 2015-10-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
-  name: httpclient
+  name: string-utility
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.0.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.0.1
+- !ruby/object:Gem::Dependency
+  name: json
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -26,7 +40,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: json
+  name: httpclient
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="