How to renew \chapter command to include endnotes (i.e., \printnotes)

Question

I'm using the pagenote package to put my endnotes at the end of every chapter. For example, see the accepted answer for this question: How does one get endnotes at the end of every chapter?

In short:

\chapter{First Chapter}
 some text
\printnotes*

How can I renew the \chapter command so that I don't have to include \printnotes* at the end of every chapter?

I don't want to make the \printnotes* command happen before the \chapter command because then I'll get notes before the first chapter (which doesn't make sense).

Answer 1

Here is a solution using the \pretocmd provided by the etoolbox package.

\documentclass{book}
\usepackage{etoolbox}
\usepackage{pagenote}
\makepagenote
\begin{document}

\chapter{Chap 1}

\pretocmd{\chapter}{\printnotes*}{}{}

\chapter{Chap 2}
\chapter{Chap 3}
\chapter{Chap 4}
\printnotes*

\end{document}

As you can see the command may be used after the initial \chapter.

Edit #1

As I've made the correction on the code above, another \printnotes* command is required at the end of the final chapter.

Edit #2

It might feel more convenient to automate the process in the preamble itself with the help of logic.

\documentclass{book}

\usepackage{etoolbox}
\usepackage{pagenote}
\makepagenote

\renewcommand*{\notedivision}{\section*{\notesname\ to chapter~\thechapter}}
\renewcommand*{\pagenotesubhead}[2]{}

\pretocmd{\chapter}{%
    \ifnum\value{chapter}>0\relax\printnotes*\fi%
}{}{}
%\pretocmd{\backmatter}{\printnotes*}{}{}
\AtEndDocument{\printnotes*}    

\begin{document}

\chapter{Chap 1}
\chapter{Chap 2}
\chapter{Chap 3}

\end{document} 

In the presence of backmatter the line \pretocmd{\backmatter}{\printnotes*}{}{} can be uncommented while \AtEndDocument may be omitted.