[dw-free] Error handling for WTF backend

[mark]

[commit: http://hg.dwscoalition.org/dw-free/rev/3f64af57ebea]

http://bugs.dwscoalition.org/show_bug.cgi?id=368

Make add_edge call the methods to validate if the user can watch/trust.

Patch by mark.

Files modified:

• cgi-bin/DW/User/Edges/WatchTrust.pm

--------------------------------------------------------------------------------
diff -r 817d2372b6fa -r 3f64af57ebea cgi-bin/DW/User/Edges/WatchTrust.pm
--- a/cgi-bin/DW/User/Edges/WatchTrust.pm	Fri Mar 06 08:28:34 2009 +0000
+++ b/cgi-bin/DW/User/Edges/WatchTrust.pm	Fri Mar 06 08:34:51 2009 +0000
@@ -69,6 +69,10 @@ sub _add_wt_edge {
     $watch_edge ||= {};
     my $do_trust = $trust_edge ? 1 : 0;
     $trust_edge ||= {};
+
+    # throw errors if we're not allowed
+    return 0 if $do_watch && ! $from_u->can_watch( $to_u );
+    return 0 if $do_trust && ! $from_u->can_trust( $to_u );
 
     # get current record, so we know what to modify
     my $dbh = LJ::get_db_writer();
--------------------------------------------------------------------------------