#!/usr/local/bin/ruby

# MySpace friend adder script.
#
# - Searches Google for a term you give it.
# - Rips friend ID's to a SQLite database in memory.
# - Adds those friends to your MySpace.
#
# !!! THIS SCRIPT IS FREE TO DISTRIBUTE, JUST KEEP THESE COMMENTS INTACT !!!
#
# Work by Subimage.com
# http://sublog.subimage.com/articles/2007/07/11/ruby-myspace-friend-adder-script
#
# REQUIRED GEMS:
#
# - activerecord
# - hpricot
# - mechanize
# - sqlite3-ruby
#

# GLOBALS =====================================================================

# STUFF YOU SHOULD CHANGE

RECORDS_PER_GOOGLE_PAGE = 100
GOOGLE_PAGES = 10
# Should enter these here, otherwise it'll prompt you for it...
MYSPACE_USERNAME = ''
MYSPACE_PASSWORD = ''

# mac
BROWSER_EXE = 'open -a firefox'
# windoze
# BROWSER_EXE = 'c:/progra~1/mozill~1/firefox.exe'

# Maximum sleep time between page reloads in seconds
MAX_SLEEP_TIME = 10

# DON'T CHANGE THIS STUFF

SCRIPT_VERSION = "1.0"
DBFILE = "myspace_adder_db.sqlite"
SEPARATOR = "--------------------------------------------------------------------------------"
# User agents
UA = [
   'Windows IE 6' ,
   'Windows Mozilla',
   'Mac Safari' ,
   'Mac Mozilla' ,
   'Linux Mozilla', 
   'Linux Konqueror' ]

# SETUP CONFIGURATION =========================================================
begin
  require 'rubygems'
  require 'active_record'
  require 'open-uri'
  require 'mechanize'
  require 'fileutils'
rescue
  puts "!!! SETUP ERROR !!!"
  puts "Please make sure you have followed the setup directions at: "
  puts "http://sublog.subimage.com/articles/2007/07/11/ruby-myspace-friend-adder-script"
end

class Friend < ActiveRecord::Base

end

class MySpaceAdder

  # Initializes...
  def initialize
    # Initialize mechanize agent
    @agent = WWW::Mechanize.new
    # Set random user agent
    @agent.user_agent_alias =  UA[rand(UA.size)]
    @myspace_username = MYSPACE_USERNAME
    if @myspace_username.blank?
      printf "Please enter your MySpace email address / username: "
      @myspace_username = gets.chomp
    end
    @myspace_password = MYSPACE_PASSWORD
    if @myspace_password.blank?
      printf "Please enter your MySpace password: "
      @myspace_password = gets.chomp
    end
  end

  # GOOGLE SEARCH
  #
  # Looks at google for phrase given and snags friend IDs
  #
  def do_google_search

    puts SEPARATOR
    puts ""
    puts "Let's search Google to find friends to add..."
    printf "What should we search for: "

    search_string = gets

    query  = "http://www.google.com/search?q=interests+#{search_string}"
    query << "+site:www.myspace.com&num=#{RECORDS_PER_GOOGLE_PAGE}"

    # Open url, get up to 10 pages of links since that's all google gives
    # Each page contains 100 search results...(or whatever you've set the variable to)
    puts "\n\nGrabbing page results..."
    for page_index in 0..GOOGLE_PAGES do
      puts "...getting page #{page_index}"
      query << "&start=#{page_index * RECORDS_PER_GOOGLE_PAGE}"
      page = @agent.get(query)
      # Rip out all the friend names
      page.links.each do |link|
        str_to_look_for = "www.myspace.com/"
        start_pos = link.href.index(str_to_look_for)
        next if !start_pos
        # Offset start slice pos
        start_pos = start_pos+str_to_look_for.size
        next if link.href.index("+")
        next if link.href.index("&")

        username = link.href[start_pos, link.href.length-1]

        if !Friend.find(:first, :conditions => ["username = ?", username])
          Friend.create(
           :username => username,
           :link => link.href
          ) 
        end
      end
      # Sleep between page grabs so google doesn't freak out...
      sleep_time = rand(MAX_SLEEP_TIME)
      puts "...sleeping #{sleep_time} seconds so Google doesn't freak out..."
      next if page_index == GOOGLE_PAGES
      sleep(sleep_time)
    end

    puts SEPARATOR

    puts "Total friend count: #{Friend.count()}"
    puts "Done searching...time to add these suckers"
  end

  # Tries to login to myspace...
  #
  def myspace_login
    # Login to myspace...
    puts "...Trying to login to MySpace..."
    page = @agent.get("http://www.myspace.com/?Mytoken=90f1eeb2-f110-401b-a150-d9053692048f")
    # Fill in the form
    login_form = page.form("theForm")
    login_form.email = @myspace_username
    login_form.password = @myspace_password
    # Click the login button
    page = @agent.submit(login_form)
    if !page.form("theForm")
      puts "...We're logged in!"
    else
      puts "...Please check your login and password. We couldn't log you in."
      exit
    end
  end

  # ADD FRIENDS
  #
  # Adds friends stored in the SQLite database.
  #
  def add_friends

    puts SEPARATOR

    self.myspace_login()

    # Loop through all friends
    Friend.find(:all).each do |f|
      # Navigate to person's page, find the add link, go there...
      puts "\n\nTrying to add: #{f.username}"
      # Remove friend from the db...just because we don't want to
      # do any odd condition checking later.
      f.destroy
      #
      @agent.read_timeout = 30
      begin
        page = @agent.get(f.link)
      rescue Timeout::Error
        "Getting the page is taking too long...skipping."
        next
      rescue
        "General error trying to load page...skipping"
        next
      end

      add_link = nil
      # Mechanize's find method on urls is shitty, so lets do this
      # oldschool
      while add_link == nil do
        page.links.each do |link|
          add_link = link.href if link.href && link.href.index('addfriend')
        end
        # Can't find the add link, move on...
        puts "Add link: #{add_link}"
        break if !add_link
      end
      # Get contents so we can parse...
      page_contents = @agent.get_file(add_link)
      page = @agent.get(add_link)

      # Look for the "confirm add friend button"
        if (page_contents =~ /person.is.already.your.Friend/)
            puts "#{f.username} is already your friend."
            next
        elsif (page_contents =~ /pending.friend.request/)
            puts "Pending add request."
            next
        elsif (page_contents =~ /only.accepts.add.requests.from/)
            puts "Private profile. Moving on."
            next
        elsif (page_contents =~ /from.bands/)
          puts "#{f.username} doesn't accept requests from bands..."
          next
        elsif (page_contents =~ /Invalid.FriendID/i)
          puts "Invalid friend ID, moving on..."
          next
        elsif (page_contents =~ /CAPTCHA/)
        puts "********************"
        puts "* CAPTCHA DETECTED *"
        puts "********************"
            puts add_link

            # I put escaped quotes around the '&' in the next line so that my bash 
            # wouldn't interpret it as a line break and foul up the system command
            system("#{BROWSER_EXE} \"#{add_link}\"")
          puts "Press enter when you're typing it in..."
          gets
          next
        end

        add_form = page.form("addFriend")
        if add_form && add_form.class != String
          @agent.submit(add_form)
        else
          puts "Couldn't find the add form...moving on"
        end

      # Sleep between friend adds grabs so myspace doesn't freak out...
      sleep_time = rand(MAX_SLEEP_TIME)
      puts "...sleeping #{sleep_time} seconds so MySpace doesn't freak out..."
      sleep(sleep_time)

    end
  end
end # / END CLASS

# DO THIS!@
puts "********************************************************************************"
puts "        S U B I M A G E     M Y S P A C E     F R I E N D     A D D E R"
puts "********************************************************************************"

#ActiveRecord::Base.logger = Logger.new(STDERR)
ActiveRecord::Base.colorize_logging = false

ActiveRecord::Base.establish_connection(
  :adapter => "sqlite3",
  :dbfile  => DBFILE
)

adder = MySpaceAdder.new()

if File.size(DBFILE) > 0
  puts "An existing database filled with (#{Friend.count()}) friends has been found."
  printf "Would you like to use it to finish your last adding session? (Y/n): "
  init_response = gets.chomp
end

if (File.size(DBFILE) == 0 || (init_response == "n" || init_response == "N"))
  puts "\nInitializing the database..."
  ActiveRecord::Schema.define do
    begin
      drop_table :friends
    rescue
      # Who cares if we can't drop it...
    end
    create_table :friends do |table|
      table.column :link, :string
      table.column :username, :string
    end
  end
  puts "Done...\n\n"
  adder.do_google_search()
  adder.add_friends()
else
  adder.add_friends()
end

puts "No more friends to add...start a new search."