lib/Frey/Server.pm

package Frey::Server;

use strict;
use warnings;

use Continuity;
use Continuity::REPL;
use Data::Dump qw/dump/;

use base 'Frey';
use Frey::HTML;

my @messages;    # Global (shared) list of messages
my $got_message; # Flag to indicate that there is a new message to display

use vars qw( $repl $server );

#$repl = Continuity::REPL->new;
$server = Continuity->new(
  port => 16001,
  path_session => 1,
  cookie_session => 'sid',
  callback => \&main,
);
$server->debug_level( 2 );
sub run {

        $server->loop;
}

my @callbacks;
my $callback_count;

sub gen_link {
        my ($text, $code) = @_;
        $callbacks[$callback_count++] = $code;
        return qq{<a href="?cb=$callback_count">$text</a>};
}

sub process_links {
        my $request = shift;
        my $cb = $request->param('cb');
        if (exists $callbacks[$cb]) {
                $callbacks[$cb]->($request);
                delete $callbacks[$cb];
        }
}

# This is the main entrypoint. We are looking for one of three things -- a
# pushstream, a sent message, or a request for the main HTML. We delegate each
# of these cases, none of which will return (they all loop forever).
sub main {
  my ($req) = @_;
  
  my $path = $req->request->url->path;
  warn "REQUEST: $path\n";

        warn $req->request->header('User_Agent');
#warn dump( $req );

  # If this is a request for the pushtream, then give them that
  if($path =~ /pushstream/) {
        pushstream($req);
  }
  
  # If they are sending us a message, we give them a thread for that too
  if($path =~ /sendmessage/) {
        send_message($req);
  }

  # Otherwise, lets give them page
  send_page($req);
}

# Here we accept a connection to the browser, and keep it open. Meanwhile we
# watch the global $got_message variable, and when it gets touched we send off
# the list of messages through the held-open connection. Then we let the
# browser open a new connection and begin again.
sub pushstream {
  my ($req) = @_;
  # Set up watch event -- this will be triggered when $got_message is written
  my $w = Coro::Event->var(var => \$got_message, poll => 'w');
  while(1) {
    print STDERR "**** GOT MESSAGE, SENDING ****\n";
    my $log = join "<br>", @messages;
    $req->print($log);
    $req->next;
    print STDERR "**** Waiting for got_message indicator ****\n";
    $w->next;
  }
}


# Watch for the user to send us a message. As soon as we get it, we add it to
# our list of messages and touch the $got_message flag to let all the
# pushstreams know.
sub send_message {
  my ($req) = @_;
  while(1) {
    my $msg = $req->param('message');
    my $name = $req->param('username');
    if($msg) {
      unshift @messages, "$name: $msg";
      pop @messages if $#messages > 15; # Only keep the recent 15 messages
    }
    $got_message = 1;
    $req->print("Got it!");
    $req->next;
  }
}

# This isn't a pushstream, nor a new message. It is just the main page. We loop
# in case they ask for it multiple times :)
sub send_page {
        my ($req) = @_;
        my $templates = Template::Declare->templates;
        while(1) {
                warn "param = ",dump($req->param);
                my $path = $req->request->url->path;

                my $html;

                if ( $path =~ m/::/ ) {
                        my ( undef, $module, $method ) = split(m!/!, $path, 3);

                        if ( ! defined( $templates->{$module} ) ) {
                                $req->conn->send_status_line( 404, "$module" );
                                $html = "Package $module not found";
                        } elsif ( ! $method ) {
                                $html = Frey::HTML->page( 'package-methods', $req, $module );
                        } elsif ( grep(/^\Q$method\E$/, @{ $templates->{$module} }) ) {
                                $html = Frey::HTML->page( $method, $req );
                        } else {
                                $req->conn->send_status_line( 404, "$module $method" );
                                $html = "Package $module doesn't have $method";
                        }
                } else {
                        warn "fallback to status page\n";
                        $html = Frey::HTML->page( 'status' );
                }

                $req->print( $html );
                warn ">> ",length( $html ), " bytes\n";
                $req->next;
        }
}

1;
1	package Frey::Server;
2
3	use strict;
4	use warnings;
5
6	use Continuity;
7	use Continuity::REPL;
8	use Data::Dump qw/dump/;
9
10	use base 'Frey';
11	use Frey::HTML;
12
13	my @messages; # Global (shared) list of messages
14	my $got_message; # Flag to indicate that there is a new message to display
15
16	use vars qw( $repl $server );
17
18	#$repl = Continuity::REPL->new;
19	$server = Continuity->new(
20	port => 16001,
21	path_session => 1,
22	cookie_session => 'sid',
23	callback => \&main,
24	);
25	$server->debug_level( 2 );
26	sub run {
27
28	$server->loop;
29	}
30
31	my @callbacks;
32	my $callback_count;
33
34	sub gen_link {
35	my ($text, $code) = @_;
36	$callbacks[$callback_count++] = $code;
37	return qq{<a href="?cb=$callback_count">$text</a>};
38	}
39
40	sub process_links {
41	my $request = shift;
42	my $cb = $request->param('cb');
43	if (exists $callbacks[$cb]) {
44	$callbacks[$cb]->($request);
45	delete $callbacks[$cb];
46	}
47	}
48
49	# This is the main entrypoint. We are looking for one of three things -- a
50	# pushstream, a sent message, or a request for the main HTML. We delegate each
51	# of these cases, none of which will return (they all loop forever).
52	sub main {
53	my ($req) = @_;
54
55	my $path = $req->request->url->path;
56	warn "REQUEST: $path\n";
57
58	warn $req->request->header('User_Agent');
59	#warn dump( $req );
60
61	# If this is a request for the pushtream, then give them that
62	if($path =~ /pushstream/) {
63	pushstream($req);
64	}
65
66	# If they are sending us a message, we give them a thread for that too
67	if($path =~ /sendmessage/) {
68	send_message($req);
69	}
70
71	# Otherwise, lets give them page
72	send_page($req);
73	}
74
75	# Here we accept a connection to the browser, and keep it open. Meanwhile we
76	# watch the global $got_message variable, and when it gets touched we send off
77	# the list of messages through the held-open connection. Then we let the
78	# browser open a new connection and begin again.
79	sub pushstream {
80	my ($req) = @_;
81	# Set up watch event -- this will be triggered when $got_message is written
82	my $w = Coro::Event->var(var => \$got_message, poll => 'w');
83	while(1) {
84	print STDERR "** GOT MESSAGE, SENDING **\n";
85	my $log = join "<br>", @messages;
86	$req->print($log);
87	$req->next;
88	print STDERR "** Waiting for got_message indicator **\n";
89	$w->next;
90	}
91	}
92
93
94	# Watch for the user to send us a message. As soon as we get it, we add it to
95	# our list of messages and touch the $got_message flag to let all the
96	# pushstreams know.
97	sub send_message {
98	my ($req) = @_;
99	while(1) {
100	my $msg = $req->param('message');
101	my $name = $req->param('username');
102	if($msg) {
103	unshift @messages, "$name: $msg";
104	pop @messages if $#messages > 15; # Only keep the recent 15 messages
105	}
106	$got_message = 1;
107	$req->print("Got it!");
108	$req->next;
109	}
110	}
111
112	# This isn't a pushstream, nor a new message. It is just the main page. We loop
113	# in case they ask for it multiple times :)
114	sub send_page {
115	my ($req) = @_;
116	my $templates = Template::Declare->templates;
117	while(1) {
118	warn "param = ",dump($req->param);
119	my $path = $req->request->url->path;
120
121	my $html;
122
123	if ( $path =~ m/::/ ) {
124	my ( undef, $module, $method ) = split(m!/!, $path, 3);
125
126	if ( ! defined( $templates->{$module} ) ) {
127	$req->conn->send_status_line( 404, "$module" );
128	$html = "Package $module not found";
129	} elsif ( ! $method ) {
130	$html = Frey::HTML->page( 'package-methods', $req, $module );
131	} elsif ( grep(/^\Q$method\E$/, @{ $templates->{$module} }) ) {
132	$html = Frey::HTML->page( $method, $req );
133	} else {
134	$req->conn->send_status_line( 404, "$module $method" );
135	$html = "Package $module doesn't have $method";
136	}
137	} else {
138	warn "fallback to status page\n";
139	$html = Frey::HTML->page( 'status' );
140	}
141
142	$req->print( $html );
143	warn ">> ",length( $html ), " bytes\n";
144	$req->next;
145	}
146	}
147
148	1;