add osmand sqlite output format (lib & cli)

1 files changed, 25 insertions, 11 deletions

diff --git a/hittekaart-cli/src/main.rs b/hittekaart-cli/src/main.rs
index 11133af..880ea61 100644
--- a/hittekaart-cli/src/main.rs
+++ b/hittekaart-cli/src/main.rs
@@ -7,8 +7,8 @@ use color_eyre::{
 };
 use hittekaart::{
     gpx::{self, Compression},
-    renderer::{self, heatmap, marktile, tilehunt, Renderer},
-    storage::{Folder, Sqlite, Storage},
+    renderer::{self, Renderer, heatmap, marktile, tilehunt},
+    storage::{Folder, Sqlite, Storage, TableFormat},
 };
 use indicatif::{MultiProgress, ProgressBar, ProgressDrawTarget, ProgressStyle};
 use is_terminal::IsTerminal;
@@ -25,6 +25,14 @@ enum Mode {
     Tilehunter,
 }
 
+/// Output format.
+#[derive(ValueEnum, Debug, Clone, Copy, PartialEq, Eq, Hash)]
+enum Format {
+    Folder,
+    Sqlite,
+    OsmAnd,
+}
+
 #[derive(Parser, Debug, Clone)]
 #[command(author, version, about)]
 struct Args {
@@ -49,10 +57,9 @@ struct Args {
     #[arg(long, short)]
     output: Option<PathBuf>,
 
-    /// Store the tiles in a SQLite database. If given, `--output` will determine the SQLite
-    /// filename.
+    /// Output format.
     #[arg(long)]
-    sqlite: bool,
+    format: Format,
 
     /// Generation mode.
     #[arg(value_enum, long, short, default_value_t = Mode::Heatmap)]
@@ -115,12 +122,19 @@ fn run<R: Renderer>(renderer: R, args: Args) -> Result<()> {
     let tracks = tracks.into_iter().collect::<Result<Vec<_>>>()?;
     bar.finish();
 
-    let mut storage: Box<dyn Storage + Send> = if args.sqlite {
-        let output = args.output.unwrap_or_else(|| "tiles.sqlite".into());
-        Box::new(Sqlite::connect(output)?)
-    } else {
-        let output = args.output.unwrap_or_else(|| "tiles".into());
-        Box::new(Folder::new(output))
+    let mut storage: Box<dyn Storage + Send> = match args.format {
+        Format::Folder => {
+            let output = args.output.unwrap_or_else(|| "tiles".into());
+            Box::new(Folder::new(output))
+        }
+        Format::Sqlite => {
+            let output = args.output.unwrap_or_else(|| "tiles.sqlite".into());
+            Box::new(Sqlite::connect(output, TableFormat::Simple)?)
+        }
+        Format::OsmAnd => {
+            let output = args.output.unwrap_or_else(|| "tiles.sqlitedb".into());
+            Box::new(Sqlite::connect(output, TableFormat::OsmAnd)?)
+        }
     };
     storage.prepare()?;