Class: RbsActiverecord::RakeTask

Inherits:
Rake::TaskLib
  • Object
show all
Defined in:
lib/rbs_activerecord/rake_task.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(name = :"rbs:activerecord", &block) ⇒ RakeTask

Returns a new instance of RakeTask.



13
14
15
16
17
18
19
20
21
22
23
24
25
 
# File 'lib/rbs_activerecord/rake_task.rb', line 13

def initialize(name = :"rbs:activerecord", &block) #: void
  super()

  @name = name
  @pure_accessors = false
  @signature_root_dir = Rails.root / "sig/activerecord"

  block&.call(self)

  define_setup_task
  define_clean_task
  define_generate_task
end

Instance Attribute Details

#name ⇒ Object

: Symbol



7
8
9
 
# File 'lib/rbs_activerecord/rake_task.rb', line 7

def name
  @name
end

#pure_accessors ⇒ Object

: bool



8
9
10
 
# File 'lib/rbs_activerecord/rake_task.rb', line 8

def pure_accessors
  @pure_accessors
end

#signature_root_dir ⇒ Object

: Pathname



9
10
11
 
# File 'lib/rbs_activerecord/rake_task.rb', line 9

def signature_root_dir
  @signature_root_dir
end

Instance Method Details

#define_clean_task ⇒ Object

: void



34
35
36
37
38
39
 
# File 'lib/rbs_activerecord/rake_task.rb', line 34

def define_clean_task #: void
  desc "Clean RBS files for ActiveRecord models"
  task "#{name}:clean" do
    signature_root_dir.rmtree if signature_root_dir.exist?
  end
end

#define_generate_task ⇒ Object

: void # rubocop:disable Metrics/AbcSize



41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
 
# File 'lib/rbs_activerecord/rake_task.rb', line 41

def define_generate_task #: void  # rubocop:disable Metrics/AbcSize
  desc "Generate RBS files for ActiveRecord models"
  task("#{name}:generate": :environment) do
    require "rbs_activerecord" # load RbsActiverecord lazily

    Rails.application.eager_load!

    signature_root_dir.mkpath

    ActiveRecord::Base.descendants.each do |klass|
      next if klass.abstract_class?
      next unless klass.connection.table_exists?(klass.table_name)

      rbs = Generator.new(klass, pure_accessors:).generate

      path = signature_root_dir / "app/models/#{klass.name.underscore}.rbs"
      path.dirname.mkpath
      path.write(rbs)
    end
  end
end

#define_setup_task ⇒ Object

: void



27
28
29
30
31
32
 
# File 'lib/rbs_activerecord/rake_task.rb', line 27

def define_setup_task #: void
  desc "Run all tasks of rbs_activerecord"

  deps = [:"#{name}:clean", :"#{name}:generate"]
  task("#{name}:setup" => deps)
end