jupyter notebook support #151

Author: o2sh

resources/jupyter.ascii

@@ -0,0 +1,28 @@
+{1}                                                                             
+{1}                     ***************            {2}.--.  
+{1}                 ***********************       {2}/    \
+{1}       {2} _     {1}*****************************    {2}\    /
+{1}       {2}(_)  {1}*********************************   {2}'--'
+{1}          ********                     ********
+{1}        *****                               *****
+{1}       ***                                     ***
+{1}      *                             {0} _            {1}*
+{0}                                    | |                
+{0}         _   _   _   _ __    _   _  | |_    ___   _ __ 
+{0}        | | | | | | | '_ \  | | | | | __|  / _ \ | '__|
+{0}        | | | |_| | | |_) | | |_| | | |_  |  __/ | |   
+{0}    _   | |  \___,| | .__/   \__, |  \__|  \___| |_|   
+{0}   | |__| |         | |       __/ |                    
+{0}    \____/          |_|      |___/                     
+{1}        
+{1}      *                                           *
+{1}       ***                                     ***
+{1}        *****                               *****
+{1}          ********                     ********          
+{1}           ***********************************
+{1}             *******************************
+{1}                ************************
+{2}           .--.      {1}***************
+{2}          /    \
+{2}          \    /
+{2}           '--'

src/language.rs

@@ -38,6 +38,8 @@ pub enum Language {
     Java,
     JavaScript,
     Julia,
+    #[strum(serialize = "jupyter-notebooks")]
+    Jupyter,
     Kotlin,
     Lisp,
     Lua,
@@ -99,6 +101,7 @@ impl std::fmt::Display for Language {
             Language::Java => write!(f, "Java"),
             Language::JavaScript => write!(f, "JavaScript"),
             Language::Julia => write!(f, "Julia"),
+            Language::Jupyter => write!(f, "Jupyter-Notebooks"),
             Language::Kotlin => write!(f, "Kotlin"),
             Language::Lisp => write!(f, "Lisp"),
             Language::Lua => write!(f, "Lua"),
@@ -162,6 +165,7 @@ impl From<tokei::LanguageType> for Language {
             tokei::LanguageType::Java => Language::Java,
             tokei::LanguageType::JavaScript => Language::JavaScript,
             tokei::LanguageType::Julia => Language::Julia,
+            tokei::LanguageType::Jupyter => Language::Jupyter,
             tokei::LanguageType::Kotlin => Language::Kotlin,
             tokei::LanguageType::Lisp => Language::Lisp,
             tokei::LanguageType::Lua => Language::Lua,
@@ -224,6 +228,7 @@ impl Language {
             Language::Java => include_str!("../resources/java.ascii"),
             Language::JavaScript => include_str!("../resources/javascript.ascii"),
             Language::Julia => include_str!("../resources/julia.ascii"),
+            Language::Jupyter => include_str!("../resources/jupyter.ascii"),
             Language::Kotlin => include_str!("../resources/kotlin.ascii"),
             Language::Lisp => include_str!("../resources/lisp.ascii"),
             Language::Lua => include_str!("../resources/lua.ascii"),
@@ -302,6 +307,7 @@ impl Language {
                 Color::Red,
                 Color::BrightMagenta,
             ],
+            Language::Jupyter => vec![Color::BrightWhite, Color::Yellow, Color::White],
             Language::Kotlin => vec![Color::Blue, Color::Yellow, Color::Magenta],
             Language::Lisp => vec![Color::Yellow],
             Language::Lua => vec![Color::Blue],
@@ -421,6 +427,7 @@ fn get_all_language_types() -> Vec<tokei::LanguageType> {
         tokei::LanguageType::Java,
         tokei::LanguageType::JavaScript,
         tokei::LanguageType::Julia,
+        tokei::LanguageType::Jupyter,
         tokei::LanguageType::Kotlin,
         tokei::LanguageType::Lisp,
         tokei::LanguageType::Lua,