--
-- (C) 2017 - ntop.org
--

dirs = ntop.getDirs()
package.path = dirs.installdir .. "/scripts/lua/modules/?.lua;" .. package.path
require "lua_utils"
local host_pools_utils = require "host_pools_utils"
if(ntop.isPro()) then
  package.path = dirs.installdir .. "/pro/scripts/lua/modules/?.lua;" .. package.path
  shaper_utils = require "shaper_utils"
end

sendHTTPHeader('text/html; charset=iso-8859-1')

ntop.dumpFile(dirs.installdir .. "/httpdocs/inc/header.inc")
active_page = "admin"
dofile(dirs.installdir .. "/scripts/lua/inc/menu.lua")

if not isAdministrator() then
  return
end

if _POST["edit_pools"] ~= nil then
  local config = paramsPairsDecode(_POST, true)

  for pool_id, pool_name in pairs(config) do
    -- create or rename
    host_pools_utils.createPool(ifId, pool_id, pool_name)

    if(interface.isBridgeInterface(ifId) == true) then
      -- create default shapers
      shaper_utils.setProtocolShapers(ifId,
          pool_id,
          shaper_utils.POOL_SHAPER_DEFAULT_PROTO_KEY,
          shaper_utils.DEFAULT_SHAPER_ID --[[ ingress shaper --]],
          shaper_utils.DEFAULT_SHAPER_ID --[[ egress shaper --]],
          true) -- Set only if the key does NOT exist
    end
  end
  -- Note: do not call reload here
elseif _POST["pool_to_delete"] ~= nil then
  host_pools_utils.deletePool(ifId, _POST["pool_to_delete"])
  interface.reloadHostPools()
elseif (_POST["edit_members"] ~= nil) then
  local pool_to_edit = _POST["pool_id"]
  local config = paramsPairsDecode(_POST, true)

  -- This code handles member address changes

  -- delete old addresses
  for k,old_member in pairs(config) do
    if((not isEmptyString(old_member)) and (k ~= old_member)) then
      host_pools_utils.deleteFromPoll(ifId, pool_to_edit, old_member)
    end
  end

  -- add new addresses
  for new_member,k in pairs(config) do
    if k ~= new_member then
      host_pools_utils.addToPool(ifId, pool_to_edit, new_member)
    end
  end

  interface.reloadHostPools()
elseif (_POST["member_to_delete"] ~= nil) then
  local pool_to_edit = _POST["pool_id"]

  host_pools_utils.deleteFromPoll(ifId, pool_to_edit, _POST["member_to_delete"])
  interface.reloadHostPools()
end

function printPoolNameField(pool_id_str)
  print[[<div class="form-group has-feedback" style="margin-bottom:0;">]]
  print[[<input name="pool_' + ]] print(pool_id_str) print[[ + '" class="form-control" data-unique="unique" placeholder="]] print(i18n("host_pools.specify_pool_name")) print[[" required/>]]
  print[[<div class="help-block with-errors" style="margin-bottom:0;"></div>]]
  print[[</div>]]
end

function printMemberNameField(member_str, origin_value_str)
  print[[<div class="form-group has-feedback" style="margin-bottom:0;">]]
  print[[<input name="member_' + ]] print(member_str) print[[ + '" class="form-control" data-address="address" data-member="member" placeholder="]] print(i18n("host_pools.specify_member_address")) print[["]]
  if not isEmptyString(origin_value_str) then
    print[[ data-origin-value="' + ]] print(origin_value_str) print[[ + '"]]
  end
  print[[ required/>]]
  print[[<div class="help-block with-errors" style="margin-bottom:0;"></div>]]
  print[[</div>]]
end

function printMemberVlanField(member_str)
  print[[<div class="form-group has-feedback" style="margin-bottom:0;">]]
  print[[<input name="member_' + ]] print(member_str) print[[ + '_vlan" class="form-control text-right" data-member="member" style="width:5em; padding-right:1em; margin: 0 auto;" type="number" min="0" value="0" required/>]]
  print[[<div class="help-block with-errors" style="margin-bottom:0;"></div>]]
  print[[</div>]]
end

--------------------------------------------------------------------------------

local selected_pool_id = _GET["pool"]
local available_pools = {}

for _, pool_id in host_pools_utils.listPools(ifId) do
  available_pools[#available_pools + 1] = {id=pool_id, name=host_pools_utils.getPoolName(ifId, pool_id)}
end

if tonumber(selected_pool_id) == nil then
  if #available_pools > 0 then
    selected_pool_id = available_pools[1].id
  else
    selected_pool_id = "0"
  end
end

--------------------------------------------------------------------------------

print [[
<hr>
<h2>]] print(i18n("host_pools.edit_host_pools")) print[[</h2>
<br>
  <ul id="hostPoolsNav" class="nav nav-tabs" role="tablist">
    <li><a data-toggle="tab" role="tab" href="#manage">]] print(i18n("host_pools.manage_pools")) print[[</a></li>
    <li><a data-toggle="tab" role="tab" href="#create">]] print(i18n("host_pools.create_pools")) print[[</a></li>
  </ul>
  <div class="tab-content">
    <div id="manage" class="tab-pane">
]]

  print('<br/>') print(i18n("host_pools.pool")) print(': <select class="form-control pool-selector" style="display:inline;" onchange="document.location.href=\'?pool=\' + $(this).val() + \'#manage\';">')
  for _,pool in ipairs(available_pools) do
    print('<option value="'..tostring(pool.id)..'"')
    if pool.id == selected_pool_id then
      print(" selected")
    end
    print('>'..(pool.name)..'</option>\n')
  end
  print('</select>\n')

  print[[
      <form id="table-manage-form">
        <br/><div id="table-manage"></div>
        <button class="btn btn-primary" style="float:right; margin-right:1em;" disabled="disabled" type="submit">]] print(i18n("save_settings")) print[[</button>
      </form>
]]

print[[
      <br/><br/>
    </div>
    <div id="create" class="tab-pane">
      <form id="table-create-form">
        <br/><div id="table-create"></div>
        <button class="btn btn-primary" style="float:right; margin-right:1em;" disabled="disabled" type="submit">]] print(i18n("save_settings")) print[[</button>
      </form>
      <br/><br/>
    </div>
  </div>
]]

--------------------------------------------------------------------------------

print[[
  <script>
    /* Make the pair address,vlan unique */
    function addressValidator(input) {
      var member = input.val();

      /* this will be checked separately */
      if (! member)
        return true;

      return is_mac_address(member) || is_network_mask(member, true);
    }
  
    function memberValidator(input) {
      var member = input.val();

      /* this will be checked separately */
      if (! member)
        return true;

      var address_value;
      var vlan_value;
      if (input.attr("name").endsWith("_vlan")) {
        var name = input.attr("name").split("_vlan")[0];
        address_value = $("input[name='" + name + "']", $("#table-manage-form")).val();
        vlan_value = member;
      } else {
        var name = input.attr("name") + "_vlan";
        address_value = member;
        vlan_value = $("input[name='" + name + "']", $("#table-manage-form")).val();
      }

      var identifier = address_value + "@" + vlan_value;
      var count = 0;

      $('input[name^="member_"]:not([name$="_vlan"])', $("#table-manage-form")).each(function() {
        var address_value = $(this).val();
        var name = $(this).attr("name") + "_vlan";
        vlan_value = $("input[name='" + name + "']", $("#table-manage-form")).val();

        var aggregated = address_value + "@" + vlan_value;
        if (aggregated === identifier)
          count++;
      });

      return count == 1;
    }
  </script>
]]

-- ==== Manage tab ====

print [[

  <script>
    var maxMembersNum = ]] print(tostring(host_pools_utils.MAX_MEMBERS_NUM)) print[[;
    var addedMemberCtr = 0;

    function addPoolMember() {
      if (datatableIsEmpty("#table-manage"))
         datatableRemoveEmptyRow("#table-manage");

      var member_id = addedMemberCtr++;
      var newid = "member_" + member_id;

      var tr = $('<tr id=' + newid + '><td>]] printMemberNameField('member_id') print[[</td><td class="text-center">]] printMemberVlanField('member_id') print[[</td><td class="text-center"></td></tr>');
      datatableAddDeleteButtonCallback.bind(tr)(3, "datatableUndoAddRow('#" + newid + "', ']] print(i18n("host_pools.empty_pool")) print[[', '#addPoolMemberBtn')", "]] print(i18n('undo')) print[[");
      $("#table-manage table").append(tr);
      $("input", tr).first().focus();

      aysRecheckForm("#table-manage-form");
      recheckMemberAddButton();
    }

    function deletePoolMember(member_id) {
      var form = $("#table-manage-form");
      var field = form.find("input[name='member_" + member_id + "']");

      if (field.attr("data-origin-value")) {
        var params = {};
        params.pool_id = ]] print(selected_pool_id) print[[;
        params.member_to_delete = field.attr("data-origin-value");
        params.csrf = "]] print(ntop.getRandomCSRFValue()) print[[";
        paramsToForm('<form method="post"></form>', params).appendTo('body').submit();
      }
    }

    $("#table-manage").datatable({
      url: "]]
   print (ntop.getHttpPrefix())
   print [[/lua/get_host_pools.lua?ifid=]] print(ifId.."") print[[&pool=]] print(selected_pool_id) print[[",
      title: "",
      hidePerPage: true,
      hideDetails: true,
      showPagination: false,
      perPage: maxMembersNum,
      forceTable: true,
      
      buttons: [
         '<a id="addPoolMemberBtn" onclick="addPoolMember()" role="button" class="add-on btn" data-toggle="modal"><i class="fa fa-plus" aria-hidden="true"></i></a>'
      ], columns: [
         {
            title: "]] print(i18n("host_pools.member_address")) print[[",
            field: "column_member",
            css: {
               textAlign: 'left',
               verticalAlign: 'middle'
            }
         }, {
            title: "VLAN",
            field: "column_vlan",
            css: {
               width: '10%',
               textAlign: 'center',
               verticalAlign: 'middle'
            }
         }, {
            title: "]] print(i18n("actions")) print[[",
            css : {
               width: '10%',
               textAlign: 'center',
               verticalAlign: 'middle'
            }
         }
      ], tableCallback: function() {
        if (]] print(selected_pool_id) print[[ == ]] print(host_pools_utils.DEFAULT_POOL_ID) print[[) {
          datatableAddEmptyRow("#table-manage", "]] print(i18n("host_pools.default_pool_read_only")) print[[");
          $("#addPoolMemberBtn").attr("disabled", "disabled");
        } else if(datatableIsEmpty("#table-manage")) {
          datatableAddEmptyRow("#table-manage", "]] print(i18n("host_pools.empty_pool")) print[[");
        } else {
          datatableForEachRow("#table-manage", function() {
            var member_address = $("td:nth-child(1)", $(this));
            var vlan = $("td:nth-child(2)", $(this));
            var member_id = addedMemberCtr++;

            /* Make member name editable */
            var input = $(']] printMemberNameField('member_id', 'member_address.html() + \'@\' + vlan.html()') print[[');
            $("input", input).first().val(member_address.html());
            member_address.html(input);

            /* Make vlan editable */
            var input = $(']] printMemberVlanField('member_id') print[[');
            $("input", input).first().val(vlan.html());
            vlan.html(input);

            datatableAddDeleteButtonCallback.bind(this)(3, "deletePoolMember('" + member_id + "')", "]] print(i18n('delete')) print[[");
          });

          aysResetForm('#table-manage-form');
        }

        recheckMemberAddButton();
        $("#table-manage-form")
            .validator(validator_options)
            .on('submit', checkManagePoolForm);
      }
    });

    function checkManagePoolForm(event) {
      if (event.isDefaultPrevented())
        return false;

      var form = $("#table-manage-form");
      
      // build the settings object
      var settings = {};
      $("input[name^='member_']", form).each(function() {
        var vlan_name = $(this).attr("name") + "_vlan";
        var vlan_field = $("input[name=" + vlan_name + "]", form);
        if (vlan_field.length == 1) {
          var original;

          if($(this).attr("data-origin-value"))
            original = $(this).attr("data-origin-value");
          else
            original = "";

          var member;
          if((member = is_network_mask($(this).val(), true)))
            member = member.address + "/" + member.mask;
          else
            member = $(this).val();
          settings[member + "@" + vlan_field.val()] = original;
        }
      });

      // reset ays so that we can submit a custom form
      aysResetForm(form);

      // create a form with key-values encoded
      var params = paramsPairsEncode(settings);
      params.edit_members = "";
      params.pool_id = ]] print(selected_pool_id) print[[;
      params.csrf = "]] print(ntop.getRandomCSRFValue()) print[[";
      paramsToForm('<form method="post"></form>', params).appendTo('body').submit();
      return false;
    }

    function recheckMemberAddButton() {
      if(addedMemberCtr >= maxMembersNum)
        $("#addPoolMemberBtn").attr("disabled", "disabled");
    }
  </script>
]]

-- ==== Create tab ====

print [[
  <script>
    var nextPoolId = 1;
    var maxPoolNum = ]] print(tostring(host_pools_utils.MAX_NUM_POOLS)) print[[;

    function deletePool(pool_id) {
      var params = {};
      params.pool_to_delete = pool_id;
      params.csrf = "]] print(ntop.getRandomCSRFValue()) print[[";

      var form = paramsToForm('<form method="post"></form>', params);
      if (pool_id == ]] print(selected_pool_id) print[[)
        form.attr("action", "?#create");

      form.appendTo('body').submit();
    }

    function onPoolAddUndo(newid) {
      var pool_id = parseInt(newid.split("added_pool_")[1]);
      if (pool_id)
        nextPoolId = Math.min(nextPoolId, pool_id);
    }

    function addPool() {
      var pool_id = nextPoolId;

      while (nextPoolId < maxPoolNum) {
        nextPoolId = nextPoolId+1;

        // find a new gap
        if ($("#table-create input[name='pool_" + nextPoolId +"']").length == 0)
          break;
      }

      if (datatableIsEmpty("#table-create"))
         datatableRemoveEmptyRow("#table-create");

      if (pool_id < maxPoolNum) {
        var newid = "added_pool_" + pool_id;

        var tr = $('<tr id=' + newid + '><td class="text-center">' + pool_id + '</td><td>]] printPoolNameField('pool_id') print[[</td><td class="text-center"></td></tr>');
        datatableAddDeleteButtonCallback.bind(tr)(3, "datatableUndoAddRow('#" + newid + "', ']] print(i18n("host_pools.no_pools")) print[[', '#addNewPoolBtn', 'onPoolAddUndo')", "]] print(i18n('undo')) print[[");
        datatableOrderedInsert("#table-create table", 1, tr, pool_id);
        $("input", tr).focus();

        aysRecheckForm("#table-create-form");
      }

      recheckPoolAddButton();
    }

    function recheckPoolAddButton() {
      if(nextPoolId >= maxPoolNum)
        $("#addNewPoolBtn").attr("disabled", "disabled");
    }

    $("#table-create").datatable({
      url: "]]
   print (ntop.getHttpPrefix())
   print [[/lua/get_host_pools.lua?ifid=]] print(ifId.."") print[[",
      title: "",
      hidePerPage: true,
      hideDetails: true,
      showPagination: false,
      perPage: maxPoolNum,
      forceTable: true,
      
      perPage: maxPoolNum,
      buttons: [
         '<a id="addNewPoolBtn" onclick="addPool()" role="button" class="add-on btn" data-toggle="modal"><i class="fa fa-plus" aria-hidden="true"></i></a>'
      ], columns: [
         {
            title: "]] print(i18n("host_pools.pool_id")) print[[",
            field: "column_pool_id",
            hidden: true,
            css: {
               textAlign: 'center',
               width: '5%',
               verticalAlign: 'middle'
            }
         }, {
            title: "]] print(i18n("host_pools.pool_name")) print[[",
            field: "column_pool_name",
            css: {
               textAlign: 'left',
               width: '50%',
               verticalAlign: 'middle'
            }
         }, {
            title: "]] print(i18n("actions")) print[[",
            css : {
               width: '10%',
               textAlign: 'center',
               verticalAlign: 'middle'
            }
         }
      ], tableCallback: function() {
        if(datatableIsEmpty("#table-create")) {
          datatableAddEmptyRow("#table-create", "]] print(i18n("host_pools.no_pools")) print[[");
        } else {
         datatableForEachRow("#table-create", function() {
            var pool_id = $("td:nth-child(1)", $(this)).html();
            var pool_name = $("td:nth-child(2)", $(this));

            /* Make pool name editable */
            if (pool_id != ]]  print(host_pools_utils.DEFAULT_POOL_ID) print[[) {
              var input = $(']] printPoolNameField('pool_id') print[[');
              $("input", input).first().val(pool_name.html());
              pool_name.html(input);

              datatableAddDeleteButtonCallback.bind(this)(3, "deletePool('" + pool_id + "')", "]] print(i18n('delete')) print[[");
            }
         });

         /* pick the first unused pool ID */
         $("#table-create td:nth-child(1)").each(function() {
            var this_pool_id = parseInt($(this).html());
            if(nextPoolId == this_pool_id)
               nextPoolId += 1;
         });

         aysResetForm('#table-creaate-form');
        }

        recheckPoolAddButton();
        $("#table-create-form")
          .validator(validator_options)
          .on('submit', checkCreatePoolForm);
      }
    });

    function checkCreatePoolForm(event) {
      if (event.isDefaultPrevented())
        return false;

      var form = $("#table-create-form");
      
      // build the settings object
      var settings = {};
      $("input[name^='pool_']", form).each(function() {
        var pool_id = $(this).attr("name").split("pool_")[1];
        settings[pool_id] = $(this).val();
      });

      // reset ays so that we can submit a custom form
      aysResetForm(form);

      // create a form with key-values encoded
      var params = paramsPairsEncode(settings);
      params.edit_pools = "";
      params.csrf = "]] print(ntop.getRandomCSRFValue()) print[[";
      paramsToForm('<form method="post"></form>', params).appendTo('body').submit();

      return false;
    }
  </script>
]]

print[[
  <script>
    handle_tab_state($("#hostPoolsNav"), "manage");

    aysHandleForm("form", {
      handle_datatable: true,
      handle_tabs: true,
      ays_options: {addRemoveFieldsMarksDirty: true}
    });
    
    var validator_options = {
      disable: true,
      custom: {
         member: memberValidator,
         address: addressValidator,
         unique: makeUniqueValidator(function(field) {
            return $('input[name^="pool_"]', $("#table-create-form"));
         }),
      }, errors: {
         member: "]] print(i18n("host_pools.duplicate_member")) print[[.",
         address: "]] print(i18n("host_pools.invalid_member")) print[[.",
         unique: "]] print(i18n("host_pools.duplicate_pool")) print[[.",
      }
    }

    /* Retrigger the validation every second to clear outdated errors */
    setInterval(function() {
      $("form:data(bs.validator)").each(function(){
        $(this).data("bs.validator").validate();
      });
    }, 1000);
  </script>
]]

dofile(dirs.installdir .. "/scripts/lua/inc/footer.lua")