From 4dcf3dde286bb2602f2be96077b02d0947353e18 Mon Sep 17 00:00:00 2001
From: nsfisis <nsfisis@gmail.com>
Date: Sat, 19 Jun 2021 21:09:57 +0900
Subject: Add '--prefix' option

---
 src/lib.rs | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

(limited to 'src/lib.rs')

diff --git a/src/lib.rs b/src/lib.rs
index db8020e..330cc87 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -13,9 +13,16 @@ impl Model {
         Model { chars }
     }
 
-    pub fn generate<Rng: RngCore>(&self, rng: &mut Rng, len: usize) -> String {
+    pub fn generate<Rng: RngCore>(
+        &self,
+        rng: &mut Rng,
+        len: usize,
+        first_char: Option<char>,
+    ) -> String {
         let mut result = String::with_capacity(len);
-        let mut prefix = NUM_OF_LETTERS;
+        let mut prefix = first_char
+            .map(|c| c as usize - b'a' as usize)
+            .unwrap_or(NUM_OF_LETTERS);
         for _ in 0..len {
             let chars = &self.chars[prefix];
             let c = select_one_char(rng, &chars);
-- 
cgit v1.2.3-70-g09d2