// Copyright (c) 2020 TUXEDO Computers GmbH <tux@tuxedocomputers.com>
//
// This file is part of TUXEDO Touchpad Switch.
//
// This file is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// TUXEDO Touchpad Switch is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with TUXEDO Touchpad Switch.  If not, see <https://www.gnu.org/licenses/>.

#include "touchpad-control.h"

#include <iostream>
#include <vector>
#include <string>

#include <cstring>

#include <fcntl.h>
#include <unistd.h>
#include <sys/ioctl.h>
#include <linux/hidraw.h>

#include <libudev.h>

using std::cout;
using std::cerr;
using std::endl;

// "std::vector<std::string> *devnodes" gets amended with found "/dev/hidraw*" device paths
// returns -EXIT_FAILURE on error or the number of found "i2c-UNIW0001:00"-touchpads, starting with 0, meaning no touchpads found
static int get_touchpad_hidraw_devices(std::vector<std::string> *devnodes) {
    int result = -EXIT_FAILURE;
    
    struct udev *udev_context = udev_new();
    if (!udev_context) {
        cerr << "get_touchpad_hidraw_devices(...): udev_new(...) failed." << endl;
    }
    else {
        struct udev_enumerate *hidraw_devices = udev_enumerate_new(udev_context);
        if (!hidraw_devices) {
            cerr << "get_touchpad_hidraw_devices(...): udev_enumerate_new(...) failed." << endl;
        }
        else {
            if (udev_enumerate_add_match_subsystem(hidraw_devices, "hidraw") < 0) {
                cerr << "get_touchpad_hidraw_devices(...): udev_enumerate_add_match_subsystem(...) failed." << endl;
            }
            else {
                if (udev_enumerate_scan_devices(hidraw_devices) < 0) {
                    cerr << "get_touchpad_hidraw_devices(...): udev_enumerate_scan_devices(...) failed." << endl;
                }
                else {
                    struct udev_list_entry *hidraw_devices_iterator = udev_enumerate_get_list_entry(hidraw_devices);
                    if (!hidraw_devices_iterator) {
                        cerr << "get_touchpad_hidraw_devices(...): udev_enumerate_get_list_entry(...) failed." << endl;
                    }
                    else {
                        struct udev_list_entry *hidraw_device_entry;
                        udev_list_entry_foreach(hidraw_device_entry, hidraw_devices_iterator) {
                            if (strstr(udev_list_entry_get_name(hidraw_device_entry), "i2c-UNIW0001:00")) {
                                struct udev_device *hidraw_device = udev_device_new_from_syspath(udev_context, udev_list_entry_get_name(hidraw_device_entry));
                                if (!hidraw_device) {
                                    cerr << "get_touchpad_hidraw_devices(...): udev_device_new_from_syspath(...) failed." << endl;
                                }
                                else {
                                    std::string devnode = udev_device_get_devnode(hidraw_device);
                                    devnodes->push_back(devnode);
                                    
                                    udev_device_unref(hidraw_device);
                                }
                            }
                        }
                        
                        result = devnodes->size();
                    }
                }
            }
            
            udev_enumerate_unref(hidraw_devices);
        }
        
        udev_unref(udev_context);
    }

    return result;
}

int set_touchpad_state(int enabled) {
    std::vector<std::string> devnodes;
    int touchpad_count = get_touchpad_hidraw_devices(&devnodes);
    if (touchpad_count < 0) {
        cerr << "send_events_handler(...): get_touchpad_hidraw_devices(...) failed." << endl;
        return EXIT_FAILURE;
    }
    if (touchpad_count == 0) {
        cout << "No compatible touchpads found." << endl;
        return EXIT_FAILURE;
    }
    
    int result = EXIT_SUCCESS;
    
    for (auto it = devnodes.begin(); it != devnodes.end(); ++it) {
        int hidraw = open((*it).c_str(), O_WRONLY|O_NONBLOCK);
        if (hidraw < 0) {
            cerr << "send_events_handler(...): open(\"" << *it << "\", O_WRONLY|O_NONBLOCK) failed." << endl;
            result = EXIT_FAILURE;
        }
        else {
            // The two rightmost bits control the touchpad status
            // In order, they are:
            // 1. LED on, touchpad off
            // 2. clicks are off
            // so, the options are:
            // 0x00 LED on, touchpad off, touchpad click off
            // 0x01 LED off, touchpad on, touchpad click off
            // 0x03 LED off, touchpad on, touchpad click on
            char buffer[2] = {0x07, 0x00};
            if (enabled) {
                buffer[1] = 0x03;
            }
            
            int result = ioctl(hidraw, HIDIOCSFEATURE(sizeof(buffer)/sizeof(buffer[0])), buffer);
            if (result < 0) {
                cerr << "send_events_handler(...): ioctl(...) on " << *it << " failed." << endl;
                result = EXIT_FAILURE;
            }
            // else {}
            
            close(hidraw);
        }
    }
    
    return result;
}