cgi-bin/ljlinks.pl

   1 #!/usr/bin/perl
   2
   3 #
   4 # Functions for lists of links created by users for display in their journals
   5 #
   6
   7 use strict;
   8
   9 package LJ::Links;
  10
  11 # linkobj structure:
  12 #
  13 # $linkobj = [
  14 #    { 'title'     => 'link title',
  15 #      'url'       => 'http://www.somesite.com',
  16 #      'children'  => [ ... ],
  17 #    },
  18 #    { ... },
  19 #    { ... },
  20 # ];
  21
  22 sub load_linkobj
  23 {
  24     my ($u, $use_master) = @_;
  25     return unless LJ::isu($u);
  26
  27     # check memcache for linkobj
  28     my $memkey = [$u->{'userid'}, "linkobj:$u->{'userid'}"];
  29     my $linkobj = LJ::MemCache::get($memkey);
  30     return $linkobj if defined $linkobj;
  31
  32     # didn't find anything in memcache
  33     $linkobj = [];
  34
  35     {
  36         # not in memcache, need to build one from db
  37         my $db = $use_master ? LJ::get_cluster_def_reader($u) : LJ::get_cluster_reader($u);
  38
  39         local $" = ",";
  40         my $sth = $db->prepare("SELECT ordernum, parentnum, title, url " .
  41                                "FROM links WHERE journalid=?");
  42         $sth->execute($u->{'userid'});
  43         push @$linkobj, $_ while $_ = $sth->fetchrow_hashref;
  44     }
  45
  46     # sort in perl-space
  47     @$linkobj = sort { $a->{'ordernum'} <=> $b->{'ordernum'} } @$linkobj;
  48
  49     # fix up the data structure
  50     foreach (@$linkobj) {
  51
  52         # TODO: build child relationships
  53         #       and store in $_->{'children'}
  54
  55         # ordernum/parentnum are only exposed via the
  56         # array structure, delete them here
  57         delete $_->{'ordernum'};
  58         delete $_->{'parentnum'};
  59     }
  60
  61     # set linkobj in memcache
  62     LJ::MemCache::set($memkey, $linkobj);
  63
  64     return $linkobj;
  65 }
  66
  67 sub save_linkobj
  68 {
  69     my ($u, $linkobj) = @_;
  70     return undef unless LJ::isu($u) && ref $linkobj eq 'ARRAY' && $u->writer;
  71
  72     # delete old links, we'll rebuild them shortly
  73     $u->do("DELETE FROM links WHERE journalid=?", undef, $u->{'userid'});
  74
  75     # only save allowed number of links
  76     my $numlinks = @$linkobj;
  77     my $caplinks = LJ::get_cap($u, "userlinks");
  78     $numlinks = $caplinks if $numlinks > $caplinks;
  79
  80     if ($numlinks) {
  81         # build insert query
  82         my (@bind, @vals);
  83         foreach my $ct (1..$numlinks) {
  84             my $it = $linkobj->[$ct-1];
  85             # journalid, ordernum, parentnum, url, title
  86             push @bind, "(?,?,?,?,?)";
  87             push @vals, ($u->{'userid'}, $ct, 0, $it->{'url'}, $it->{'title'});
  88         }
  89         my $binds = join(",", @bind);
  90         $u->do("INSERT INTO links (journalid, ordernum, parentnum, url, title) VALUES $binds", undef, @vals);
  91     }
  92
  93     # invalidate memcache
  94     my $memkey = [$u->{'userid'}, "linkobj:$u->{'userid'}"];
  95     LJ::MemCache::delete($memkey);
  96
  97 }
  98
  99 sub make_linkobj_from_form
 100 {
 101     my ($u, $post) = @_;
 102     return unless LJ::isu($u) && ref $post eq 'HASH';
 103
 104     my $linkobj = [];
 105
 106     # remove leading and trailing spaces
 107     my $stripspaces = sub {
 108         my $str = shift;
 109         $str =~ s/^\s*//;
 110         $str =~ s/\s*$//;
 111         return $str;
 112     };
 113
 114     # find number of links allowed
 115     my $numlinks = $post->{'numlinks'};
 116     my $caplinks = LJ::get_cap($u, "userlinks");
 117     $numlinks = $caplinks if $numlinks > $caplinks;
 118
 119     foreach my $num (sort { $post->{"link_${a}_ordernum"} <=>
 120                             $post->{"link_${b}_ordernum"} } (1..$numlinks)) {
 121
 122         # title is required
 123         my $title = $post->{"link_${num}_title"};
 124         $title = $stripspaces->($title);
 125         next unless $title;
 126
 127         my $url = $post->{"link_${num}_url"};
 128         $url = $stripspaces->($url);
 129
 130         # smartly add http:// to url unless they are just inserting a blank line
 131         if ($url && $title ne '-') {
 132             $url = LJ::CleanHTML::canonical_url($url);
 133         }
 134
 135         # build link object element
 136         $post->{"link_${num}_url"} = $url;
 137         push @$linkobj, { 'title' => $title, 'url' => $url };
 138
 139         # TODO: build child relationships
 140         #       push @{$linkobj->[$parentnum-1]->{'children'}}, $myself
 141     }
 142
 143     return $linkobj;
 144 }
 145
 146 1;