class A0::TZMigration::DataGenerator

Attributes

out[R]
path[R]
timezones[R]
url[R]
versions[R]

Public Class Methods

new(out, url: 'https://github.com/tzinfo/tzinfo-data.git') click to toggle source
# File lib/a0/tzmigration/data_generator.rb, line 12
def initialize(out, url: 'https://github.com/tzinfo/tzinfo-data.git')
  @out = out
  @url = url
  @path = File.join(@out, 'repo', 'tzinfo-data')

  FileUtils.mkdir_p @out

  repo
end

Public Instance Methods

data_from_transition(transition) click to toggle source
# File lib/a0/tzmigration/data_generator/process.rb, line 95
def data_from_transition(transition)
  offset = transition.offset
  previous_offset = transition.previous_offset

  { utc_offset: offset.utc_total_offset,
    utc_prev_offset: previous_offset.utc_total_offset,
    utc_timestamp: transition.time.to_i,
    utc_time: transition.datetime,
    local_ini_str: transition.local_end_time.strftime("%F %T #{previous_offset.abbreviation}"),
    local_fin_str: transition.local_start_time.strftime("%F %T #{offset.abbreviation}") }
end
generate() click to toggle source
# File lib/a0/tzmigration/data_generator/process.rb, line 10
def generate
  load_release_dates

  generate_versions
  generate_timezones

  save_versions
  save_versions_index
  save_timezones
  save_timezones_index
  save_index
end
generate_timezones() click to toggle source
# File lib/a0/tzmigration/data_generator/process.rb, line 34
def generate_timezones
  @timezones = {}
  @versions.each do |version_data|
    version_data[:timezones].each do |name, timezone_data|
      @timezones[name] ||= { name: name, versions: {} }
      @timezones[name][:versions][version_data[:version]] = { tag: version_data[:tag], released_at: version_data[:released_at] }.merge timezone_data
    end
  end

  @timezones = @timezones.to_a.sort_by(&:first).to_h
end
generate_versions() click to toggle source
# File lib/a0/tzmigration/data_generator/process.rb, line 23
def generate_versions
  tags = repo.tags.map(&:name)
  n_proc = Etc.nprocessors || 2

  @versions = Parallel.map(tags, in_processes: n_proc, progress: 'Processing') do |tag|
    process_tag tag
  end

  @versions.sort_by! { |version_data| version_data[:version] }
end
load_release_dates() click to toggle source
# File lib/a0/tzmigration/data_generator/process.rb, line 46
def load_release_dates
  @released_at = {}

  File.open(File.join(@path, 'data', 'NEWS')).each_line do |line|
    next unless line.start_with? 'Release '

    release, time = line.split(' - ')
    version = release.split.last
    next unless time

    @released_at[version] = Time.parse(time)
  end
end
process(tag) click to toggle source
# File lib/a0/tzmigration/data_generator/process.rb, line 69
def process(tag)
  version = TZInfo::Data::Version::TZDATA
  timezones = {}
  meta = { tag: tag, version: version, released_at: @released_at[version], timezones: timezones }

  TZInfo::Timezone.all.each do |timezone|
    process_timezone timezones, timezone.send(:real_timezone)
  end

  meta
end
process_tag(tag) click to toggle source
# File lib/a0/tzmigration/data_generator/process.rb, line 60
def process_tag(tag)
  repo_use tag

  tzinfo_unload
  tzinfo_load

  process tag
end
process_timezone(timezones, real_timezone) click to toggle source
# File lib/a0/tzmigration/data_generator/process.rb, line 81
def process_timezone(timezones, real_timezone)
  name = real_timezone.name
  canonical_name = real_timezone.canonical_identifier

  if name == canonical_name
    transitions = real_timezone.send(:info).instance_variable_get(:@transitions)&.map do |transition|
      data_from_transition(transition)
    end
    timezones[name] = { transitions: transitions }
  else
    timezones[name] = { alias: canonical_name }
  end
end
repo() click to toggle source
# File lib/a0/tzmigration/data_generator/git.rb, line 8
def repo
  return @repo if defined? @repo

  File.exist?(@path) ? repo_update : repo_clone
end
repo_clone() click to toggle source
# File lib/a0/tzmigration/data_generator/git.rb, line 23
def repo_clone
  puts "Cloning #{@url} repo at #{@path}"

  @repo = Git.clone @url, @path
end
repo_update() click to toggle source
# File lib/a0/tzmigration/data_generator/git.rb, line 14
def repo_update
  puts "Updating #{@url} repo at #{@path}"

  @repo = Git.open @path
  @repo.pull

  @repo
end
repo_use(tag) click to toggle source
# File lib/a0/tzmigration/data_generator/git.rb, line 29
def repo_use(tag)
  tag_path = File.join(@out, 'repo', tag)
  lib_path = File.join(tag_path, 'lib')
  $LOAD_PATH.unshift File.expand_path(lib_path)

  return if File.exist? lib_path

  FileUtils.rm_rf tag_path
  FileUtils.copy_entry @path, tag_path

  tag_repo = Git.open tag_path
  tag_repo.checkout tag
end
save_index() click to toggle source
# File lib/a0/tzmigration/data_generator/save.rb, line 56
def save_index
  File.open(File.join(@out, 'index.md'), 'w') do |file|
    file.write version_head
    @timezones.keys.sort.map do |timezone_name|
      file.write timezones_row(timezone_name)
    end
  end
end
save_timezones() click to toggle source
# File lib/a0/tzmigration/data_generator/save.rb, line 18
def save_timezones
  FileUtils.mkdir_p File.join(@out, 'timezones')

  @timezones.each do |name, timezone_data|
    dir = name.split('/')[0...-1]
    FileUtils.mkdir_p File.join(@out, 'timezones', dir) unless dir.empty?

    File.open(File.join(@out, 'timezones', "#{name}.json"), 'w') do |file|
      file.write(JSON.generate(timezone_data))
    end
  end
end
save_timezones_index() click to toggle source
# File lib/a0/tzmigration/data_generator/save.rb, line 43
def save_timezones_index
  FileUtils.mkdir_p File.join(@out, 'timezones')

  File.open(File.join(@out, 'timezones', '00-index.json'), 'w') do |file|
    object = @timezones.map do |name, timezone_data|
      versions = timezone_data[:versions].keys
      [name, { versions: versions }]
    end.to_h

    file.write(JSON.generate(timezones: object))
  end
end
save_versions() click to toggle source
# File lib/a0/tzmigration/data_generator/save.rb, line 8
def save_versions
  FileUtils.mkdir_p File.join(@out, 'versions')

  @versions.each do |version_data|
    File.open(File.join(@out, 'versions', "#{version_data[:version]}.json"), 'w') do |file|
      file.write(JSON.generate(version_data))
    end
  end
end
save_versions_index() click to toggle source
# File lib/a0/tzmigration/data_generator/save.rb, line 31
def save_versions_index
  FileUtils.mkdir_p File.join(@out, 'versions')

  File.open(File.join(@out, 'versions', '00-index.json'), 'w') do |file|
    versions = @versions.map do |version_data|
      [version_data[:version], { released_at: version_data[:released_at], timezones: version_data[:timezones].keys }]
    end.to_h

    file.write(JSON.generate(versions: versions))
  end
end
timezones_row(timezone_name) click to toggle source
# File lib/a0/tzmigration/data_generator/save.rb, line 75
def timezones_row(timezone_name)
  row = @versions.map do |version_data|
    data = @timezones[timezone_name][:versions][version_data[:version]]
    if data && data[:alias]
      "→ #{data[:alias]}"
    elsif data
      "#{data[:transitions].count} "
    end
  end.join ' | '

  "| [#{timezone_name}](timezones/#{timezone_name}.json) | #{row} |\n"
end
tzinfo_load() click to toggle source
# File lib/a0/tzmigration/data_generator.rb, line 22
def tzinfo_load
  old_verbose = $VERBOSE
  $VERBOSE = nil

  require 'tzinfo'
  require 'tzinfo/data/indexes/countries'
  require 'tzinfo/data/indexes/timezones'
  require 'tzinfo/data'

  $VERBOSE = old_verbose
end
tzinfo_unload() click to toggle source
# File lib/a0/tzmigration/data_generator.rb, line 34
def tzinfo_unload
  $LOADED_FEATURES.delete_if { |x| x.include?('tzinfo') }
  Object.send(:remove_const, :TZInfo) if defined? TZInfo
end
version_head() click to toggle source
# File lib/a0/tzmigration/data_generator/save.rb, line 65
def version_head
  head_a = @versions.map do |version_data|
    version = version_data[:version]
    "[#{version}](versions/#{version}.json)"
  end.join ' | '
  head_b = @versions.map { '---' }.join ' | '

  "| | #{head_a} |\n| --- | #{head_b} |\n"
end