Class: Spring::Client::Binstub

Inherits:
Command
  • Object
show all
Defined in:
lib/spring/client/binstub.rb

Defined Under Namespace

Classes: Item

Constant Summary collapse

SHEBANG = 
/\#\!.*\n/
LOADER =

If loading the bin/spring file works, it’ll run spring which will eventually call Kernel.exit. This means that in the client process we will never execute the lines after this block. But if the spring client is not invoked for whatever reason, then the Kernel.exit won’t happen, and so we’ll fall back to the lines after this block, which should cause the “unsprung” version of the command to run.

<<CODE
begin
  load File.expand_path("../spring", __FILE__)
rescue LoadError
end
CODE
SPRING =

The defined? check ensures these lines don’t execute when we load the binstub from the application process. Which means that in the application process we’ll execute the lines which come after the LOADER block, which is what we want.

Parsing the lockfile in this way is pretty nasty but reliable enough The regex ensures that the match must be between a GEM line and an empty line, so it won’t go on to the next section.

<<'CODE'
#!/usr/bin/env ruby

# This file loads spring without using Bundler, in order to be fast.
# It gets overwritten when you run the `spring binstub` command.

unless defined?(Spring)
  require "rubygems"
  require "bundler"

  if match = Bundler.default_lockfile.read.match(/^GEM$.*?^    (?:  )*spring \((.*?)\)$.*?^$/m)
    Gem.paths = { "GEM_PATH" => [Bundler.bundle_path.to_s, *Gem.path].uniq }
    gem "spring", match[1]
    require "spring/binstub"
  end
end
CODE
OLD_BINSTUB = 
%{if !Process.respond_to?(:fork) || Gem::Specification.find_all_by_name("spring").empty?}

Instance Attribute Summary collapse

Attributes inherited from Command

#args, #env

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(args) ⇒ Binstub

Returns a new instance of Binstub.



132
133
134
135
136
137
138
139
140
141
142
 
# File 'lib/spring/client/binstub.rb', line 132

def initialize(args)
  super

  @bindir = env.root.join("bin")
  @all    = false
  @mode   = :add
  @items  = args.drop(1)
                .map { |name| find_commands name }
                .inject(Set.new, :|)
                .map { |command| Item.new(command) }
end

Instance Attribute Details

#bindir ⇒ Object (readonly)

Returns the value of attribute bindir.



117
118
119
 
# File 'lib/spring/client/binstub.rb', line 117

def bindir
  @bindir
end

#items ⇒ Object (readonly)

Returns the value of attribute items.



117
118
119
 
# File 'lib/spring/client/binstub.rb', line 117

def items
  @items
end

Class Method Details

.call(args) ⇒ Object 



127
128
129
130
 
# File 'lib/spring/client/binstub.rb', line 127

def self.call(args)
  require "spring/commands"
  super
end

.description ⇒ Object 



119
120
121
 
# File 'lib/spring/client/binstub.rb', line 119

def self.description
  "Generate spring based binstubs. Use --all to generate a binstub for all known commands."
end

.rails_command ⇒ Object 



123
124
125
 
# File 'lib/spring/client/binstub.rb', line 123

def self.rails_command
  @rails_command ||= CommandWrapper.new("rails")
end

Instance Method Details

#call ⇒ Object 



166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
 
# File 'lib/spring/client/binstub.rb', line 166

def call
  case @mode
  when :add
    bindir.mkdir unless bindir.exist?

    File.write(spring_binstub, SPRING)
    spring_binstub.chmod 0755

    items.each(&:add)
  when :remove
    spring_binstub.delete if @all
    items.each(&:remove)
  else
    raise ArgumentError
  end
end

#find_commands(name) ⇒ Object 



144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
 
# File 'lib/spring/client/binstub.rb', line 144

def find_commands(name)
  case name
  when "--all"
    @all = true
    commands = Spring.commands.dup
    commands.delete_if { |command_name, _| command_name.start_with?("rails_") }
    commands.values + [self.class.rails_command]
  when "--remove"
    @mode = :remove
    []
  when "rails"
    [self.class.rails_command]
  else
    if command = Spring.commands[name]
      [command]
    else
      $stderr.puts "The '#{name}' command is not known to spring."
      exit 1
    end
  end
end

#spring_binstub ⇒ Object 



183
184
185
 
# File 'lib/spring/client/binstub.rb', line 183

def spring_binstub
  bindir.join("spring")
end