Support function-level states

demo.hxml

@@ -4,3 +4,5 @@
 
 -lib coconut.storybook
 -lib coconut.vdom
+
+-D analyzer-optimize

extraParams.hxml

@@ -0,0 +1 @@
+--macro coconut.storybook.Setup.setup()

haxe_libraries/coconut.storybook.hxml

@@ -1,4 +1,5 @@
 -lib coconut.ui
--cp ${SCOPE_DIR}/./src
+-cp ${SCOPE_DIR}/../coconut.storybook/src
 -D coconut.storybook=0.0.0
+${SCOPE_DIR}/../coconut.storybook/extraParams.hxml
 --macro Sys.println("haxe_libraries/coconut.storybook.hxml:3: [Warning] Using dev version of library coconut.storybook")

src/coconut/storybook/Component.macro.hx

@@ -1,14 +1,20 @@
 package coconut.storybook;
 
 import haxe.macro.Context;
+import haxe.macro.Expr;
 
 using tink.MacroApi;
 
 class Component {
 	public static function build() {
 		var builder = new ClassBuilder();
+
 		if (!builder.target.meta.has(':tink'))
 			builder.target.meta.add(':tink', [], Context.currentPos());
+
+		if (!builder.target.meta.has(':storybook'))
+			builder.target.meta.add(':storybook', [], Context.currentPos());
+
 		return null;
 	}
 

src/coconut/storybook/Setup.hx

@@ -0,0 +1,103 @@
+package coconut.storybook;
+
+import haxe.macro.Context;
+import haxe.macro.Expr;
+import tink.SyntaxHub;
+
+using StringTools;
+using tink.MacroApi;
+
+class Setup {
+	public static function setup() {
+		var debug = false;
+		SyntaxHub.classLevel.after('tink.lang.Sugar', builder -> {
+			if (!builder.target.meta.has(':tink'))
+				return false;
+
+			for (field in builder)
+				switch [field.kind, field.metaNamed(':story'), field.metaNamed(':state')] {
+					case [FFun(func), stories, v] if (stories.length > 0):
+						// wrap with Isolated
+						function subst(e:Expr)
+							return switch e {
+								case macro return $ret:
+									macro return coconut.ui.Isolated.fromHxx({}, {children: $ret});
+								case e:
+									e;
+							}
+						func.expr = func.expr.map(subst);
+
+						// add states
+						for (states in v)
+							for (state in states.params)
+								switch state.expr {
+									case EVars(vars):
+										for (v in vars) {
+											if (v.expr == null)
+												state.pos.error('@:state var ${v.name} requires a initializer');
+											if (v.type == null)
+												state.pos.error('@:state var ${v.name} requires a type hint');
+
+											var name = v.name;
+											var alias = getAlias(name);
+											var ct = v.type;
+
+											var original = func.expr;
+											func.expr = macro {
+												var $name = new tink.state.State<$ct>(${v.expr});
+												var $alias = $i{name};
+												${func.expr}
+											}
+										}
+
+									case _:
+										state.pos.error('Only supports EVars expressions');
+								}
+
+					case _:
+						// skip
+				}
+
+			return true;
+		});
+
+		SyntaxHub.exprLevel.inward.after(_ -> true, {
+			appliesTo: builder -> builder.target.meta.has(':storybook'),
+			apply: (e:Expr) -> {
+				function transform(name:String, original:Expr, transformed:Expr) {
+					return switch Context.getLocalTVars()[name] {
+						case null:
+							original;
+						case {t: _.getID() => 'tink.state.State'}:
+							transformed;
+						case _:
+							original;
+					}
+				}
+
+				switch e.expr {
+					case EConst(CIdent(name)) if (!isAlias(name)):
+						transform(name, e, macro $i{getAlias(name)}.value);
+
+					case EBinop(OpAssign, macro $i{name}, rhs) if (!isAlias(name)):
+						transform(name, e, macro $i{getAlias(name)}.set($rhs));
+
+					case EBinop(OpAssignOp(binop), macro $i{name}, rhs) if (!isAlias(name)):
+						var rhs = EBinop(binop, macro $i{getAlias(name)}.value, rhs).at(e.pos);
+						transform(name, e, macro $i{getAlias(name)}.set($rhs));
+
+					case _:
+						e;
+				}
+			}
+		});
+	}
+
+	inline static function getAlias(name:String) {
+		return '__alias_${name}';
+	}
+
+	inline static function isAlias(name:String) {
+		return name.startsWith('__alias_');
+	}
+}

src/coconut/storybook/Storybook.macro.hx

@@ -7,6 +7,12 @@ using tink.MacroApi;
 
 class Storybook {
 	public static macro function add(exprs:Array<Expr>):Expr {
+		switch exprs {
+			case [{expr: EArrayDecl(values)}]:
+				exprs = values;
+			case _:
+		}
+
 		var ret = [];
 
 		for (expr in exprs) {

tests/Demo.hx

@@ -5,7 +5,12 @@ import coconut.ui.*;
 
 class Demo {
 	static function main() {
-		Storybook.add(new Button(), new foo.Bar());
+		Storybook.add([
+			// @formatter:off
+			new Button(),
+			new foo.Bar(),
+			// @formatter:on
+		]);
 	}
 }
 
@@ -31,6 +36,14 @@ class Button extends Component {
 		</button>
 	';
 
+	@:story
+	@:state(var value:Int = 0)
+	function withState() '
+		<button onclick=${value += 1}>
+			Clicked ${value} time(s)
+		</button>
+	';
+
 	function wrap(f:()->RenderResult) '
 		<div style=${{backgroundColor: 'black'}}>${f()}</div>
 	';
@@ -43,3 +56,10 @@ extern class Knobs {
 	static function boolean(name:String, value:Bool):Bool;
 	static function text(name:String, value:String):String;
 }
+
+class Isolated extends View {
+	@:attribute var children:Children;
+
+	function render()
+		'<>${...children}</>';
+}