/*
   * Copyright (C) 2015 Cenk Gündoğan <cnkgndgn@gmail.com>
   *
   * This file is subject to the terms and conditions of the GNU Lesser General
   * Public License v2.1. See the file LICENSE in the top level directory for
   * more details.
   */
  
  /**
   * @ingroup     sys_shell_commands.h
   * @{
   *
   * @file
   *
   * @author      Cenk Gündoğan <cnkgndgn@gmail.com>
   */
  
  #include <string.h>
  #include <stdio.h>
  #include "net/gnrc/ipv6/netif.h"
  #include "net/gnrc/rpl.h"
  #include "net/gnrc/rpl/structs.h"
  #include "net/gnrc/rpl/dodag.h"
  #include "utlist.h"
  #include "trickle.h"
  #ifdef MODULE_GNRC_RPL_P2P
  #include "net/gnrc/rpl/p2p.h"
  #include "net/gnrc/rpl/p2p_dodag.h"
  #include "net/gnrc/rpl/p2p_structs.h"
  #endif
  
  int _gnrc_rpl_init(char *arg)
  {
      kernel_pid_t iface_pid = atoi(arg);
      gnrc_ipv6_netif_t *entry = gnrc_ipv6_netif_get(iface_pid);
  
      if (entry == NULL) {
          puts("unknown interface specified");
          return 1;
      }
  
      gnrc_rpl_init(iface_pid);
      printf("successfully initialized RPL on interface %d\n", iface_pid);
      return 0;
  }
  
  int _gnrc_rpl_dodag_root(char *arg1, char *arg2)
  {
      uint8_t instance_id = atoi(arg1);
      ipv6_addr_t dodag_id;
  
      if (ipv6_addr_from_str(&dodag_id, arg2) == NULL) {
          puts("error: <dodag_id> must be a valid IPv6 address");
          return 1;
      }
  
      gnrc_rpl_instance_t *inst = gnrc_rpl_root_init(instance_id, &dodag_id, false, false);
      if (inst == NULL) {
          char addr_str[IPV6_ADDR_MAX_STR_LEN];
          printf("error: could not add DODAG (%s) to instance (%d)\n",
                  ipv6_addr_to_str(addr_str, &dodag_id, sizeof(addr_str)), instance_id);
          return 1;
      }
  
      printf("successfully added a new RPL DODAG\n");
      return 0;
  }
  
  #ifdef MODULE_GNRC_RPL_P2P
  int _gnrc_rpl_find(char *arg1, char *arg2)
  {
      uint8_t instance_id = atoi(arg1);
      ipv6_addr_t dodag_id;
      ipv6_addr_t target;
  
      if (ipv6_addr_from_str(&dodag_id, arg1) == NULL) {
          puts("<dodag_id> must be a valid IPv6 address");
          return 1;
      }
  
      if (ipv6_addr_from_str(&target, arg2) == NULL) {
          puts("<target> must be a valid IPv6 address");
          return 1;
      }
  
      if (gnrc_rpl_p2p_root_init(0, &dodag_id, &target, true) == NULL) {
          char addr_str[IPV6_ADDR_MAX_STR_LEN];
          printf("error: could not add DODAG (%s) to instance (%d)\n",
                  ipv6_addr_to_str(addr_str, &dodag_id, sizeof(addr_str)), instance_id);
          return 1;
      }
  
      printf("successfully initiated a P2P-RPL Route Discovery\n");
      return 0;
  }
  #endif
  
  int _gnrc_rpl_instance_remove(char *arg1)
  {
      uint8_t instance_id = atoi(arg1);
      gnrc_rpl_instance_t *inst;
  
      if ((inst = gnrc_rpl_instance_get(instance_id)) == NULL) {
          printf("error: could not find the instance (%d)\n", instance_id);
          return 1;
      }
  
      if (gnrc_rpl_instance_remove(inst) == false) {
          printf("error: could not remove instance (%d)\n", instance_id);
          return 1;
      }
  
      printf("success: removed instance (%d)\n", instance_id);
      return 0;
  }
  
  int _gnrc_rpl_trickle_reset(char *arg1)
  {
      uint8_t instance_id = atoi(arg1);
      gnrc_rpl_instance_t *inst;
      char addr_str[IPV6_ADDR_MAX_STR_LEN];
  
      if ((inst = gnrc_rpl_instance_get(instance_id)) == NULL) {
          puts("error: could not find the <instance_id>");
          return 1;
      }
  
      trickle_reset_timer(&(inst->dodag.trickle));
  
      printf("success: reset trickle timer of DODAG (%s) from instance (%d)\n",
              ipv6_addr_to_str(addr_str, &(inst->dodag.dodag_id), sizeof(addr_str)),
              instance_id);
      return 0;
  }
  
  int _gnrc_rpl_trickle_stop(char *arg1)
  {
      uint8_t instance_id = atoi(arg1);
      gnrc_rpl_instance_t *inst;
      char addr_str[IPV6_ADDR_MAX_STR_LEN];
  
      if ((inst = gnrc_rpl_instance_get(instance_id)) == NULL) {
          puts("error: could not find the <instance_id>");
          return 1;
      }
      trickle_stop(&(inst->dodag.trickle));
  
      printf("success: stopped trickle timer of DODAG (%s) from instance (%d)\n",
              ipv6_addr_to_str(addr_str, &(inst->dodag.dodag_id), sizeof(addr_str)),
              instance_id);
      return 0;
  }
  
  int _gnrc_rpl_trickle_start(char *arg1)
  {
      uint8_t instance_id = atoi(arg1);
      gnrc_rpl_instance_t *inst;
      char addr_str[IPV6_ADDR_MAX_STR_LEN];
  
      if ((inst = gnrc_rpl_instance_get(instance_id)) == NULL) {
          puts("error: could not find the <instance_id>");
          return 1;
      }
  
      trickle_start(gnrc_rpl_pid, &(inst->dodag.trickle), GNRC_RPL_MSG_TYPE_TRICKLE_MSG,
                    (1 << inst->dodag.dio_min), inst->dodag.dio_interval_doubl,
                    inst->dodag.dio_redun);
  
      printf("success: started trickle timer of DODAG (%s) from instance (%d)\n",
              ipv6_addr_to_str(addr_str, &(inst->dodag.dodag_id), sizeof(addr_str)),
              instance_id);
      return 0;
  }
  
  int _gnrc_rpl_send_dis(void)
  {
      gnrc_rpl_send_DIS(NULL, (ipv6_addr_t *) &ipv6_addr_all_rpl_nodes);
  
      puts("success: send a DIS\n");
      return 0;
  }
  
  #ifdef MODULE_NETSTATS_RPL
  int _stats(void)
  {
      puts(  "Statistics        (ucast) RX / TX                  RX / TX (mcast)");
      printf("DIO     #packets: %10" PRIu32 " / %-10" PRIu32 "  %10" PRIu32 " / %-10" PRIu32 "\n",
             gnrc_rpl_netstats.dio_rx_ucast_count, gnrc_rpl_netstats.dio_tx_ucast_count,
             gnrc_rpl_netstats.dio_rx_mcast_count, gnrc_rpl_netstats.dio_tx_mcast_count);
      printf("DIO       #bytes: %10" PRIu32 " / %-10" PRIu32 "  %10" PRIu32 " / %-10" PRIu32 "\n",
             gnrc_rpl_netstats.dio_rx_ucast_bytes, gnrc_rpl_netstats.dio_tx_ucast_bytes,
             gnrc_rpl_netstats.dio_rx_mcast_bytes, gnrc_rpl_netstats.dio_tx_mcast_bytes);
      printf("DIS     #packets: %10" PRIu32 " / %-10" PRIu32 "  %10" PRIu32 " / %-10" PRIu32 "\n",
             gnrc_rpl_netstats.dis_rx_ucast_count, gnrc_rpl_netstats.dis_tx_ucast_count,
             gnrc_rpl_netstats.dis_rx_mcast_count, gnrc_rpl_netstats.dis_tx_mcast_count);
      printf("DIS       #bytes: %10" PRIu32 " / %-10" PRIu32 "  %10" PRIu32 " / %-10" PRIu32 "\n",
             gnrc_rpl_netstats.dis_rx_ucast_bytes, gnrc_rpl_netstats.dis_tx_ucast_bytes,
             gnrc_rpl_netstats.dis_rx_mcast_bytes, gnrc_rpl_netstats.dis_tx_mcast_bytes);
      printf("DAO     #packets: %10" PRIu32 " / %-10" PRIu32 "  %10" PRIu32 " / %-10" PRIu32 "\n",
             gnrc_rpl_netstats.dao_rx_ucast_count, gnrc_rpl_netstats.dao_tx_ucast_count,
             gnrc_rpl_netstats.dao_rx_mcast_count, gnrc_rpl_netstats.dao_tx_mcast_count);
      printf("DAO       #bytes: %10" PRIu32 " / %-10" PRIu32 "  %10" PRIu32 " / %-10" PRIu32 "\n",
             gnrc_rpl_netstats.dao_rx_ucast_bytes, gnrc_rpl_netstats.dao_tx_ucast_bytes,
             gnrc_rpl_netstats.dao_rx_mcast_bytes, gnrc_rpl_netstats.dao_tx_mcast_bytes);
      printf("DAO-ACK #packets: %10" PRIu32 " / %-10" PRIu32 "  %10" PRIu32 " / %-10" PRIu32 "\n",
             gnrc_rpl_netstats.dao_ack_rx_ucast_count, gnrc_rpl_netstats.dao_ack_tx_ucast_count,
             gnrc_rpl_netstats.dao_ack_rx_mcast_count, gnrc_rpl_netstats.dao_ack_tx_mcast_count);
      printf("DAO-ACK   #bytes: %10" PRIu32 " / %-10" PRIu32 "  %10" PRIu32 " / %-10" PRIu32 "\n",
             gnrc_rpl_netstats.dao_ack_rx_ucast_bytes, gnrc_rpl_netstats.dao_ack_tx_ucast_bytes,
             gnrc_rpl_netstats.dao_ack_rx_mcast_bytes, gnrc_rpl_netstats.dao_ack_tx_mcast_bytes);
      return 0;
  }
  #endif
  
  int _gnrc_rpl_dodag_show(void)
  {
      printf("instance table:\t");
      for (uint8_t i = 0; i < GNRC_RPL_INSTANCES_NUMOF; ++i) {
          if (gnrc_rpl_instances[i].state == 0) {
              printf("[ ]");
          }
          else {
              printf("[X]");
          }
          putchar('\t');
      }
  
      putchar('\n');
  
      printf("parent table:\t");
      for (uint8_t i = 0; i < GNRC_RPL_PARENTS_NUMOF; ++i) {
          if (gnrc_rpl_parents[i].state == 0) {
              printf("[ ]");
          }
          else {
              printf("[X]");
          }
          putchar('\t');
      }
  
      putchar('\n');
  
  #ifdef MODULE_GNRC_RPL_P2P
      printf("p2p-rpl table:\t");
      for (int8_t i = 0; i < GNRC_RPL_P2P_EXTS_NUMOF; ++i) {
          if (gnrc_rpl_p2p_exts[i].state == 0) {
              printf("[ ]");
          }
          else {
              printf("[X]");
          }
          putchar('\t');
      }
  #endif
  
      putchar('\n');
  
      gnrc_rpl_dodag_t *dodag = NULL;
      char addr_str[IPV6_ADDR_MAX_STR_LEN];
      int8_t cleanup;
      uint64_t tc, xnow = xtimer_now_usec64();
  
      for (uint8_t i = 0; i < GNRC_RPL_INSTANCES_NUMOF; ++i) {
          if (gnrc_rpl_instances[i].state == 0) {
              continue;
          }
  
          dodag = &gnrc_rpl_instances[i].dodag;
  
          printf("instance [%d | Iface: %" PRIkernel_pid " | mop: %d | ocp: %d | mhri: %d | mri %d]\n",
                  gnrc_rpl_instances[i].id, dodag->iface,
                  gnrc_rpl_instances[i].mop, gnrc_rpl_instances[i].of->ocp,
                  gnrc_rpl_instances[i].min_hop_rank_inc, gnrc_rpl_instances[i].max_rank_inc);
  
          tc = (((uint64_t) dodag->trickle.msg_timer.long_target << 32)
                  | dodag->trickle.msg_timer.target) - xnow;
          tc = (int64_t) tc < 0 ? 0 : tc / US_PER_SEC;
  
          cleanup = dodag->instance->cleanup < 0 ? 0 : dodag->instance->cleanup;
  
          printf("\tdodag [%s | R: %d | OP: %s | PIO: %s | CL: %ds | "
                 "TR(I=[%d,%d], k=%d, c=%d, TC=%" PRIu32 "s)]\n",
                 ipv6_addr_to_str(addr_str, &dodag->dodag_id, sizeof(addr_str)),
                 dodag->my_rank, (dodag->node_status == GNRC_RPL_LEAF_NODE ? "Leaf" : "Router"),
                 ((dodag->dio_opts & GNRC_RPL_REQ_DIO_OPT_PREFIX_INFO) ? "on" : "off"),
                 (int) cleanup, (1 << dodag->dio_min), dodag->dio_interval_doubl, dodag->trickle.k,
                 dodag->trickle.c, (uint32_t) (tc & 0xFFFFFFFF));
  
  #ifdef MODULE_GNRC_RPL_P2P
          if (dodag->instance->mop == GNRC_RPL_P2P_MOP) {
              gnrc_rpl_p2p_ext_t *p2p_ext = gnrc_rpl_p2p_ext_get(dodag);
              printf("\tP2P-Ext [%s | HBH: %s | R: %s | L: %us]\n",
                      ipv6_addr_to_str(addr_str, &p2p_ext->target, sizeof(addr_str)),
                      p2p_ext->hop_by_hop ? "True" : "False",
                      p2p_ext->reply ? "True" : "False", p2p_ext->lifetime_sec);
          }
  #endif
  
          gnrc_rpl_parent_t *parent;
          LL_FOREACH(gnrc_rpl_instances[i].dodag.parents, parent) {
              printf("\t\tparent [addr: %s | rank: %d | lifetime: %" PRIu32 "s]\n",
                      ipv6_addr_to_str(addr_str, &parent->addr, sizeof(addr_str)),
                      parent->rank, parent->lifetime);
          }
      }
      return 0;
  }
  
  int _gnrc_rpl_operation(bool leaf, char *arg1)
  {
      uint8_t instance_id = atoi(arg1);
      gnrc_rpl_instance_t *inst;
  
      if ((inst = gnrc_rpl_instance_get(instance_id)) == NULL) {
          printf("error: could not find the instance (%d)\n", instance_id);
          return 1;
      }
  
      if (leaf) {
          gnrc_rpl_leaf_operation(&inst->dodag);
      }
      else {
          gnrc_rpl_router_operation(&inst->dodag);
      }
  
      printf("success: operate in instance (%d) as %s\n", instance_id, leaf ? "leaf" : "router");
      return 0;
  }
  
  #ifndef GNRC_RPL_WITHOUT_PIO
  int _gnrc_rpl_set_pio(char *inst_id, bool status)
  {
      uint8_t instance_id = atoi(inst_id);
      gnrc_rpl_instance_t *inst;
  
      if ((inst = gnrc_rpl_instance_get(instance_id)) == NULL) {
          printf("error: could not find the instance (%d)\n", instance_id);
          return 1;
      }
  
      gnrc_rpl_config_pio(&inst->dodag, status);
  
      printf("success: %sactivated PIO transmissions\n", status ? "" : "de");
      return 0;
  }
  #endif
  
  int _gnrc_rpl(int argc, char **argv)
  {
      if ((argc < 2) || (strcmp(argv[1], "show") == 0)) {
          return _gnrc_rpl_dodag_show();
      }
      else if ((argc == 3) && strcmp(argv[1], "init") == 0) {
          return _gnrc_rpl_init(argv[2]);
      }
      else if ((argc == 4) && strcmp(argv[1], "root") == 0) {
          return _gnrc_rpl_dodag_root(argv[2], argv[3]);
      }
      else if (strcmp(argv[1], "rm") == 0) {
          if (argc == 3) {
              return _gnrc_rpl_instance_remove(argv[2]);
          }
      }
      else if (strcmp(argv[1], "trickle") == 0) {
          if ((argc == 4) && (strcmp(argv[2], "reset") == 0)) {
              return _gnrc_rpl_trickle_reset(argv[3]);
          }
          else if ((argc == 4) && (strcmp(argv[2], "stop") == 0)) {
              return _gnrc_rpl_trickle_stop(argv[3]);
          }
          else if ((argc == 4) && (strcmp(argv[2], "start") == 0)) {
              return _gnrc_rpl_trickle_start(argv[3]);
          }
      }
      else if (strcmp(argv[1], "send") == 0) {
          if ((argc == 3) && (strcmp(argv[2], "dis") == 0)) {
              return _gnrc_rpl_send_dis();
          }
      }
      else if (strcmp(argv[1], "leaf") == 0) {
          if (argc == 3) {
              return _gnrc_rpl_operation(true, argv[2]);
          }
      }
      else if (strcmp(argv[1], "router") == 0) {
          if (argc == 3) {
              return _gnrc_rpl_operation(false, argv[2]);
          }
      }
      else if (strcmp(argv[1], "set") == 0) {
          if (argc > 2) {
  #ifndef GNRC_RPL_WITHOUT_PIO
              if (strcmp(argv[2], "pio") == 0) {
                  if ((argc == 5) && (strcmp(argv[3], "on") == 0)) {
                      return _gnrc_rpl_set_pio(argv[4], true);
                  }
                  else if ((argc == 5) && (strcmp(argv[3], "off") == 0)) {
                      return _gnrc_rpl_set_pio(argv[4], false);
                  }
              }
  #endif
          }
      }
  #ifdef MODULE_GNRC_RPL_P2P
      else if (strcmp(argv[1], "find") == 0) {
          if (argc == 4) {
              return _gnrc_rpl_find(argv[2], argv[3]);
          }
      }
  #endif
  #ifdef MODULE_NETSTATS_RPL
      else if (strcmp(argv[1], "stats") == 0) {
          return _stats();
      }
  #endif
  
  #ifdef MODULE_GNRC_RPL_P2P
      puts("* find <dodag_id> <target>\t\t\t- initiate a P2P-RPL route discovery");
  #endif
      puts("* help\t\t\t\t\t- show usage");
      puts("* init <if_id>\t\t\t\t- initialize RPL on the given interface");
      puts("* leaf <instance_id>\t\t\t- operate as leaf in the instance");
      puts("* trickle reset <instance_id>\t\t- reset the trickle timer");
      puts("* trickle start <instance_id>\t\t- start the trickle timer");
      puts("* trickle stop <instance_id>\t\t- stop the trickle timer");
      puts("* rm <instance_id>\t\t\t- delete the given instance and related dodag");
      puts("* root <inst_id> <dodag_id>\t\t- add a dodag to a new or existing instance");
      puts("* router <instance_id>\t\t\t- operate as router in the instance");
      puts("* send dis\t\t\t\t- send a multicast DIS");
  #ifndef GNRC_RPL_WITHOUT_PIO
      puts("* set pio <on/off> <instance_id>\t- (de-)activate PIO transmissions in DIOs");
  #endif
      puts("* show\t\t\t\t\t- show instance and dodag tables");
      return 0;
  }
  /**
   * @}
   */