The best books on discipline are Willpower, Mastery, Mini Habits, No Excuses, etc. To name a few.