contrib/auto-naming/update-named-status.rb

   1 #!/usr/bin/ruby
   2
   3 # update-named-status.rb - update the named status of routers in our database
   4 #
   5 # Copyright (c) 2007 Peter Palfrader
   6 #
   7 # Permission is hereby granted, free of charge, to any person obtaining a copy
   8 # of this software and associated documentation files (the "Software"), to deal
   9 # in the Software without restriction, including without limitation the rights
  10 # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  11 # copies of the Software, and to permit persons to whom the Software is
  12 # furnished to do so, subject to the following conditions:
  13 #
  14 # The above copyright notice and this permission notice shall be included in
  15 # all copies or substantial portions of the Software.
  16 #
  17 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  18 # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  19 # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  20 # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  21 # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  22 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  23 # SOFTWARE.
  24
  25 require "yaml"
  26
  27 require 'db'
  28 require 'db-config'
  29
  30 def do_update(verbose)
  31         now = $db.query_row("SELECT max(last_seen) AS max FROM router_claims_nickname")['max']
  32         unless now
  33                 STDERR.puts "Could not find the latest last_seen timestamp.  Is the database empty still?"
  34                 return
  35         end
  36         now = "TIMESTAMP '" + now.to_s + "'"
  37
  38         denamed = $db.do("
  39                         UPDATE router_claims_nickname
  40                         SET named=false
  41                         WHERE named
  42                           AND last_seen < #{now} - INTERVAL '6 months'")
  43         puts "de-named: #{denamed}" if verbose
  44
  45         named = $db.do("
  46                         UPDATE router_claims_nickname
  47                         SET named=true
  48                         WHERE NOT named
  49                           AND first_seen < #{now} - INTERVAL '2 weeks'
  50                           AND last_seen  > #{now} - INTERVAL '2 days'
  51                           AND NOT EXISTS (SELECT *
  52                                FROM router_claims_nickname AS innertable
  53                                WHERE named
  54                                  AND router_claims_nickname.nickname_id=innertable.nickname_id) "+ # if that nickname is already named, we lose.
  55                         " AND NOT EXISTS (SELECT *
  56                                FROM router_claims_nickname AS innertable
  57                                WHERE router_claims_nickname.nickname_id=innertable.nickname_id
  58                                  AND router_claims_nickname.router_id <> innertable.router_id
  59                                  AND last_seen > #{now} - INTERVAL '1 month') ") # if nobody else wanted that nickname in the last month we are set
  60         puts "named: #{named}" if verbose
  61 end
  62
  63 if __FILE__ == $0
  64         $db = Db.new($CONFIG['database']['dbhost'], $CONFIG['database']['dbname'], $CONFIG['database']['user'], $CONFIG['database']['password'])
  65         verbose = ARGV.first == "-v"
  66
  67         $db.transaction_begin
  68         do_update verbose
  69         $db.transaction_commit
  70 end