// Copyright 2017 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 <assert.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <threads.h>
#include <unistd.h>

#include <ddk/binding.h>
#include <ddk/debug.h>
#include <ddk/device.h>
#include <ddk/driver.h>
#include <ddk/platform-defs.h>
#include <ddk/protocol/i2c.h>
#include <lib/device-protocol/i2c.h>
#include <ddk/protocol/platform/device.h>

#include <zircon/process.h>
#include <zircon/syscalls.h>

typedef struct {
    zx_device_t* zdev;
    i2c_protocol_t i2c;
    thrd_t thread;
    bool done;
} i2c_test_t;

static void i2c_test_release(void* ctx) {
    i2c_test_t* bus = ctx;
    bus->done = true;
    thrd_join(bus->thread, NULL);
    free(bus);
}

static zx_protocol_device_t i2c_test_device_protocol = {
    .version = DEVICE_OPS_VERSION,
    .release = i2c_test_release,
};

static void i2c_complete(void* cookie, zx_status_t status, const i2c_op_t* ops, size_t cnt) {
    if (status != ZX_OK) {
        zxlogf(ERROR, "hikey960-i2c-test i2c_complete error: %d\n", status);
    }
    ZX_ASSERT(cnt == 1);
    if (ops[0].data_size != 8) {
        zxlogf(ERROR, "hikey960-i2c-test received %zd bytes instead of 8\n", ops[0].data_size);
    }
    const uint8_t* data = ops[0].data_buffer;
    zxlogf(INFO, "hikey-i2c-test: %02X %02X %02X %02X %02X %02X %02X %02X\n", data[0], data[1],
           data[2], data[3], data[4], data[5], data[6], data[7]);
}

static int i2c_test_thread(void* arg) {
    i2c_test_t* i2c_test = arg;

    while (!i2c_test->done) {
        char write_buf[1] = { 0x0 };
        i2c_write_read(&i2c_test->i2c, write_buf, sizeof(write_buf), 8, i2c_complete, NULL);
        sleep(1);
    }

    return ZX_OK;
}


static zx_status_t i2c_test_bind(void* ctx, zx_device_t* parent) {
    i2c_test_t* i2c_test = calloc(1, sizeof(i2c_test_t));
    if (i2c_test == NULL) {
        return ZX_ERR_NO_MEMORY;
    }

    if (device_get_protocol(parent, ZX_PROTOCOL_I2C, &i2c_test->i2c) != ZX_OK) {
        free(i2c_test);
        return ZX_ERR_NOT_SUPPORTED;
    }

    device_add_args_t args = {
        .version = DEVICE_ADD_ARGS_VERSION,
        .name = "hikey960-i2c-test",
        .ctx = i2c_test,
        .ops = &i2c_test_device_protocol,
        .flags = DEVICE_ADD_NON_BINDABLE,
    };

    zx_status_t status  = device_add(parent, &args, NULL);
    if (status != ZX_OK) {
        free(i2c_test);
        return status;
    }

    thrd_create_with_name(&i2c_test->thread, i2c_test_thread, i2c_test, "i2c_test_thread");
    return ZX_OK;
}

static zx_driver_ops_t i2c_test_driver_ops = {
    .version = DRIVER_OPS_VERSION,
    .bind = i2c_test_bind,
};

ZIRCON_DRIVER_BEGIN(hikey960_i2c_test, i2c_test_driver_ops, "zircon", "0.1", 4)
    BI_ABORT_IF(NE, BIND_PROTOCOL, ZX_PROTOCOL_PDEV),
    BI_ABORT_IF(NE, BIND_PLATFORM_DEV_VID, PDEV_VID_96BOARDS),
    BI_ABORT_IF(NE, BIND_PLATFORM_DEV_PID, PDEV_PID_HIKEY960),
    BI_MATCH_IF(EQ, BIND_PLATFORM_DEV_DID, PDEV_DID_HIKEY960_I2C_TEST),
ZIRCON_DRIVER_END(hikey960_i2c_test)