From 47c98245a30f228132e1e64469f7a1cd3886e1a3 Mon Sep 17 00:00:00 2001
From: Zoe Roux <zoe.roux@sdg.moe>
Date: Wed, 16 Nov 2022 00:04:02 +0900
Subject: [PATCH] Add pressable support on react native

---
 .eslintrc.json                            |  3 +-
 examples/expo-example/src/app.tsx         | 28 ++++++++--
 packages/yoshiki/package.json             |  9 +--
 packages/yoshiki/src/native/generator.tsx | 68 ++++++++++++++++++-----
 packages/yoshiki/src/native/hover.tsx     | 51 +++++++++++++++++
 packages/yoshiki/src/native/index.ts      |  2 +
 6 files changed, 136 insertions(+), 25 deletions(-)
 create mode 100644 packages/yoshiki/src/native/hover.tsx

diff --git a/.eslintrc.json b/.eslintrc.json
index 05beca7..d012a7c 100644
--- a/.eslintrc.json
+++ b/.eslintrc.json
@@ -17,6 +17,7 @@
 				""
 			],
 			1
-		]
+		],
+		"@typescript-eslint/ban-ts-ignore": "off"
 	}
 }
diff --git a/examples/expo-example/src/app.tsx b/examples/expo-example/src/app.tsx
index bdef548..8b6f434 100644
--- a/examples/expo-example/src/app.tsx
+++ b/examples/expo-example/src/app.tsx
@@ -6,7 +6,7 @@
 import { StatusBar } from "expo-status-bar";
 import { Text, View } from "react-native";
 import { registerRootComponent } from "expo";
-import { Stylable, useYoshiki } from "yoshiki/native";
+import { Pressable, Stylable, useYoshiki } from "yoshiki/native";
 
 const CustomBox = ({ color, ...props }: { color: string } & Stylable) => {
 	const { css } = useYoshiki();
@@ -22,9 +22,29 @@ const BoxWithoutProps = (props: Stylable) => {
 	const { css } = useYoshiki();
 
 	return (
-		<View {...css({ backgroundColor: { xs: "#00ff00", md: "#ff0000" } }, props)}>
-			<Text>Text inside the box without props (green on small screens, red on bigs)</Text>
-		</View>
+		<Pressable
+			{...css(
+				{
+					backgroundColor: { xs: "#00ff00", md: "#ff0000" },
+					hover: { alignContent: "center" },
+					press: { alignContent: "center" },
+					focus: { alignContent: "center" },
+				},
+				props,
+			)}
+		>
+			<Text
+				{...css(
+					{
+						backgroundColor: { xs: "#00ff00", md: "#ff0000" },
+						hover: { alignContent: "center" },
+					},
+					props,
+				)}
+			>
+				Text inside the box without props (green on small screens, red on bigs)
+			</Text>
+		</Pressable>
 	);
 };
 
diff --git a/packages/yoshiki/package.json b/packages/yoshiki/package.json
index fd9eee1..a264e19 100644
--- a/packages/yoshiki/package.json
+++ b/packages/yoshiki/package.json
@@ -13,14 +13,9 @@
 	"bugs": {
 		"url": "https://github.com/AnonymusRaccoon/yoshiki/issues"
 	},
-	"main": "dist/index.js",
-	"types": "dist/index.d.ts",
+	"main": "src/index.ts",
+	"types": "src/index.ts",
 	"source": "src/index.ts",
-	"exports": {
-		".": "./dist/index.ts",
-		"./native": "./dist/native/index.ts",
-		"./web": "./dist/index.ts"
-	},
 	"scripts": {
 		"build": "tsc --build ."
 	},
diff --git a/packages/yoshiki/src/native/generator.tsx b/packages/yoshiki/src/native/generator.tsx
index 6b3a371..f83a500 100644
--- a/packages/yoshiki/src/native/generator.tsx
+++ b/packages/yoshiki/src/native/generator.tsx
@@ -54,28 +54,70 @@ const propertyMapper = <
 	return [[key, value]];
 };
 
+type WithState<Style> = {
+	hover: Style;
+	focus: Style;
+	press: Style;
+};
+type AtLeastOne<T, U = {[K in keyof T]: Pick<T, K> }> = Partial<T> & U[keyof U]
+
+const hasState = <Style,>(obj: unknown): obj is WithState<Style> => {
+	if (!obj || typeof obj !== "object") return false;
+	return "hover" in obj || "focus" in obj || "press" in obj;
+};
+
 export const useYoshiki = () => {
 	const breakpoint = useBreakpoint();
 	const theme = useTheme();
 
 	return {
-		css: <Style extends ViewStyle | TextStyle | ImageStyle>(
-			css: EnhancedStyle<Style>,
+		css: <
+			Style extends ViewStyle | TextStyle | ImageStyle,
+			State extends Partial<WithState<EnhancedStyle<Style>>> | Record<string, never>,
+		>(
+			css: EnhancedStyle<Style> & State,
 			leftOvers?: { style?: Style },
-		): { style: Style } => {
+		): State extends AtLeastOne<WithState<unknown>>
+			? { style: (state: { pressed: boolean; focused: boolean; hovered: boolean }) => Style }
+			: { style: Style } => {
 			const { style, ...leftOverProps } = leftOvers ?? {};
 
-			// @ts-ignore propertyMapper always returns key that are valid for the current Style
-			const inline: Style = Object.fromEntries(
-				Object.entries(css).flatMap(([key, value]) =>
-					propertyMapper(key, value, { breakpoint, theme }),
-				),
-			);
-
-			return {
-				style: { ...inline, ...style },
-				...leftOverProps,
+			const processStyle = (styleList: EnhancedStyle<Style>): Style => {
+				// @ts-ignore propertyMapper always returns key that are valid for the current Style
+				return Object.fromEntries(
+					Object.entries(styleList).flatMap(([key, value]) =>
+						propertyMapper(key, value, { breakpoint, theme }),
+					),
+				);
 			};
+
+			if (hasState<Style>(css)) {
+				const { hover, focus, press, ...inline } = css;
+				const ret: {
+					style: (state: { hovered: boolean; focused: boolean; pressed: boolean }) => Style;
+				} = {
+					style: ({ hovered, focused, pressed }) => ({
+						// @ts-ignore EnhancedStyle<Style> is not assignable to EnhancedStyle<Style>...
+						...processStyle(inline),
+						...(hovered ? hover : {}),
+						...(focused ? focus : {}),
+						...(pressed ? press : {}),
+						...style,
+					}),
+					...leftOverProps,
+				};
+
+				// @ts-ignore Ts is not able to identify the type of the return type (the condition throws him off).
+				return ret;
+			} else {
+				const ret: { style: Style } = {
+					style: { ...processStyle(css), ...style },
+					...leftOverProps,
+				};
+
+				// @ts-ignore Ts is not able to identify the type of the return type (the condition throws him off).
+				return ret;
+			}
 		},
 		theme: theme,
 	};
diff --git a/packages/yoshiki/src/native/hover.tsx b/packages/yoshiki/src/native/hover.tsx
new file mode 100644
index 0000000..dc72dca
--- /dev/null
+++ b/packages/yoshiki/src/native/hover.tsx
@@ -0,0 +1,51 @@
+//
+// Copyright (c) Zoe Roux and contributors. All rights reserved.
+// Licensed under the MIT license. See LICENSE file in the project root for details.
+//
+
+import { useState } from "react";
+import { Pressable as RPressable, PressableProps, StyleProp, ViewStyle } from "react-native";
+
+export const Pressable = ({
+	children,
+	onHoverIn,
+	onHoverOut,
+	onFocus,
+	onBlur,
+	style,
+	...props
+}: {
+	style:
+		| PressableProps["style"]
+		| ((state: { pressed: boolean; hovered: boolean; focused: boolean }) => StyleProp<ViewStyle>);
+} & PressableProps) => {
+	const [hovered, setHover] = useState(false);
+	const [focused, setFocus] = useState(false);
+
+	return (
+		<RPressable
+			onHoverIn={(e) => {
+				onHoverIn?.call(null, e);
+				setHover(true);
+			}}
+			onHoverOut={(e) => {
+				onHoverOut?.call(null, e);
+				setHover(false);
+			}}
+			onFocus={(e) => {
+				onFocus?.call(null, e);
+				setFocus(true);
+			}}
+			onBlur={(e) => {
+				onBlur?.call(null, e);
+				setFocus(false);
+			}}
+			style={
+				typeof style === "function" ? ({ pressed }) => style({ pressed, hovered, focused }) : style
+			}
+			{...props}
+		>
+			{children}
+		</RPressable>
+	);
+};
diff --git a/packages/yoshiki/src/native/index.ts b/packages/yoshiki/src/native/index.ts
index 35bd12b..9868a34 100644
--- a/packages/yoshiki/src/native/index.ts
+++ b/packages/yoshiki/src/native/index.ts
@@ -6,3 +6,5 @@
 export { type Theme, breakpoints, useTheme, ThemeProvider } from "../theme";
 
 export { useYoshiki, type Stylable } from "./generator";
+
+export { Pressable } from "./hover";