module TwitterAuth
  module OauthUser
    def self.included(base)
      base.class_eval do
        attr_accessor :access_token, :access_secret
      end

      base.extend TwitterAuth::OauthUser::ClassMethods
      base.extend TwitterAuth::Dispatcher::Shared
    end
    
    module ClassMethods
      def identify_or_create_from_access_token(token, secret=nil)
        raise ArgumentError, 'Must authenticate with an OAuth::AccessToken or the string access token and secret.' unless (token && secret) || token.is_a?(OAuth::AccessToken)
        
        token = OAuth::AccessToken.new(TwitterAuth.consumer, token, secret) unless token.is_a?(OAuth::AccessToken)
        
        response = token.get(TwitterAuth.path_prefix + '/account/verify_credentials.json')
        user_info = handle_response(response)
        
        if user = User.find_by_login(user_info['screen_name'].to_s)
          if user.onboard_status == 1
            user.onboard_status = 2 
          end
            user.login = user_info['screen_name']
            user.assign_twitter_attributes(user_info)
            user.access_token = token.token
            user.access_secret = token.secret
            user.save
            user
        else
          User.create_from_twitter_hash_and_token(user_info, token) 
        end
      end

      def create_from_twitter_hash_and_token(user_info, access_token)
        user = User.new_from_twitter_hash(user_info)
        user.access_token = access_token.token
        user.access_secret = access_token.secret
        user.onboard_status = 0
        user.save
        user
      end
    end

    def token
      OAuth::AccessToken.new(TwitterAuth.consumer, access_token, access_secret)
    end 
  end
end