Unmagic::Enum

gem 'unmagic-enum'

class Status < Unmagic::Enum
  ACTIVE = new("active")
  PENDING = new("pending")
  ARCHIVED = new("archived")
end

# Usage
status = Status::ACTIVE
status.active?              # => true
status == "active"          # => true
status.to_s                 # => "active"
Status["active"]            # => Status::ACTIVE

class Priority < Unmagic::Enum
  attribute :label
  attribute :color
  attribute :level, default: 0

  HIGH = new("high", label: "High Priority", color: "red", level: 3)
  MEDIUM = new("medium", label: "Medium Priority", color: "yellow", level: 2)
  LOW = new("low", label: "Low Priority", color: "green", level: 1)
end

priority = Priority::HIGH
priority.label              # => "High Priority"
priority.color              # => "red"
priority.level              # => 3

class Message < ApplicationRecord
  class State < Unmagic::Enum
    DRAFT = new("draft")
    SENT = new("sent")
    DELIVERED = new("delivered")
  end

  # Use the column type for proper serialization
  attribute :state, State.column_type
  
  # Create scopes
  scope :delivered, -> { where(state: State::DELIVERED) }
end

# Usage
message = Message.new(state: "sent")
message.state               # => Message::State::SENT
message.state.sent?         # => true

class MessageType < Unmagic::Enum
  USER = new("user")                    # key and value both "user"
  ENTITY = new("entity", value: "bot")  # key: "entity", value: "bot" (legacy DB)
  SYSTEM = new("system", value: "s")    # key: "system", value: "s" (short code)
end

MessageType::ENTITY.key     # => "entity"
MessageType::ENTITY.value   # => "bot"
MessageType["bot"]          # => MessageType::ENTITY

class User < ApplicationRecord
  class Type < Unmagic::Enum
    # Pass the actual class - no constantize needed!
    CUSTOMER = new(Customer)
    ADMIN = new(Admin)
    MODERATOR = new(Moderator)
  end

  attribute :type, Type.column_type

  def self.find_sti_class(type_name)
    if enum_value = Type[type_name]
      enum_value.key  # Returns the class directly
    else
      super
    end
  end
end