module Oxidized
  module API
    class Mig
      def initialize(hash_router_db, cloginrc, path_new_router)
        @hash_router_db = hash_router_db
        @cloginrc = cloginrc
        @path_new_router = path_new_router
      end

      # read cloginrc and return a hash with node name, which a hash value which contains user,
      # password, eventually enable
      def cloginrc clogin_file
        close_file = clogin_file
        file = close_file.read
        file = file.gsub('add', '')

        hash = {}
        file.each_line do |line|
          # stock all device name, and password and enable if there is one
          line = line.split(' ')
          for i in 0..line.length
            if line[i] == 'user'
              # add the equipment and user if not exist
              unless hash[line[i + 1]]
                hash[line[i + 1]] = { user: line[i + 2] }
              end
            # if the equipment exist, add password and enable password
            elsif line[i] == 'password'
              if hash[line[i + 1]]
                if line.length > i + 2
                  h = hash[line[i + 1]]
                  h[:password] = line[i + 2]
                  if /\s*/.match(line[i + 3])
                    h[:enable] = line[i + 3]
                  end
                  hash[line[i + 1]] = h
                elsif line.length = i + 2
                  h = hash[line[i + 1]]
                  h[:password] = line[i + 2]
                  hash[line[i + 1]] = h
                end
              end
            end
          end
        end
        close_file.close
        hash
      end

      def model_dico model
        dico = { 'cisco' => 'ios', 'foundry' => 'ironware' }
        model = model.gsub("\n", '')
        if dico[model]
          model = dico[model]
        end
        model
      end

      # add node and group for an equipment (take a list of router.db)
      def rancid_group router_db_list
        model = {}
        hash = cloginrc @cloginrc
        router_db_list.each do |router_db|
          group = router_db[:group]
          file_close = router_db[:file]
          file = file_close.read
          file = file.gsub(':up', '')
          file.gsub(' ', '')

          file.each_line do |line|
            line = line.split(':')
            node = line[0]
            if hash[node]
              h = hash[node]
              model = model_dico line[1].to_s
              h[:model] = model
              h[:group] = group
            end
          end
          file_close.close
        end
        hash
      end
      
      # write a router.db conf, need the hash and the path of the file we whish create
      def write_router_db hash
        router_db = File.new(@path_new_router, 'w')
        hash.each do |key, value|
          line = key.to_s
          line += ":#{value[:model]}"
          line += ":#{value[:user]}"
          line += ":#{value[:password]}"
          line += ":#{value[:group]}"
          if value[:enable]
            line += ":#{value[:enable]}"
          end
          router_db.puts(line)
        end
        router_db.close
      end

      def edit_conf_file path_conf, router_db_path
        file_close = File.open(path_conf, 'r')
        file = file_close
        file = file.read
        source_reached = false
        new_file = []
        file.each_line do |line|
          if source_reached
            unless /^\w/.match(line)
              next
            end
            source_reached = false
          end
          new_file.push(line)
          if /source:/.match(line)
            source_reached = true
            new_file.push("  default: csv\n")
            new_file.push("  csv:\n")
            new_file.push("    file: #{router_db_path}\n")
            new_file.push("    delimiter: !ruby/regexp /:/\n")
            new_file.push("    map:\n")
            new_file.push("      name: 0\n")
            new_file.push("      model: 1\n")
            new_file.push("      username: 2\n")
            new_file.push("      password: 3\n")
            new_file.push("      group: 4\n")
            new_file.push("    vars_map:\n")
            new_file.push("      enable: 5\n")
            next
          end
        end
        file_close.close

        new_conf = File.new(path_conf, "w")
        new_file.each do |line|
          new_conf.puts(line)
        end
        new_conf.close
      end

      def go_rancid_migration
        hash = rancid_group @hash_router_db
        write_router_db hash
        edit_conf_file "#{ENV['HOME']}/.config/oxidized/config", @path_new_router
      end
    end
  end
end