user.rb

class User
  include MongoMapper::Document

  devise :database_authenticatable, :registerable,
         :recoverable, :rememberable, :trackable, :validatable
         
  key :friend_ids, Array
  key :pending_request_ids, Array
  key :visible_post_ids, Array

  one :person, :class_name => 'Person', :foreign_key => :owner_id

  many :friends, :in => :friend_ids, :class_name => 'Person'
  many :pending_requests, :in => :pending_request_ids, :class_name => 'Request'
  many :raw_visible_posts, :in => :visible_post_ids, :class_name => 'Post'

  many :groups, :class_name => 'Group'

  after_validation_on_create :setup_person
  
  ######## Making things work ########
  key :email, String
  ensure_index :email

  def method_missing(method, *args)
    self.person.send(method, *args)
  end


  def real_name
    "#{person.profile.first_name.to_s} #{person.profile.last_name.to_s}"
  end
  
  ######### Groups ######################
  def group( opts = {} )
    opts[:user] = self
    Group.create(opts)
  end

  ######## Posting ########
  def post(class_name, options = {})
    options[:person] = self.person

    group_id = options[:group_id]
    options.delete(:group_id)

    model_class = class_name.to_s.camelize.constantize
    post = model_class.instantiate(options)
    post.creator_signature = post.sign_with_key(encryption_key)
    post.save


    if group_id
      group = self.groups.find_by_id(group_id)
      group.posts << post
      group.save
      post.push_to( group.people.all )
    else
      post.push_to( self.friends.all )
    end

    post.socket_to_uid(id) if post.respond_to?(:socket_to_uid)

    self.raw_visible_posts << post
    self.save
    post
  end
 
  def visible_posts( opts = {} )
    if opts[:by_members_of]
      group = self.groups.find_by_id( opts[:by_members_of].id )
      group.posts
    end
  end

  ######## Commenting  ########
  def comment(text, options = {})
    raise "must comment on something!" unless options[:on]
    comment = Comment.new(:person_id => self.person.id, :text => text, :post => options[:on])
    comment.creator_signature = comment.sign_with_key(encryption_key)
    if comment.save
      dispatch_comment comment
      comment.socket_to_uid id
      comment
    else
      Rails.logger.warn "this failed to save: #{comment.inspect}"
      false
    end
  end
  
  def dispatch_comment( comment )
    if owns? comment.post
      comment.post_creator_signature = comment.sign_with_key(encryption_key)
      comment.save
      comment.push_downstream
    elsif owns? comment
      comment.save
      comment.push_upstream
    end
  end
  
  ######### Posts and Such ###############

  def retract( post )
    post.unsocket_from_uid(self.id) if post.respond_to? :unsocket_from_uid
    retraction = Retraction.for(post)
    retraction.creator_signature = retraction.sign_with_key( encryption_key ) 
    retraction.push_to( self.friends.all )
    retraction
  end

  ########### Profile ######################
  def update_profile(params)
    if self.person.update_attributes(params)
      self.profile.push_to( self.friends.all )
      true
    else
      false
    end
  end

  ######### Friend Requesting ###########
  def send_friend_request_to(friend_url, group_id)
    unless self.friends.detect{ |x| x.receive_url == friend_url}
      request = Request.instantiate(:to => friend_url, :from => self.person, :into => group_id)
      if request.save
        self.pending_requests << request
        self.save

        group = self.group_by_id(group_id)

        group.requests << request
        group.save
        
        request.push_to_url friend_url
      end
      request
    end
  end 

  def accept_friend_request(friend_request_id, group_id)
    request = Request.find_by_id(friend_request_id)
    pending_requests.delete(request)
    
    activate_friend(request.person, group_by_id(group_id))

    request.reverse_for(self)
    request
  end
  
  def dispatch_friend_acceptance(request)
    request.push_to_url(request.callback_url)
    request.destroy unless request.callback_url.include? url
  end 
  
  def accept_and_respond(friend_request_id, group_id)
    dispatch_friend_acceptance(accept_friend_request(friend_request_id, group_id))
  end

  def ignore_friend_request(friend_request_id)
    request = Request.find_by_id(friend_request_id)
    person  = request.person

    person.user_refs -= 1

    self.pending_requests.delete(request)
    self.save

    (person.user_refs > 0 || person.owner.nil? == false) ?  person.save : person.destroy
    request.destroy
  end

  def receive_friend_request(friend_request)
    Rails.logger.info("receiving friend request #{friend_request.to_json}")

    if request_from_me?(friend_request)
      group = self.group_by_id(friend_request.group_id)
      activate_friend(friend_request.person, group)

      Rails.logger.info("#{self.real_name}'s friend request has been accepted")

      friend_request.destroy
    else

      friend_request.person.user_refs += 1
      friend_request.person.save
      self.pending_requests << friend_request
      self.save
      Rails.logger.info("#{self.real_name} has received a friend request")
      friend_request.save
    end
  end

  def unfriend(bad_friend)
    Rails.logger.info("#{self.real_name} is unfriending #{bad_friend.inspect}")
    retraction = Retraction.for(self)
    retraction.creator_signature = retraction.sign_with_key(encryption_key)
    retraction.push_to_url(bad_friend.receive_url) 
    remove_friend(bad_friend)
  end
  
  def remove_friend(bad_friend)
    raise "Friend not deleted" unless self.friend_ids.delete( bad_friend.id )
    groups.each{|g| g.person_ids.delete( bad_friend.id )}
    self.save

    self.raw_visible_posts.find_all_by_person_id( bad_friend.id ).each{|post|
      self.visible_post_ids.delete( post.id )
      post.user_refs -= 1
      (post.user_refs > 0 || post.person.owner.nil? == false) ?  post.save : post.destroy
    }
    self.save

    bad_friend.user_refs -= 1
    (bad_friend.user_refs > 0 || bad_friend.owner.nil? == false) ?  bad_friend.save : bad_friend.destroy
  end

  def unfriended_by(bad_friend)
    Rails.logger.info("#{self.real_name} is being unfriended by #{bad_friend.inspect}")
    remove_friend bad_friend
  end

  def send_request(rel_hash, group)
    if rel_hash[:friend]
      self.send_friend_request_to(rel_hash[:friend], group)
    else
      raise "you can't do anything to that url"
    end
  end
  
  def activate_friend(person, group)
    person.user_refs += 1
    group.people << person
    friends << person
    person.save
    group.save
    save
  end

  def request_from_me?(request)
    pending_requests.detect{|req| (req.callback_url == person.receive_url) && (req.destination_url == person.receive_url)}
  end

  ###### Receiving #######
  def receive xml
    object = Diaspora::Parser.from_xml(xml)
    Rails.logger.debug("Receiving object:\n#{object.inspect}")
    raise "Signature was not valid on: #{object.inspect}" unless object.signature_valid?
    if object.is_a? Retraction
      if object.type == 'Person' && object.signature_valid?

        Rails.logger.info( "the person id is #{object.post_id} the friend found is #{visible_person_by_id(object.post_id).inspect}")
        unfriended_by visible_person_by_id(object.post_id)

      else
        object.perform self.id
        groups = self.groups_with_person(object.person)
        groups.each{ |group| group.post_ids.delete(ensure_bson(object.post_id))
                             group.save
        }
      end
    elsif object.is_a? Request
      person = Diaspora::Parser.get_or_create_person_object_from_xml( xml )
      person.serialized_key ||= object.exported_key
      object.person = person
      object.person.save
      old_request =  Request.first(:id => object.id)
      object.group_id = old_request.group_id if old_request
      object.save
      receive_friend_request(object)
    elsif object.is_a? Profile
      person = Diaspora::Parser.owner_id_from_xml xml
      person.profile = object
      person.save  

    elsif object.is_a?(Comment) 
      dispatch_comment object unless owns?(object)
    else
      Rails.logger.debug("Saving object: #{object}")
      object.user_refs += 1
      object.save
      
      self.raw_visible_posts << object
      self.save

      groups = self.groups_with_person(object.person)
      groups.each{ |group| 
        group.posts << object
        group.save
        object.socket_to_uid(id, :group_id => group.id) if (object.respond_to?(:socket_to_uid) && !self.owns?(object))
      }

    end

  end

  ###Helpers############

  def terse_url
    terse= self.url.gsub(/https?:\/\//, '')
    terse.gsub!(/www\./, '')
    terse = terse.chop! if terse[-1, 1] == '/'
    terse
  end
 
  def visible_person_by_id( id )
    id = ensure_bson id
    return self.person if id == self.person.id
    friends.detect{|x| x.id == id }
  end

  def group_by_id( id )
    id = ensure_bson id
    groups.detect{|x| x.id == id }
  end

  def groups_with_person person
    id = ensure_bson person.id
    groups.select {|group| group.person_ids.include? id}
  end

  def setup_person
    self.person.serialized_key ||= generate_key.export
    self.person.email = email
    self.person.save!
  end

  protected
  
  def assign_key
    self.person.serialized_key ||= generate_key.export
  end

  def generate_key
    OpenSSL::PKey::RSA::generate 1024 
  end

  def self.generate_key
    OpenSSL::PKey::RSA::generate 1024 
  end

  def ensure_bson id 
    id.class == String ? BSON::ObjectID(id) : id 
  end
end