extract4.rb

#!/usr/bin/env ruby
require 'json'
require 'htmlentities'

# Calculate person who cites the most number of different XKCDs and print output
# to stdout

# Load xkcd.jsonl
imgs = {}
File.foreach("xkcd.jsonl") do |line|
  data = JSON.parse(line)
  imgs[data["img"]] = data["id"]
end

stats = {}
# Load raw.jsonl
File.foreach("raw.jsonl") do |line|
  data = JSON.parse(line)
  if !stats[data["by"]] then
    stats[data["by"]] = {}
  end
  text = HTMLEntities.new.decode(data["text"])
  h = {}

  # Grab xkcd.com/<id>, www.xkcd.com/<id>
  # and m.xkcd.com/<id>.
  m = text.scan(/(?:www\.|m\.)?xkcd\.com\/([0-9]+)/i).flatten
  m.each do |id|
    id = id.to_i
    h[id]=true
  end

  # Grab various forms of "xkcd #1234"
  m = text.scan(/xkcd [^a-z0-9]*([0-9]+)/i).flatten
  m.each do |id|
    id = id.to_i
    h[id]=true
  end

  # Grab imgs.xkcd.com/<img> and convert to comic id
  m = text.scan(/(imgs\.xkcd\.com\/.*?(?:png|gif|jpg))/i).flatten
  m.each{|img|
    i = "https://" + img
    if imgs[i] then
      h[imgs[i]] = true
    else
      #puts "unknown: " + i
    end
  }

  h.each do |key, value|
    stats[data["by"]][key] = true
  end
end

# Sort and emit top 10
stats.each do |key, value|
  stats[key] = value.length
end
puts stats.sort_by {|_key, value| value}.last(10).reverse().to_h