abuse-the-force/lib/abusetheforce.rb

305 lines
9.8 KiB
Ruby
Raw Normal View History

2013-04-18 16:09:05 +00:00
require "metaforce"
require "base64"
module AbuseTheForce
attr_accessor :client
# Write error to screen
def self.pute(s, fatal=false)
puts "Error: #{s}"
# If fatal error, exit
if fatal
exit 1
end
end
# Write warning to screen
def self.putw(s)
puts "Warning: #{s}"
end
# builds the client instance of Metaforce
def self.build_client
target = Atf_Config.active_target
puts target.username
2013-04-18 16:09:05 +00:00
@client = Metaforce.new :username => target.username,
:password => target.get_password,
:security_token => target.security_token
Metaforce.configuration.host = target.host
# Supress some of the verbose logging
Metaforce.configuration.log = false
2013-04-18 16:09:05 +00:00
end
# Fetches a single file from the server
def self.retrieve_file(metadata_type, full_name)
if @client == nil
build_client
end
# Backup old Manifest
FileUtils.copy(
File.join(Atf_Config.get_project_path, 'package.xml'),
File.join(Atf_Config.get_project_path, 'package.xml-bak')
)
2013-04-18 16:09:05 +00:00
manifest = Metaforce::Manifest.new(metadata_type => [full_name])
@client.retrieve_unpackaged(manifest).
extract_to(Atf_Config.get_project_path).
2013-04-18 16:09:05 +00:00
on_complete { |job| puts "Finished retrieve #{job.id}!" }.
on_error { |job| puts "Something bad happened!" }.
on_poll { |job| puts "Polling for #{job.id}!" }.
2013-04-18 16:09:05 +00:00
perform
# Restore old Manifest
FileUtils.move(
File.join(Atf_Config.get_project_path, 'package.xml-bak'),
File.join(Atf_Config.get_project_path, 'package.xml')
)
2013-04-18 16:09:05 +00:00
end
# Fetches a whole project from the server
def self.retrieve_project()
if @client == nil
build_client
end
if File.file? File.join(Atf_Config.get_project_path, 'package.xml')
2013-04-18 16:09:05 +00:00
@client.retrieve_unpackaged(File.expand_path(Atf_Config.src + '/package.xml')).
extract_to(Atf_Config.get_project_path).
2013-04-18 16:09:05 +00:00
on_complete { |job| puts "Finished retrieve #{job.id}!" }.
on_error { |job| puts "Something bad happened!" }.
on_poll { |job| puts "Polling for #{job.id}!" }.
2013-04-18 16:09:05 +00:00
perform
else
puts "#{Atf_Config.get_project_path}: Not a valid project path"
2013-04-18 16:09:05 +00:00
end
end
def self.deploy_project(dpath=Atf_Config.get_project_path)
2013-04-18 16:09:05 +00:00
if @client == nil
build_client
end
if File.file? File.join(dpath, 'package.xml')
2013-04-18 16:09:05 +00:00
@client.deploy(File.expand_path(dpath)).
on_complete { |job|
puts "Finished deploy #{job.id}!"
result = job.result
if result != nil
puts "\nDeploy #{result.success ? "SUCCESS" : "FAILURE"}"
# If a failed deploy, print errors
if result.success == false
# Need messages in an array
unless result.messages.kind_of? Array
result.messages = [].push result.messages
end
puts "ERRORS: #{result.messages.size}"
result.messages.each do |m|
# If the path is not from the project, fix it
unless m.file_name.starts_with? Atf_Config.src
m.file_name = m.file_name.sub(/[a-zA-Z._-]*\//, Atf_Config.src + '/')
end
# Print our error in teh format "filename:line:column type in object message"
puts "#{m.file_name}:#{m.line_number}:#{m.column_number} #{m.problem_type} in #{m.full_name} #{m.problem}"
end
end
end
}.
2013-04-18 16:09:05 +00:00
on_error { |job| puts "Something bad happened!" }.
on_poll { |job| puts "Polling for #{job.id}!" }.
2013-04-18 16:09:05 +00:00
perform
else
puts "#{dpath}: Not a valid project path"
end
end
class Atf_Config
class << self
attr_accessor :targets, :active_target, :src, :root_dir
SETTINGS_FILE=".abusetheforce.yaml"
def locate_root(path = '.')
temp_path = path
# Look for a settings file in this path and up to root
until File.file? File.join(temp_path, SETTINGS_FILE)
# If we hit root, stop
if temp_path == '/'
break
end
# Didn't find one so go up one level
temp_path = File.absolute_path File.dirname(temp_path)
end
# If we actually found it
if File.file? File.join(temp_path, SETTINGS_FILE)
# Return
return temp_path
else
# Return the original path
return path
end
end
2013-04-18 16:09:05 +00:00
# Loads configurations from yaml
def load()
# Get the project root directory
@root_dir = locate_root
if File.file? File.join(@root_dir, SETTINGS_FILE)
settings = YAML.load_file File.join(@root_dir, SETTINGS_FILE)
@targets = settings[:targets]
@src = settings[:src]
else
2013-04-18 16:09:05 +00:00
puts "No settings file found, creating one now"
# Settings file doesn't exist
# Create it
@targets = {}
@active_target = nil
@src = './src'
dump_settings
end
# Set the default target
@targets.values.each do |target|
# Check if this one is active
if target.active == true
# Set it if there is no default target set yet
if @active_target == nil
@active_target = target
else
puts "Two active targets set. Using #{@active_target.print}"
end
end
end
end
# write settings to a yaml file
def dump_settings
File.open(SETTINGS_FILE, 'w') do |out|
YAML.dump( { :targets => @targets, :src => @src }, out)
end
end
# Adds a new target to the config
def add_target(target)
# If there are no targets yet, use this one as the default
if @active_target == nil #@targets.empty?
target.active = true
@active_target = target
end
# Push the new target
@targets[target.name] = target
#write out the config
dump_settings
puts "Target Added"
target.print
2013-04-18 16:09:05 +00:00
end
# Selects one target as the active target for deployment
def set_active_target(name)
# Empty out current active target
@active_target = nil
# Go through each pair
@targets.each_pair do |target_name, target|
# If you find a matching item
if name == target_name
# Check if there is already a default
if @active_target == nil
# Set active
target.active = true
# Make it default
@active_target = target
end
else # name != name
# make not active
target.active = false
end
end
if @active_target != nil
# Save to yaml
dump_settings
# Notify the user
puts "Active target changed"
@active_target.print
2013-04-18 16:09:05 +00:00
else
AbuseTheForce.pute "Target with alias #{name} was not found."
end
2013-04-18 16:09:05 +00:00
end
# Sets project path from default ./src
def set_project_path(ppath)
if File.file? File.join(ppath, 'package.xml')
2013-04-18 16:09:05 +00:00
@src = ppath
dump_settings
else
pute("No package.xml found in #{ppath}", true)
end
end
# Gets the full path to the src folder
def get_project_path
return File.absolute_path File.join(root_dir, src)
end
2013-04-18 16:09:05 +00:00
end
end
# Class for holding a target
class Atf_Target
attr_accessor :name, :username, :password, :security_token, :host, :active
def initialize(name, username, password, security_token, host="login.salesforce.com")
@name = name
@username = username
set_password(password)
@security_token = security_token
@host = host
@active = false;
end
# TODO: Provide 2 way encryption with a lock to decode passwords
def set_password(password)
@password = Base64.encode64(password)
end
def get_password()
return Base64.decode64(@password)
end
def print
puts "#{@name}\t#{@username}\t#{@host}\t#{(@active && 'Active') || ''}"
end
end
end