123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116 |
- # Copyright (C) 2012-2022 Zammad Foundation, https://zammad-foundation.org/
- module CanBePublished
- class StateMachine
- include AASM
- delegate :current_state, to: :aasm
- def initialize(record)
- @record = record
- aasm.current_state = calculated_state
- end
- def calculated_state
- matching_time = Time.zone.now
- state = %i[archived published internal].find { |state_name| calculated_state_valid?(state_name, matching_time) }
- state || :draft
- end
- def calculated_state_valid?(state_name, time)
- date = @record.send "#{state_name}_at"
- date.present? && date < time
- end
- def set_timestamp
- override_unarchived_state if aasm.to_state == :unarchived
- @record.send "#{aasm.to_state}_at=", Time.zone.now.change(sec: 0)
- end
- def override_unarchived_state
- aasm.to_state = @record.published_at.present? ? :published : :internal
- end
- def update_using_current_user(user)
- %i[archived internal published].each { |state_name| update_state_using_current_user(user, state_name) }
- end
- def update_state_using_current_user(user, state_name)
- return if !@record.send("#{state_name}_at_changed?")
- new_value = @record.send("#{state_name}_at").present? ? user : nil
- @record.send("#{state_name}_by=", new_value)
- end
- def clear_archived
- @record.archived_at = nil
- end
- def save_record
- @record.save!
- end
- aasm do
- state :draft, initial: true
- state :internal
- state :published
- state :archived
- state :unarchived # magic
- event :internal do
- transitions from: :draft,
- to: :internal,
- guard: :guard_internal?,
- after: :set_timestamp
- end
- event :publish do
- transitions from: %i[draft internal],
- to: :published,
- guard: :guard_publish?,
- after: :set_timestamp
- end
- event :archive do
- transitions from: %i[published internal],
- to: :archived,
- guard: :guard_archive?,
- after: :set_timestamp
- end
- event :unarchive do
- transitions from: :archived,
- to: :unarchived,
- guard: :guard_unarchive?,
- after: %i[clear_archived set_timestamp]
- end
- after_all_events %i[update_using_current_user save_record mark_as_idle]
- error_on_all_events :mark_as_idle
- end
- def mark_as_idle
- aasm.send(:current_event=, nil) # nullify current_event after transitioning
- end
- def guard_internal?
- draft?
- end
- def guard_publish?
- draft? || internal?
- end
- def guard_archive?
- internal? || published?
- end
- def guard_unarchive?
- archived?
- end
- end
- end
|