slack-pixiv-bot/slack-pixiv/commands/illust.rb

# coding: utf-8
module SlackMathbot
  module Commands
    class Illust < SlackRubyBot::Commands::Base
      match /member_illust\.php(?<url>.*)$/ do |client, _data, _match|

        # Initalize Mechanize
        agent = Mechanize.new

        # Create Pixiv URL
        # If a full URL is given, extra characters are appended for some reason.
        # This ensures we catch them better
        if (_data[:text].include? "pixiv.net")
          pixiv_url = "http://www.pixiv.net/member_illust.php" + _match[:url][0..-2]
        else
          pixiv_url = "http://www.pixiv.net/member_illust.php" + _match[:url][0..-1]
        end
        pixiv_url = CGI.unescapeHTML(pixiv_url)
        puts pixiv_url

        # Create iOS Illustration URL, regex pixiv_url to
        # extract numbers only, such as /^[0-9]+$/
        #ios_url = "pixiv://illusts/" + pixiv_url.split("illust_id=")[-1]
        ios_url = "pixiv://illusts/" + pixiv_url[/\d+/]
        puts ios_url

        # Scrape profile link
        profile = agent.get(pixiv_url).at("div.userdata-row > h2.name > a").attributes['href'].to_s
        puts profile

        # Create iOS Members URL
        ios_mem_url = "pixiv://users/" + profile[/\d+/]
        puts ios_mem_url

        # Scrape page title
        title = agent.get(pixiv_url).title
        puts title

        # Scrape image
        image_url = agent.get(pixiv_url).images_with(:src => /600x600\/img-master/)[0].to_s.sub! '600x600','480x960'
        begin
          # Attempt to check on image
          page = agent.head(image_url)

          # Make sure page came back kosher
          if page.code.to_i == 200
            # Image url is fine
            puts "Image is good to go!"
          else
            # Fallback on http error to be safe
            image_url = "https://embed.pixiv.net/decorate.php?illust_id=" + pixiv_url[/\d+/]
          end
        rescue Mechanize::ResponseCodeError
          # Fallback to og:image url
          puts "Image failed to load"
          image_url = "https://embed.pixiv.net/decorate.php?illust_id=" + pixiv_url[/\d+/]
        end
        puts image_url

        client.web_client.chat_postMessage(
          channel: _data.channel,
          as_user: true,
          attachments: [
            {
              fallback: title + " - " + pixiv_url,
              text: "<" + ios_url + "|View Image> | <" + ios_mem_url + "|Open Profile>",
              title: title,
              title_link: pixiv_url,
              image_url: image_url,
              color: "#2684BD"
            }
          ].to_json
        )

      end
    end
  end
end