// Copyright 2020 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "gpioutil.h"

int ParseArgs(int argc, char** argv, GpioFunc* func, uint8_t* write_value,
              ::llcpp::fuchsia::hardware::gpio::GpioFlags* in_flag, uint8_t* out_value,
              uint64_t* ds_ua) {
  if (argc < 3) {
    return -1;
  }

  *write_value = 0;
  *in_flag = ::llcpp::fuchsia::hardware::gpio::GpioFlags::NO_PULL;
  *out_value = 0;
  *ds_ua = 0;
  unsigned long flag = 0;
  switch (argv[1][0]) {
    case 'r':
      *func = Read;
      break;
    case 'w':
      *func = Write;

      if (argc < 4) {
        return -1;
      }
      *write_value = static_cast<uint8_t>(std::stoul(argv[3]));
      break;
    case 'i':
      *func = ConfigIn;

      if (argc < 4) {
        return -1;
      }
      flag = std::stoul(argv[3]);
      if (flag > 3) {
        printf("Invalid flag\n\n");
        return -1;
      }
      *in_flag = static_cast<::llcpp::fuchsia::hardware::gpio::GpioFlags>(flag);
      break;
    case 'o':
      *func = ConfigOut;

      if (argc < 4) {
        return -1;
      }
      *out_value = static_cast<uint8_t>(std::stoul(argv[3]));
      break;
    case 'd':
      *func = SetDriveStrength;

      if (argc < 4) {
        return -1;
      }
      *ds_ua = static_cast<uint64_t>(std::stoull(argv[3]));
      break;
    default:
      *func = Invalid;
      return -1;
  }

  return 0;
}

int ClientCall(::llcpp::fuchsia::hardware::gpio::Gpio::SyncClient client, GpioFunc func,
               uint8_t write_value, ::llcpp::fuchsia::hardware::gpio::GpioFlags in_flag,
               uint8_t out_value, uint64_t ds_ua) {
  switch (func) {
    case Read: {
      auto result = client.Read();
      if ((result.status() != ZX_OK) || result->result.has_invalid_tag()) {
        printf("Could not read GPIO\n");
        return -2;
      }
      printf("GPIO Value: %u\n", result->result.response().value);
      break;
    }
    case Write: {
      auto result = client.Write(write_value);
      if ((result.status() != ZX_OK) || result->result.has_invalid_tag()) {
        printf("Could not write to GPIO\n");
        return -2;
      }
      break;
    }
    case ConfigIn: {
      auto result = client.ConfigIn(in_flag);
      if ((result.status() != ZX_OK) || result->result.has_invalid_tag()) {
        printf("Could not configure GPIO as input\n");
        return -2;
      }
      break;
    }
    case ConfigOut: {
      auto result = client.ConfigOut(out_value);
      if ((result.status() != ZX_OK) || result->result.has_invalid_tag()) {
        printf("Could not configure GPIO as output\n");
        return -2;
      }
      break;
    }
    case SetDriveStrength: {
      auto result = client.SetDriveStrength(ds_ua);
      if ((result.status() != ZX_OK) || result->result.has_invalid_tag()) {
        printf("Could not set GPIO drive strength\n");
        return -2;
      }
      printf("Set drive strength to %lu\n", result->result.response().actual_ds_ua);
      break;
    }
    default:
      printf("Invalid function\n\n");
      return -1;
  }
  return 0;
}