feat: add get_extension

xhyrom · Aug 4, 2024 · 8f5d0c7 · 8f5d0c7
1 parent 9495faa
commit 8f5d0c7
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/lsp/src/main.rs b/lsp/src/main.rs
@@ -17,6 +17,7 @@
  * along with this program.  If not, see <http://www.gnu.org/licenses/>
  */
 
+use std::ffi::OsStr;
 use std::fmt::Debug;
 use std::path::{Path, PathBuf};
 use std::sync::{Mutex, MutexGuard};
@@ -59,6 +60,15 @@ impl Document {
     fn get_filename(&self) -> &str {
         return self.path.file_name().unwrap().to_str().unwrap();
     }
+
+    fn get_extension(&self) -> &str {
+        return self
+            .path
+            .extension()
+            .unwrap_or(OsStr::new(""))
+            .to_str()
+            .unwrap();
+    }
 }
 
 impl Backend {