mirror of
https://gh.wpcy.net/https://github.com/discourse/discourse.git
synced 2026-05-09 06:34:52 +08:00
44 lines
1.1 KiB
Ruby
44 lines
1.1 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require "rrule"
|
|
|
|
class RRuleGenerator
|
|
def self.generate(
|
|
starts_at:,
|
|
timezone: "UTC",
|
|
max_years: nil,
|
|
recurrence: "every_week",
|
|
recurrence_until: nil
|
|
)
|
|
rrule = generate_hash(RRuleConfigurator.rule(recurrence_until:, recurrence:, starts_at:))
|
|
rrule = set_mandatory_options(rrule, starts_at)
|
|
|
|
::RRule::Rule
|
|
.new(stringify(rrule), dtstart: starts_at, tzid: timezone)
|
|
.between(Time.current, Time.current + 14.months)
|
|
.first(RRuleConfigurator.how_many_recurring_events(recurrence:, max_years:))
|
|
end
|
|
|
|
private
|
|
|
|
def self.stringify(rrule)
|
|
rrule.map { |k, v| "#{k}=#{v}" }.join(";")
|
|
end
|
|
|
|
def self.generate_hash(rrule)
|
|
rrule
|
|
.split(";")
|
|
.each_with_object({}) do |rr, h|
|
|
key, value = rr.split("=")
|
|
h[key] = value
|
|
end
|
|
end
|
|
|
|
def self.set_mandatory_options(rrule, time)
|
|
rrule["BYHOUR"] = time.strftime("%H")
|
|
rrule["BYMINUTE"] = time.strftime("%M")
|
|
rrule["INTERVAL"] ||= 1
|
|
rrule["WKST"] = "MO" # considers Monday as the first day of the week
|
|
rrule
|
|
end
|
|
end
|
