hundredwatt/literal_enum.rb

## 46 changes: 46 additions & 0 deletions literal_enum.rb
@@ -0,0 +1,46 @@

    # frozen_string_literal: true
# frozen_string_literal: true


    require "bundler/inline"
require "bundler/inline"


    gemfile(true) do
gemfile(true) do

      source "https://rubygems.org"
  source "https://rubygems.org"


      gem "rails"
  gem "rails"

      # If you want to test against edge Rails replace the previous line with this:
  # If you want to test against edge Rails replace the previous line with this:

      # gem "rails", github: "rails/rails", branch: "main"
  # gem "rails", github: "rails/rails", branch: "main"


      gem "sqlite3"
  gem "sqlite3"

    end
end


    require "active_record"
require "active_record"

    require "minitest/autorun"
require "minitest/autorun"

    require "logger"
require "logger"


    # This connection will do for database-independent bug reports.
# This connection will do for database-independent bug reports.

    ActiveRecord::Base.establish_connection(adapter: "sqlite3", database: ":memory:")
ActiveRecord::Base.establish_connection(adapter: "sqlite3", database: ":memory:")

    ActiveRecord::Base.logger = Logger.new(STDOUT)
ActiveRecord::Base.logger = Logger.new(STDOUT)


    ActiveRecord::Schema.define do
ActiveRecord::Schema.define do

      create_table :posts, force: true do |t|
  create_table :posts, force: true do |t|

        t.string :status
    t.string :status

      end
  end

    end
end


    class ApplicationRecord < ActiveRecord::Base
class ApplicationRecord < ActiveRecord::Base

      primary_abstract_class
  primary_abstract_class


      def self.literal_enum(name = nil, values = nil, **options)
  def self.literal_enum(name = nil, values = nil, **options)

        if values.is_a?(Array)
    if values.is_a?(Array)

          values = values.map { |value| [value] * 2 }.to_h
      values = values.map { |value| [value] * 2 }.to_h

        end
    end


        enum(name, values, **options)
    enum(name, values, **options)

      end
  end

    end
end


    class Post < ApplicationRecord
class Post < ApplicationRecord

      literal_enum :status, ['pending', 'accepted']
  literal_enum :status, ['pending', 'accepted']

    end
end


    post = Post.create! status: 'accepted'
post = Post.create! status: 'accepted'

    p post.status_before_type_cast # => 'accepted'
p post.status_before_type_cast # => 'accepted'