embulk-input-sitemap 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: f2373cfc6f002bcd47fa75bd7685d7a97a14257e58341d18f1bcc899c96d9626
+  data.tar.gz: 216200f9d64b0af61ab84baf15cf9f652bffa58e1789ff3317d3ff54f821b548
+SHA512:
+  metadata.gz: b82213ff694d42a2f9df7284222d333ee498704945e01278bdbd91b5f07ea1056c3faa1d9d3a1ca07503c9fde91f2f724900d3cabccab8f8a5f4a96b55e5f54a
+  data.tar.gz: 2e20ed289d7aa0e68351951fa88922fa86c1e94b04f9c3d42d364c9fd4a8d631186a5492b0d77cfe5946e54e5a4a7e15fcc23dc5a4076367006e4eae464901cb

data/.gitignore

@@ -0,0 +1,8 @@
+*~
+/pkg/
+/tmp/
+/.bundle/
+/Gemfile.lock
+/vendor/bundle/
+/workspaces/
+/bin/

data/.ruby-version

@@ -0,0 +1 @@
+jruby-9.1.13.0

data/Gemfile

@@ -0,0 +1,2 @@
+source 'https://rubygems.org/'
+gemspec

data/LICENSE.txt

@@ -0,0 +1,21 @@
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,32 @@
+# Sitemap input plugin for Embulk
+
+Embulk input plugin for sitemap.xml.
+
+## Overview
+
+* **Plugin type**: input
+* **Resume supported**: no
+* **Cleanup supported**: no
+* **Guess supported**: no
+
+## Configuration
+
+- **url**: The sitemap url (string, required)
+- **params**: Query parameter for sitemap url (array, default: [])
+
+## Example
+
+```yaml
+in:
+  type: sitemap
+  url: https://sem-technology.info/sitemap.xml
+  params:
+    - {name: page, value: 1}
+```
+
+
+## Build
+
+```
+$ rake
+```

data/Rakefile

@@ -0,0 +1,3 @@
+require "bundler/gem_tasks"
+
+task default: :build

data/embulk-input-sitemap.gemspec

@@ -0,0 +1,21 @@
+
+Gem::Specification.new do |spec|
+  spec.name          = "embulk-input-sitemap"
+  spec.version       = "0.1.0"
+  spec.authors       = ["ryota.yamada"]
+  spec.summary       = "Sitemap input plugin for Embulk"
+  spec.description   = "Loads records from Sitemap."
+  spec.email         = ["1987yama3@gmail.com"]
+  spec.licenses      = ["MIT"]
+  spec.homepage      = "https://github.com/sem-technology/embulk-input-sitemap"
+
+  spec.files         = `git ls-files`.split("\n") + Dir["classpath/*.jar"]
+  spec.test_files    = spec.files.grep(%r{^(test|spec)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "nokogiri"
+
+  spec.add_development_dependency 'embulk', ['>= 0.8.38']
+  spec.add_development_dependency 'bundler', ['>= 1.10.6']
+  spec.add_development_dependency 'rake', ['>= 10.0']
+end

data/examples/normal.yml

@@ -0,0 +1,5 @@
+in:
+  type: sitemap
+  url: https://sem-technology.info/sitemap.xml
+out:
+  type: stdout

data/lib/embulk/input/sitemap.rb

@@ -0,0 +1,9 @@
+ENV["SSL_CERT_FILE"] = "/Users/yamada/programing/embulk/embulk-input-sitemap/bin/cacert.pem"
+require "embulk/input/sitemap/plugin"
+require "embulk/input/sitemap/client"
+module Embulk
+  module Input
+    module Sitemap
+    end
+  end
+end

data/lib/embulk/input/sitemap/client.rb

@@ -0,0 +1,56 @@
+require "ostruct"
+require "open-uri"
+require "nokogiri"
+require "zlib"
+module Embulk
+  module Input
+    module Sitemap
+      class Client
+        def initialize(url, params)
+          @url = url
+          @params = params
+        end
+
+        def query
+          queries = @params.map{|param| "#{param["name"]}=#{param["value"]}"}
+          return "?" + queries.join("&") if queries.length > 0 and @url.include?("?") == false
+          return "&" + queries.join("&") if queries.length > 0 and @url.include?("?")
+          return ""
+        end
+
+        def invoke
+          items = []
+          Embulk.logger.info "GET #{@url}#{self.query}"
+          response = open(@url + self.query) do |f|
+            Embulk.logger.info "Content-Type = #{f.content_type}"
+            case f.content_type
+              when "application/x-gzip", "application/octet-stream" then
+                `curl #{@url} | gunzip -d`
+              else f.read
+            end
+          end
+          document = Nokogiri::XML(response)
+          sitemaps = document.css("sitemap")
+          urls = document.css("url")
+          Embulk.logger.info "Find #{sitemaps.length} sitemaps, #{urls.length} urls"
+          if sitemaps.length > 0
+            items << sitemaps.collect do |sitemap|
+              Client.new(sitemap.css("loc").first.text.to_s).invoke
+            end
+          end
+          if urls.length > 0
+            items << urls.collect do |url|
+              item = {}
+              item[:loc] = url.css("loc").first.text.to_s
+              item[:changefreq] = url.css("changefreq").first.text.to_s if url.css("changefreq").first
+              item[:priority] = url.css("priority").first.text.to_s if url.css("priority").first
+              item[:lastmod] = url.css("lastmod").first.text.to_s if url.css("lastmod").first
+              OpenStruct.new(item)
+            end
+          end
+          items.flatten
+        end
+      end
+    end
+  end
+end

data/lib/embulk/input/sitemap/plugin.rb

@@ -0,0 +1,52 @@
+module Embulk
+  module Input
+    module Sitemap
+      class Plugin < InputPlugin
+        Embulk::Plugin.register_input("sitemap", self)
+
+        def self.transaction(config, &control)
+          # configuration code:
+          task = {
+            "url" => config.param("url", :string),
+            "params" => config.param("params", :array, default: []),
+          }
+
+          columns = [
+            Column.new(0, "loc", :string),
+            Column.new(1, "lastmod", :string),
+            Column.new(2, "priority", :double),
+            Column.new(3, "changefreq", :string),
+          ]
+
+          resume(task, columns, 1, &control)
+        end
+
+        def self.resume(task, columns, count, &control)
+          task_reports = yield(task, columns, count)
+
+          next_config_diff = {}
+          return next_config_diff
+        end
+
+        def init
+        end
+
+        def run
+          client = Embulk::Input::Sitemap::Client.new(task["url"], task["params"])
+          client.invoke.each do |item|
+            page_builder.add([
+              item.loc,
+              item.lastmod,
+              item.priority,
+              item.changefreq,
+            ])
+          end
+          page_builder.finish
+
+          task_report = {}
+          return task_report
+        end
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,111 @@
+--- !ruby/object:Gem::Specification
+name: embulk-input-sitemap
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- ryota.yamada
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2017-12-12 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  name: nokogiri
+  prerelease: false
+  type: :runtime
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.8.38
+  name: embulk
+  prerelease: false
+  type: :development
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.8.38
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.10.6
+  name: bundler
+  prerelease: false
+  type: :development
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.10.6
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  name: rake
+  prerelease: false
+  type: :development
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '10.0'
+description: Loads records from Sitemap.
+email:
+- 1987yama3@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".ruby-version"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- embulk-input-sitemap.gemspec
+- examples/normal.yml
+- lib/embulk/input/sitemap.rb
+- lib/embulk/input/sitemap/client.rb
+- lib/embulk/input/sitemap/plugin.rb
+homepage: https://github.com/sem-technology/embulk-input-sitemap
+licenses:
+- MIT
+metadata: {}
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project:
+rubygems_version: 2.6.13
+signing_key:
+specification_version: 4
+summary: Sitemap input plugin for Embulk
+test_files: []