diff --git a/README.md b/README.md index e43b572..e0acb26 100644 --- a/README.md +++ b/README.md @@ -32,6 +32,7 @@ The following is shared: - `start`: (Optional) Number: Where to start. (`1` is the default, since it is the most common.) - `finish`: Number: Where to end. - `naming`: (Optional) String: How to name sections in the final output. The result is `[naming] [#]` (using section numbers). If not specified, there will be no Table of Contents. + - `automatic_naming`: (Optional) Boolean: If any line matches "Prologue" or "Chapter #" (any number), it will be made into a heading. (Note: This does not override `naming`. Both can be used together.) Example: ```json diff --git a/make-epub.lua b/make-epub.lua index 34fffd6..aee5fd1 100755 --- a/make-epub.lua +++ b/make-epub.lua @@ -223,7 +223,25 @@ local function concatenate_pages(config) for page = 1, config.page_counts[section - (config.sections.start - 1)] do local page_file, err = io.open(section_dir .. page .. ".md", "r") if not page_file then error(err) end - section_file:write(page_file:read("*a") .. "\n") -- the \n probably isn't needed, but does guarantee safety.. + if config.sections.automatic_naming then + local naming_patterns = { + "^Prologue$", + "^Chapter %d+$", + } + local line = page_file:read("*line") + while line do + for _, pattern in ipairs(naming_patterns) do + if line:find(pattern) then + line = "# " .. line + end + end + section_file:write(line .. "\n") + line = page_file:read("*line") + end + else + section_file:write(page_file:read("*a")) + end + section_file:write("\n") -- guarantees no accidental line collisions page_file:close() end end