Colocate component stories with their components

crates/ui2/src/components/keybinding.rs

@@ -165,3 +165,81 @@ impl ModifierKeys {
         self
     }
 }
+
+#[cfg(feature = "stories")]
+pub use stories::*;
+
+#[cfg(feature = "stories")]
+mod stories {
+    use itertools::Itertools;
+
+    use crate::Story;
+
+    use super::*;
+
+    #[derive(Element)]
+    pub struct KeybindingStory<S: 'static + Send + Sync + Clone> {
+        state_type: PhantomData<S>,
+    }
+
+    impl<S: 'static + Send + Sync + Clone> KeybindingStory<S> {
+        pub fn new() -> Self {
+            Self {
+                state_type: PhantomData,
+            }
+        }
+
+        fn render(&mut self, cx: &mut ViewContext<S>) -> impl Element<State = S> {
+            let all_modifier_permutations = ModifierKey::iter().permutations(2);
+
+            Story::container(cx)
+                .child(Story::title_for::<_, Keybinding<S>>(cx))
+                .child(Story::label(cx, "Single Key"))
+                .child(Keybinding::new("Z".to_string(), ModifierKeys::new()))
+                .child(Story::label(cx, "Single Key with Modifier"))
+                .child(
+                    div()
+                        .flex()
+                        .gap_3()
+                        .children(ModifierKey::iter().map(|modifier| {
+                            Keybinding::new("C".to_string(), ModifierKeys::new().add(modifier))
+                        })),
+                )
+                .child(Story::label(cx, "Single Key with Modifier (Permuted)"))
+                .child(
+                    div().flex().flex_col().children(
+                        all_modifier_permutations
+                            .chunks(4)
+                            .into_iter()
+                            .map(|chunk| {
+                                div()
+                                    .flex()
+                                    .gap_4()
+                                    .py_3()
+                                    .children(chunk.map(|permutation| {
+                                        let mut modifiers = ModifierKeys::new();
+
+                                        for modifier in permutation {
+                                            modifiers = modifiers.add(modifier);
+                                        }
+
+                                        Keybinding::new("X".to_string(), modifiers)
+                                    }))
+                            }),
+                    ),
+                )
+                .child(Story::label(cx, "Single Key with All Modifiers"))
+                .child(Keybinding::new("Z".to_string(), ModifierKeys::all()))
+                .child(Story::label(cx, "Chord"))
+                .child(Keybinding::new_chord(
+                    ("A".to_string(), ModifierKeys::new()),
+                    ("Z".to_string(), ModifierKeys::new()),
+                ))
+                .child(Story::label(cx, "Chord with Modifier"))
+                .child(Keybinding::new_chord(
+                    ("A".to_string(), ModifierKeys::new().control(true)),
+                    ("Z".to_string(), ModifierKeys::new().shift(true)),
+                ))
+        }
+    }
+}