iloveck101 0.1.1 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 4c188548911242ccb147a1306b1265c803fc9493
-  data.tar.gz: 2505c8b437d08a50044ee19df88e068883d6f18d
+  metadata.gz: e4d50e2206fa60de311ee10cd31c3c0746990c74
+  data.tar.gz: 190b1fff19e56b4f28cc1387e4bedf3deb2eb399
 SHA512:
-  metadata.gz: 3d31e869a2227b82a6d1ab8a2bca9c3d8c7008de4d1accdc4bddb98e607cacb77b40bf88d2cf30ce04d4f0d59a6ac0cc0efdab58398d566eb2caebc0ce478928
-  data.tar.gz: 06f33e9f9666d0b26233906895e180beb4b273c46b824aa51c7c0a9e2cf82e836ad6b1c96b48ea7a9713b1404a50f95f7bb05c43ffb3266c19d069223275ea4c
+  metadata.gz: 1d5924634255db648efd2b76fb4bb2ec3cd44aa2f97e7698e5f1b12db4badcf0a3a3d7867abf26db5a8e3e720c819d3d2becd2a134be09a8275c892f183d6ea3
+  data.tar.gz: be8d894be8bca6d92606a8b49a7e2e8fcd9869d37afd75c6547dbcb24f11f2a268de8b190dce950bf014c10620d4c6a3884901587615bfb47f8df1372fa92644

data/bin/iloveck101

@@ -1,8 +1,8 @@
 #!/usr/bin/env ruby
 
-require 'open-uri'
 require 'set'
 require 'fileutils'
+require 'mechanize'
 
 def iloveck101(url)
   page_uri = URI.parse(url)
@@ -19,18 +19,42 @@ def iloveck101(url)
     return
   end
 
-  page_content = page_uri.read
+  agent = Mechanize.new
 
-  title = page_content.match(%r{<title>(.+)</title>})[1]
-    .split(' - ')[0]
-    .gsub('/', '')
-    .strip
+  page = agent.get(page_uri)
+
+  if not page.search('#messagelogin').empty?
+    puts 'Need to log-in to access this page!'
+    print 'Username: '
+    username = STDIN.gets.chomp
+    print 'Password: '
+    password = STDIN.gets.chomp
+    password_hash = Digest::MD5.hexdigest(password)
+
+    login_result_page = agent.post(
+      'http://ck101.com/member.php?mod=logging&action=login&loginsubmit=yes&infloat=yes&lssubmit=yes&inajax=1',
+      {
+        :quickforawrd => 'yes',
+        :handlekey => 'ls',
+        :username => username,
+        :password => password_hash
+      })
+    if login_result_page.body.match(/succeedhandle_ls/).nil?
+      STDERR.print "Failed to log-in."
+      return
+    end
+
+    page = agent.get(page_uri)
+  end
+
+  title = page.title.split(' - ')[0].gsub('/', '').strip
+
+  puts "Parsing #{title}"
 
   image_url_set = Set.new(
-    page_content
-      .scan(%r{file="([^"]+)"})
-      .map { |e| e[0] }
-      .find_all { |e| e.index('.thumb.').nil?}
+    page.search('article img[file]')
+      .map { |image_tag| image_tag['file'] }
+      .find_all { |image_url| image_url.index('.thumb.').nil?}
   )
 
   target_folder_path = File.expand_path("~/Pictures/iloveck101/#{thread_id} - #{title}")
@@ -38,20 +62,22 @@ def iloveck101(url)
     FileUtils.mkdir_p(target_folder_path)
   end
 
+  puts "Found #{image_url_set.length} images."
+
   image_url_set.each { |image_url|
-    image_uri = page_uri.merge(image_url)
-    filename = File.expand_path(File.basename(image_uri.path), target_folder_path)
-    File.open(filename, 'wb') { |f|
-      f.write(open(image_uri).read)
-    }
+    puts "Downloading #{image_url}"
+
+    image = agent.get(image_url)
+
+    filename = File.expand_path(File.basename(image.uri.path), target_folder_path)
+    image.save(filename)
   }
 
+  puts 'Done!'
 end
 
-if __FILE__ == $0
-  if ARGV.length >= 1
-    iloveck101(ARGV[0])
-  else
-    STDERR.print "Usage: #{File.basename($0)} URL\n"
-  end
-end
+if ARGV.length >= 1
+  iloveck101(ARGV[0])
+else
+  STDERR.print "Usage: #{File.basename($0)} URL\n"
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: iloveck101
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.0
 platform: ruby
 authors:
 - WanCW
@@ -9,7 +9,21 @@ autorequire:
 bindir: bin
 cert_chain: []
 date: 2013-12-03 00:00:00.000000000 Z
-dependencies: []
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: mechanize
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.7.3
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.7.3
 description: Download images from ck101 thread, inspired by tzangms.
 email: wancw.wang@gmail.com
 executables: