package OnTheBeachDB::Messages;
    
use base qw/DBIx::Class/;  
    
# Load required DBIC stuff
__PACKAGE__->load_components(qw/PK::Auto Core/);
# Set the table name
__PACKAGE__->table('messages');
# Set columns in table
__PACKAGE__->add_columns(qw/message_id parent_id created title name link message/);
# Set the primary key for the table
__PACKAGE__->set_primary_key(qw/message_id/);

###########################################
##
##   THIS WHOLE THING IS TO BECOME A ResultSet CLASS
##
###########################################

sub create {
	
    ## MAKE ME PLUGINS #####
    require HTML::Scrubber; 
    require DateTime; #POSIX;
    ########################

    my ( $self, 
	 $c,
	 $name,
	 $title,
	 $message,
	 $parent_id ) = @_;
    
    
    ## hackish right now, needs to go in Forum::Util 
    ## or some such
    my @lt = localtime(time);
    my $date = DateTime->now; #POSIX::strftime('%Y-%m-%d %H:%M:%S', @lt);
     
    ## create the HTML::Scrubber object to un-taint 
    ## form data and dis/allow HTML
    my $scrubber = HTML::Scrubber->new( allow => [ qw[ p b i u hr br ] ] ); 
                  
    # cleaninate!                                                                            
    my $clean_name    =  $scrubber->scrub($name);
    my $clean_title   =  $scrubber->scrub($title);
    my $clean_message =  $scrubber->scrub($message);   
    
    # insert into db
    $c->model('OnTheBeachDB::Messages')->create({
    
         name      => $clean_name,
         title     => $clean_title,
         message   => $clean_message,
         created   => $date,
         parent_id => $parent_id,
    
    });
	     
	 
	
}

sub get_thread_ids {
	
	my  ( $self, $c ) = @_;
	my @ids;
	my $rs = $c->model('OnTheBeachDB::Messages')->search({ parent_id => '0' });
	
	while ( my $id = $rs->next ) {
		
		push ( @ids,  $id->message_id );
		
	}
	
	return @ids;
	
}   

sub get_thread_messages {
	
	my ($self,  $c, $parent_id, $depth, $replies ) = @_;
	my $output;
	
	## if no replies, set up the link given
	if ( !$replies) {
		
		$output .= $self->generate_link( $c, $parent_id, $depth );
		
	}
	
	## Get the ids of any children
	my $rs = $c->model('OnTheBeachDB::Messages')->search( 
	    {
	        parent_id => $parent_id	
	    },
	    {
	    	group_by  => 'created',
	    }
	);
	
	## Loop on each child found, if it has children, 
	## recurse on this function, if now just generate it's link
	while ( my $id = $rs->next ) {
		
		## Let's see if this child message has children of it's own
		if ( $self->has_children( $c, $id->message_id ) ) {
			
			$output .= $self->get_thread_messages( $c, $id->message_id, $depth + 2 );
			
		} else {
			
			$output .= $self->generate_link( $c, $id->message_id, $depth + 2 );
			
			
		}
		
	}
	
	return $output;
	
} 

## See if this ID has any children
sub has_children {
	
	my ( $self, $c, $id ) = @_;
	
	my $count = $c->model('OnTheBeachDB::Messages')->search(
	    {
		     message_id => $id,
	    }
	)->count();
	
	if ( $count > 0 ) {
		
		return 1;
		
	} else {
		
		return 0;
		
	}
	
}

## Count how many children this thread has
sub number_of_children {
	
	my ( $self, $c, $id ) = @_;
	
	my $count = $c->model('OnTheBeachDB::Messages')->search(
	    {
		     parent_id => $id,
	    }
	)->count();
	
	return $count;

}
	

## This is what you will see on the actual page
## Generate a link, with username, thread title, date
## Add    per $depth size
sub generate_link {
	
	my ( $self, $c, $id, $depth ) = @_;
	my $output;
	
	my $rs = $c->model('OnTheBeachDB::Messages')->find($id);
	
	my $title        = $rs->title;
	my $name         = $rs->name;
	my $created      = $rs->created;
	my $thread_text  = $rs->message;
	
	##### DEFINITELY want to get this into a template somehow. Todo list. ##
	## Indendation multiplied by the supplied $depth number
	my $spaces = '  ' x $depth;
	
    
	$output .= $spaces . 
	           qq!<a href="! . $c->uri_for('view/') . $id . qq!">! .
	           qq!$title</a>! . 
	           qq!&nbsp;by <a href="! . $c->uri_for('/users/') . $name . qq!">$name</a>! .
	           qq!&nbsp;on $created <br />\n! . 
	           $spaces . qq!$thread_text<br />\n! . 
	           $spaces . qq!<a href="! . $c->uri_for('reply/') . $id . qq!">comment on ! . $title . qq! </a><br /><br />!;
	##################
}
=head1 NAME
    
    OnTheBeachDB::Message - A model object representing a forum message.
    
=head1 DESCRIPTION
  
=cut
    
1;


##</code><code>##

sub index : Private {
	
    my ( $self, $c ) = @_;
    my @threads;
    
    ## set up the query to get all threads that aren't children
    my $rs = $c->model('OnTheBeachDB::Messages')->search ( { parent_id => '0' }, { order_by => 'created DESC'} );
    
    
    ## loop through the parent threads, 
    ## count the number of children,
    ## and push a reference to %thread_info into @threads
    
    while ( my $thread = $rs->next ) {
    	
    	my %thread_info;
    	     
    	$thread_info{'name'}          = $thread->name;
    	$thread_info{'created'}       = $thread->created;
    	$thread_info{'title'}         = $thread->title;
    	$thread_info{'message_id'}    = $thread->message_id;
    	$thread_info{'message'}       = $thread->message;
    	$thread_info{'comment_count'} = OnTheBeachDB::Messages->number_of_children( $c, $thread->message_id );
    	push (@threads, \%thread_info)
   
    	
    }
    
    $c->log->debug($c->model("OnTheBeachDB::Forums")->find(1)->posts_rs->next->title);
   
    $c->stash->{page}     = $c->model('OnTheBeachDB::Page')->find({ name => "forum_main" });
    $c->stash->{threads}  = \@threads;
}

sub create : Local {
	my ( $self, $c ) = @_;
	
	
}

sub do_create : Local {
	my ( $self, $c ) = @_;
	
	## for validation
	#unless ( $c->form->has_error ) {
        
       OnTheBeachDB::Messages->create(
           $c,
           $c->req->param('name'),
           $c->req->param('title'),
           $c->req->param('comments'),
           0, # since it's a new thread
       
       ); 
            
    #} else {
    	
    #	$c->detach('create');
    	
    #}
    
}

sub reply : Local {
	my ( $self, $c, $replyid ) = @_;
	
	my $thread = $c->model('OnTheBeachDB::Messages')->find($replyid);
	$c->stash->{thread} = $thread;
	
	
}

sub do_reply : Local {
	
	my ( $self, $c ) = @_;
	
	## form validation
	## change to ::Controller::FormBuilder
	#unless ( $c->form->has_error ) {
        
       OnTheBeachDB::Messages->create(
           $c,
           $c->req->param('name'),
           $c->req->param('title'),
           $c->req->param('comments'),
           $c->req->param('parent_id'), # since it's a new thread
       
       ); 
            
    #} else {
    	
    #	$c->detach('create');
    	
    #}
    
    $c->stash->{template } = 'forum/do_create.tt2';
	
}

sub view : Local {
	my ( $self, $c, $id ) = @_;
	
	my $thread = OnTheBeachDB::Messages->get_thread_messages($c, $id, 'replies');
    
    
    $c->stash->{threads}  = $thread;
}