Extract highlights and notes which are exported from Good Reader app

good_reader_annotations.rb

#/usr/bin/env ruby

raise "Set input file. $ ruby #{__FILE__} input.txt" if ARGV.empty?

PAGE_SEPARATOR = %r{--- Page (\S+) ---}
ANNOTATION_SEPARATOR = %r{(?<type>Highlight|Note) \((?<color>[^)]+)\), (?<time>.*)(?:, (?<owner>.*))?:}
lines = File.readlines(ARGV[0])

annotations = []
lines.map(&:chomp)
     .reject{|l| l == "" }
     .slice_before(PAGE_SEPARATOR)
     .select{|page| page.any?{|line| line =~ PAGE_SEPARATOR} }
     .each do |page|
  page_no = page.shift.match(PAGE_SEPARATOR)[1]

  page.slice_before(ANNOTATION_SEPARATOR).each do |annotation|
    info = annotation.shift.match(ANNOTATION_SEPARATOR)
    # text = annotation.map{|h| h.gsub(/[[:space:]]/, '') }.join(' ') # For Japanese
    text = annotation.join(' ') # For English
    annotations << {
      type: info[:type],
      page_no: page_no,
      color: info[:color],
      time: info[:time],
      owner: info[:owner],
      text: text,
    }
  end
end


### Output
# Change the following part as you like

## --- output as hash
# annotations.each do |a|
#   puts a.inspect
# end

## --- group by type
# annotations.group_by{|a| a[:type] }.each do |type, group|
#   puts "[#{type}]"
#   group.each do |a|
#     puts "#{a[:text]} (p#{a[:page_no]})"
#   end
#   puts
# end

## --- group by color
annotations.group_by{|a| a[:type] }.each do |type, group|
  puts "[#{type}]"
  group.group_by{|a| a[:color] }.each do |color, subgroup|
    puts "--- #{color} ---"
    subgroup.each do |a|
      puts "#{a[:text]} (p#{a[:page_no]})"
    end
    puts
  end
  puts
end

## --- using rainbow
# require 'rainbow'
# annotations.each do |a|
#   puts Rainbow(a[:text]).background(a[:color].to_sym) + "(p#{a[:page_no]})"
# end



## sample input.txt for testing
=begin
File: refactoring-ja-special-edition_p1_0.pdf

Annotation summary:

--- Page xi ---

Highlight (yellow), 2015/03/05 9:10:
2000 年に発行された『リファクタリング プログラミングの体質改善テクニック』


--- Page xix ---

Highlight (yellow), 2015/03/05 9:10:
リファクタリングの父は 2 人います。Ward Cunningham と Kent Beck です。

Highlight (yellow), 2015/03/05 9:10:
John Brant と Don Roberts は単に論文を書くのに止まらず、ツールの作成まで行いました。それが 「Refactoring Browser」 、すなわちリファクタリングを行うための Smalltalk のブラウザです。


--- Page 12 ---

Highlight (yellow), 2015/03/05 9:10:
変更がほんの少しであれば、それによって生じるエラーを見つけるのは簡単 なことです。


--- Page 66 ---

Highlight (red), 2015/03/06 14:56:
決してリファクタリングをしてはいけない場合もあります。第 1 の例は、変更するよりも最 初からの書き直した方が早いという場合です。

Highlight (yellow), 2015/03/06 14:56:
リファクタリングを避けるべき第 2 の例として、 期間が迫っている場合があります。 こうした状況では、リファクタリングをしても生産性の向上が見られるのは締め切り後であり、 時すでに遅しということになってしまいます。

Highlight (blue), 2015/03/06 14:56:
時間が足りなくなるというのは、たいて いの場合、リファクタリングが必要であることを示唆しているのです。

Note (yellow), 2015/03/06 14:56:
あいうえお
=end

This worked like a charm! Thank you!