From 42129c6181239298342622939c9cd3a42a3ba0d2 Mon Sep 17 00:00:00 2001
From: ISSOtm <eldredhabert0@gmail.com>
Date: Thu, 23 Jun 2022 00:05:56 +0200
Subject: [PATCH] Always open index page with `serve --open`

---
 src/cmd/build.rs | 16 ++++++----------
 src/cmd/serve.rs |  7 ++-----
 src/cmd/watch.rs | 11 +++++------
 src/main.rs      | 14 --------------
 4 files changed, 13 insertions(+), 35 deletions(-)

diff --git a/src/cmd/build.rs b/src/cmd/build.rs
index c0a28ec0..5fe73236 100644
--- a/src/cmd/build.rs
+++ b/src/cmd/build.rs
@@ -1,8 +1,7 @@
-use crate::{first_chapter, get_book_dir, open};
+use crate::{get_book_dir, open};
 use clap::{arg, App, Arg, ArgMatches};
 use mdbook::errors::Result;
 use mdbook::MDBook;
-use std::path::Path;
 
 // Create clap subcommand arguments
 pub fn make_subcommand<'help>() -> App<'help> {
@@ -39,15 +38,12 @@ pub fn execute(args: &ArgMatches) -> Result<()> {
 
     if args.is_present("open") {
         // FIXME: What's the right behaviour if we don't use the HTML renderer?
-        match first_chapter(&book)
-            .map(|path| book.build_dir_for("html").join(path).with_extension("html"))
-        {
-            Some(path) if Path::new(&path).exists() => open(path),
-            _ => {
-                error!("No chapter available to open");
-                std::process::exit(1)
-            }
+        let path = book.build_dir_for("html").join("index.html");
+        if !path.exists() {
+            error!("No chapter available to open");
+            std::process::exit(1)
         }
+        open(path);
     }
 
     Ok(())
diff --git a/src/cmd/serve.rs b/src/cmd/serve.rs
index 04ee556c..97ef3955 100644
--- a/src/cmd/serve.rs
+++ b/src/cmd/serve.rs
@@ -1,6 +1,6 @@
 #[cfg(feature = "watch")]
 use super::watch;
-use crate::{first_chapter, get_book_dir, open};
+use crate::{get_book_dir, open};
 use clap::{arg, App, Arg, ArgMatches};
 use futures_util::sink::SinkExt;
 use futures_util::StreamExt;
@@ -103,10 +103,7 @@ pub fn execute(args: &ArgMatches) -> Result<()> {
     });
 
     if open_browser {
-        let serving_url = match first_chapter(&book).map(|path| path.with_extension("html")) {
-            Some(path) => format!("http://{}/{}", address, path.display()),
-            _ => format!("http://{}", address),
-        };
+        let serving_url = format!("http://{}", address);
         info!("Serving on: {}", serving_url);
         open(serving_url);
     }
diff --git a/src/cmd/watch.rs b/src/cmd/watch.rs
index 2effd29f..9336af77 100644
--- a/src/cmd/watch.rs
+++ b/src/cmd/watch.rs
@@ -1,4 +1,3 @@
-use crate::first_chapter;
 use crate::{get_book_dir, open};
 use clap::{arg, App, Arg, ArgMatches};
 use mdbook::errors::Result;
@@ -46,12 +45,12 @@ pub fn execute(args: &ArgMatches) -> Result<()> {
 
     if args.is_present("open") {
         book.build()?;
-        match first_chapter(&book)
-            .map(|path| book.build_dir_for("html").join(path).with_extension("html"))
-        {
-            Some(path) if Path::new(&path).exists() => open(path),
-            _ => warn!("No chapter available to open"),
+        let path = book.build_dir_for("html").join("index.html");
+        if !path.exists() {
+            error!("No chapter available to open");
+            std::process::exit(1)
         }
+        open(path);
     }
 
     trigger_on_change(&book, |paths, book_dir| {
diff --git a/src/main.rs b/src/main.rs
index f993d475..35562e64 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -9,10 +9,7 @@ use clap::{App, AppSettings, Arg, ArgMatches};
 use clap_complete::Shell;
 use env_logger::Builder;
 use log::LevelFilter;
-use mdbook::book::Chapter;
 use mdbook::utils;
-use mdbook::BookItem;
-use mdbook::MDBook;
 use std::env;
 use std::ffi::OsStr;
 use std::io::Write;
@@ -140,17 +137,6 @@ fn get_book_dir(args: &ArgMatches) -> PathBuf {
     }
 }
 
-// Return the first displayable chapter of the given book, or None if no displayable
-// chapter is found (i.e. only drafts).
-fn first_chapter(book: &MDBook) -> Option<&PathBuf> {
-    book.iter().find_map(|item| match item {
-        BookItem::Chapter(Chapter {
-            path: Some(path), ..
-        }) => Some(path),
-        _ => None,
-    })
-}
-
 fn open<P: AsRef<OsStr>>(path: P) {
     info!("Opening web browser");
     if let Err(e) = opener::open(path) {