#include "transport/transport.h"
#include "starter/starter.h"
#include "curl/curl.h"

void prepare_amqp_connection(int argc, char const *const *argv,struct connection_data *condata) {
  int status;
  if(secure_getenv("AMQP_HOSTNAME")==NULL || secure_getenv("AMQP_HOSTNAME")==NULL || secure_getenv("AMQP_EXCHANGE")==NULL || secure_getenv("AMQP_BINDINGKEY")==NULL) {
    fprintf(stderr,"no amqp connection parameters in environment");
    exit(1);
  }
  condata->hostname = getenv("AMQP_HOSTNAME");
  condata->port = atoi(getenv("AMQP_PORT"));
  condata->exchange = getenv("AMQP_EXCHANGE");
  condata->bindingkey = getenv("AMQP_BINDINGKEY");

  condata->conn = amqp_new_connection();

  condata->socket = amqp_tcp_socket_new(condata->conn);
  if (!condata->socket) {
    error_log("failed to create TCP socket");
  }

  status = amqp_socket_open(condata->socket, condata->hostname, condata->port);
  if (status) {
    error_log("failed to open TCP socket");
  }

  amqp_assert(amqp_login(condata->conn, "/", 0, 131072, 0, AMQP_SASL_METHOD_PLAIN, "rmuser", "rmpassword"));
  amqp_channel_open(condata->conn, 1);
  amqp_assert(amqp_get_rpc_reply(condata->conn));

  {
    amqp_queue_declare_ok_t *r = amqp_queue_declare( condata->conn, 1, amqp_empty_bytes, 0, 0, 0, 1, amqp_empty_table);
    amqp_assert(amqp_get_rpc_reply(condata->conn));
    condata->queuename = amqp_bytes_malloc_dup(r->queue);
    if (condata->queuename.bytes == NULL) {
      fprintf(stderr, "Out of memory while copying queue name");
      exit(1);
    }
  }

  amqp_queue_bind(condata->conn, 1, condata->queuename, amqp_cstring_bytes(condata->exchange), amqp_cstring_bytes(condata->bindingkey), amqp_empty_table);
  amqp_assert(amqp_get_rpc_reply(condata->conn));

  amqp_basic_consume(condata->conn, 1, condata->queuename, amqp_empty_bytes, 0, 1, 0, amqp_empty_table);
  amqp_assert(amqp_get_rpc_reply(condata->conn));
}

void download_file(char* uri, char* path) {
  FILE* f = fopen(path, "w");
  //fwrite("test", 4,1,f);
  CURL *handle;
  handle = curl_easy_init();
  curl_easy_setopt(handle, CURLOPT_URL, uri);
  curl_easy_setopt(handle, CURLOPT_WRITEDATA, f);
  CURLcode code = curl_easy_perform(handle);
  if(code==CURLE_OK) {
      int response_code;
      curl_easy_getinfo(handle, CURLINFO_RESPONSE_CODE, &response_code);
      printf("response code:%d", response_code);
  } else {puts("not success");}
  sleep(1);
  //puts(res);
  curl_easy_cleanup(handle);
  fclose(f);
}

int main(int argc, char const *const *argv) {
  curl_global_init (CURL_GLOBAL_ALL);
  struct connection_data condata;
  prepare_amqp_connection(argc,argv,&condata);

  {
    for (;;) {
      amqp_rpc_reply_t res;
      amqp_envelope_t envelope;

      amqp_maybe_release_buffers(condata.conn);

      res = amqp_consume_message(condata.conn, &envelope, NULL, 0);

      if (AMQP_RESPONSE_NORMAL != res.reply_type) {
        break;
      }

      Runner__V1__Instruction *inst = runner__v1__instruction__unpack(NULL,envelope.message.body.len,envelope.message.body.bytes);
      if(inst==NULL) {
        fprintf(stderr, "error reading buffer");
        exit(1);
      }
      switch(inst->instruction_case) {
        case RUNNER__V1__INSTRUCTION__INSTRUCTION_BUILD:
          puts("build request recieved");
          printf("solution_id: %d\nbinding_key:%s\nlanguage:%d\nsolution uri:%s\nexecutable uri: %s\n\n",inst->build->solution_id,inst->build->binding_key,inst->build->language,inst->build->solution_uri, inst->build->executable_uri);
          download_file(inst->build->solution_uri, "/home/dragonmuffin/ms-runner/runs/solution");
          //struct limits limits;
          //limits.memory = 10000000;
          //limits.core = 2;
          //limits.time = 3000;
          //limits.processes = 1;

          //struct params params;
          //params.shared_folder = "/home/dragonmuffin/ms-runner/runs/";
          //params.argv = (char*[]){"/bin/ls", "/", NULL};

          //starter("/home/dragonmuffin/ms-runner/runs", "/home/dragonmuffin/ms-runner/starter", limits, params); // fix this

          break;
        case RUNNER__V1__INSTRUCTION__INSTRUCTION_RUN:
          puts("run request recieved");
          printf("solution_id: %d\ntest_id:%d\nbinding_key:%s\n\n",inst->run->solution_id,inst->run->test_id,inst->run->binding_key);
          break;
        case RUNNER__V1__INSTRUCTION__INSTRUCTION__NOT_SET:
          puts("empty request recieved");
          break;
        default:
          puts("unknown request recieved");
      }
      runner__v1__instruction__free_unpacked(inst, NULL);
      amqp_destroy_envelope(&envelope);
    }
  }

  amqp_bytes_free(condata.queuename);

  amqp_assert(amqp_channel_close(condata.conn, 1, AMQP_REPLY_SUCCESS));
  amqp_assert(amqp_connection_close(condata.conn, AMQP_REPLY_SUCCESS));
  amqp_code_assert(amqp_destroy_connection(condata.conn));
  return 0;
}