birthdaze/lib/birthdaze.rb

require "carddav"
require "icalendar"
require "thor"
require "yaml"

class Birthdaze < Thor
  desc "generate", "Generate calendars"
  def generate
    puts "Writing ical file to #{config['ical_output']}"
    File.open(config["ical_output"], 'w') { |file| file.write(calendar.to_ical) }
  end

  desc "list", "List birthdays"
  def list
    display = birthdays.map do |name, birthday|
      {
        name: name,
        month: month_of(start_date(birthday)),
        day: day_of(start_date(birthday)),
      }
    end
    display.sort! { |a, b| a[:month] == b[:month] ? a[:day] <=> b[:day] : a[:month] <=> b[:month] }
    display.each { |d| puts "🎂 #{d[:month]}/#{d[:day]} - #{d[:name]}" }
  end

  private

  def config(config_file = "#{ENV["HOME"]}/.config/birthdaze.yaml")
    unless File.file?(config_file)
      puts "Please add a configuration file"
      return
    end

    @config ||= YAML.load_file(config_file)
  end

  def client(url: config["url"], username: config["username"], password: config["password"])
    @client ||= Carddav::Client.new(url, username, password)
  end

  def birthdays
    return @birthdays if defined? @birthdays

    @birthdays = []
    client.cards.each do |card|
      card = card.parsed.to_s
      birthday = birthday_regex.match(card)[1] if birthday_regex.match?(card)
      name = name_regex.match(card)[1] if name_regex.match?(card)
      @birthdays << [ name, birthday ] if name && birthday
    end
    @birthdays
  end

  def calendar
    return @calendar if defined? @calendar

    @calendar = Icalendar::Calendar.new
    birthdays.each do |name, birthday|
      @calendar.event do |event|
        puts name
        event.dtstart = Icalendar::Values::Date.new(start_date(birthday))
        event.dtend = Icalendar::Values::Date.new(end_date(birthday))
        event.summary = summary(name, birthday)
      end
    end
    @calendar.publish
  end

  # Takes a birthday string, with or without a year, and returns a start date
  def start_date(birthday)
    year = Date.today.year
    birthday = birthday.tr("-", "")
    birthday = birthday.gsub("1604", "") if birthday.start_with?("1604")
    if birthday.length < 8 # No year specified
      "#{year}#{birthday[0..3]}"
    else
      "#{year}#{birthday[4..7]}"
    end
  end

  def end_date(birthday)
    year = Date.today.year
    birthday = birthday.tr("-", "")
    birthday = birthday.gsub("1604", "") if birthday.start_with?("1604")
    if birthday.length < 8 # No year specified
      "#{year}#{birthday[0..1]}#{birthday[2..3].to_i + 1}"
    else
      "#{year}#{birthday[4..5]}#{birthday[6..7].to_i + 1}"
    end
  end

  def summary(name, birthday)
    return "🎂 #{name}’s birthday"
  end

  def set_reminders
  end

  def birthday_regex
    # We need the dash for dates which don’t specify a year
    /.*BDAY.*:([\d-]*).*/
  end

  def name_regex
    /FN.*:(.*)/
  end

  def month_of(date)
    date.length < 8 ? date[0..1] : date[4..5]
  end

  def day_of(date)
    date.length < 8 ? date[2..3] : date[6..7]
  end
end