/*
 *  TP-LINK TL-WR940N v4 and v6 board support
 *
 *  Copyright (C) 2016 David Lutz <kpanic@ff3l.net>
 *
 *  This program is free software; you can redistribute it and/or modify it
 *  under the terms of the GNU General Public License version 2 as published
 *  by the Free Software Foundation.
 */

#include <linux/gpio.h>
#include <linux/platform_device.h>

#include <asm/mach-ath79/ath79.h>
#include <asm/mach-ath79/ar71xx_regs.h>

#include "common.h"
#include "dev-eth.h"
#include "dev-gpio-buttons.h"
#include "dev-leds-gpio.h"
#include "dev-m25p80.h"
#include "dev-wmac.h"
#include "machtypes.h"


#define TL_WR940N_V4_GPIO_LED_QSS		3
#define TL_WR940N_V4_GPIO_LED_WAN		14
#define TL_WR940N_V4_GPIO_LED_WAN_RED		15
#define TL_WR940N_V4_GPIO_LED_LAN4		4
#define TL_WR940N_V4_GPIO_LED_LAN3		18
#define TL_WR940N_V4_GPIO_LED_LAN2		6
#define TL_WR940N_V4_GPIO_LED_LAN1		8
#define TL_WR940N_V4_GPIO_LED_WLAN		7
#define TL_WR940N_V4_GPIO_LED_SYSTEM		5
/* WR940N v6 specific GPIO*/
#define TL_WR940N_V6_GPIO_LED_DIAG_ORANGE	15
#define TL_WR940N_V6_GPIO_LED_WAN_BLUE		14

#define TL_WR940N_V4_GPIO_BTN_RESET		1
#define TL_WR940N_V4_GPIO_BTN_RFKILL		2

#define TL_WR940N_KEYS_POLL_INTERVAL		20
#define TL_WR940N_KEYS_DEBOUNCE_INTERVAL	(3 * TL_WR940N_KEYS_POLL_INTERVAL)


static struct gpio_led tl_wr940n_v4_leds_gpio[] __initdata = {
	{
		.name		= "tp-link:blue:qss",
		.gpio		= TL_WR940N_V4_GPIO_LED_QSS,
		.active_low	= 1,
	},
	{
		.name		= "tp-link:blue:wan",
		.gpio		= TL_WR940N_V4_GPIO_LED_WAN,
		.active_low	= 1,
	},
	{
		.name		= "tp-link:red:wan",
		.gpio		= TL_WR940N_V4_GPIO_LED_WAN_RED,
		.active_low	= 0,
	},
	{
		.name		= "tp-link:blue:lan1",
		.gpio		= TL_WR940N_V4_GPIO_LED_LAN1,
		.active_low	= 1,
	},
	{
		.name		= "tp-link:blue:lan2",
		.gpio		= TL_WR940N_V4_GPIO_LED_LAN2,
		.active_low	= 1,
	},
	{
		.name		= "tp-link:blue:lan3",
		.gpio		= TL_WR940N_V4_GPIO_LED_LAN3,
		.active_low	= 1,
	},
	{
		.name		= "tp-link:blue:lan4",
		.gpio		= TL_WR940N_V4_GPIO_LED_LAN4,
		.active_low	= 1,
	},
	{
		.name		= "tp-link:blue:wlan",
		.gpio		= TL_WR940N_V4_GPIO_LED_WLAN,
		.active_low	= 1,
	},
	{
		.name		= "tp-link:blue:system",
		.gpio		= TL_WR940N_V4_GPIO_LED_SYSTEM,
		.active_low	= 1,
	},
};

static struct gpio_keys_button tl_wr940n_v4_gpio_keys[] __initdata = {
	{
		.desc		= "Reset button",
		.type		= EV_KEY,
		.code		= KEY_RESTART,
		.debounce_interval = TL_WR940N_KEYS_DEBOUNCE_INTERVAL,
		.gpio		= TL_WR940N_V4_GPIO_BTN_RESET,
		.active_low	= 1,
	}, {
		.desc		= "RFKILL button",
		.type		= EV_KEY,
		.code		= KEY_RFKILL,
		.debounce_interval = TL_WR940N_KEYS_DEBOUNCE_INTERVAL,
		.gpio		= TL_WR940N_V4_GPIO_BTN_RFKILL,
		.active_low	= 1,
	}
};

static struct gpio_led tl_wr940n_v6_leds_gpio[] __initdata = {
	{
		.name		= "tp-link:blue:wan",
		.gpio		= TL_WR940N_V6_GPIO_LED_WAN_BLUE,
		.active_low	= 1,
	},
	{
		.name		= "tp-link:orange:diag",
		.gpio		= TL_WR940N_V6_GPIO_LED_DIAG_ORANGE,
		.active_low	= 0,
	},
};


static const char *tl_wr940n_part_probes[] = {
	"tp-link",
	NULL,
};

static struct flash_platform_data tl_wr940n_flash_data = {
	.part_probes	= tl_wr940n_part_probes,
};


static void __init tl_wr940n_setup(void)
{
	u8 *mac = (u8 *) KSEG1ADDR(0x1f01fc00);
	u8 *ee = (u8 *) KSEG1ADDR(0x1fff1000);

	ath79_register_m25p80(&tl_wr940n_flash_data);

	ath79_register_mdio(0, 0x0);

	ath79_init_mac(ath79_eth0_data.mac_addr, mac, 1);
	ath79_init_mac(ath79_eth1_data.mac_addr, mac, 0);

	ath79_switch_data.phy4_mii_en = 1;

	ath79_register_eth(0);
	ath79_register_eth(1);

	ath79_register_wmac(ee, mac);

}

static void __init tl_wr940n_v4_setup(void)
{
	tl_wr940n_setup();

	ath79_register_leds_gpio(-1, ARRAY_SIZE(tl_wr940n_v4_leds_gpio),
				 tl_wr940n_v4_leds_gpio);

	ath79_register_gpio_keys_polled(-1, TL_WR940N_KEYS_POLL_INTERVAL,
					ARRAY_SIZE(tl_wr940n_v4_gpio_keys),
					tl_wr940n_v4_gpio_keys);
}

static void __init tl_wr940n_v6_setup(void)
{
	tl_wr940n_setup();

	ath79_register_leds_gpio(-1, ARRAY_SIZE(tl_wr940n_v6_leds_gpio),
				 tl_wr940n_v6_leds_gpio);

	ath79_register_gpio_keys_polled(-1, TL_WR940N_KEYS_POLL_INTERVAL,
					ARRAY_SIZE(tl_wr940n_v4_gpio_keys),
					tl_wr940n_v4_gpio_keys);
}

MIPS_MACHINE(ATH79_MACH_TL_WR940N_V4, "TL-WR940N-v4", "TP-LINK TL-WR940N v4",
	     tl_wr940n_v4_setup);
MIPS_MACHINE(ATH79_MACH_TL_WR940N_V6, "TL-WR940N-v6", "TP-LINK TL-WR940N v6",
	     tl_wr940n_v6_setup);