feat: make encoding configurable

Author: wfxr

completions/bash/code-minimap.bash

@@ -26,7 +26,7 @@ _code-minimap() {
 
     case "${cmd}" in
         code-minimap)
-            opts=" -h -H -V  --help --version --horizontal-scale --vertical-scale --padding  <FILE>  completion help"
+            opts=" -h -H -V  --help --version --horizontal-scale --vertical-scale --padding --encoding  <FILE>  completion help"
             if [[ ${cur} == -* || ${COMP_CWORD} -eq 1 ]] ; then
                 COMPREPLY=( $(compgen -W "${opts}" -- "${cur}") )
                 return 0
@@ -53,6 +53,10 @@ _code-minimap() {
                     COMPREPLY=($(compgen -f "${cur}"))
                     return 0
                     ;;
+                --encoding)
+                    COMPREPLY=($(compgen -W "UTF8 UTF8Lossy" -- "${cur}"))
+                    return 0
+                    ;;
                 *)
                     COMPREPLY=()
                     ;;

completions/elvish/code-minimap.elv

@@ -20,6 +20,7 @@ edit:completion:arg-completer[code-minimap] = [@words]{
             cand -V 'Specify vertical scale factor'
             cand --vertical-scale 'Specify vertical scale factor'
             cand --padding 'Specify padding width'
+            cand --encoding 'Specify input encoding'
             cand -h 'Prints help information'
             cand --help 'Prints help information'
             cand --version 'Prints version information'

completions/fish/code-minimap.fish

@@ -1,6 +1,7 @@
 complete -c code-minimap -n "__fish_use_subcommand" -s H -l horizontal-scale -d 'Specify horizontal scale factor'
 complete -c code-minimap -n "__fish_use_subcommand" -s V -l vertical-scale -d 'Specify vertical scale factor'
 complete -c code-minimap -n "__fish_use_subcommand" -l padding -d 'Specify padding width'
+complete -c code-minimap -n "__fish_use_subcommand" -l encoding -d 'Specify input encoding' -r -f -a "UTF8 UTF8Lossy"
 complete -c code-minimap -n "__fish_use_subcommand" -s h -l help -d 'Prints help information'
 complete -c code-minimap -n "__fish_use_subcommand" -l version -d 'Prints version information'
 complete -c code-minimap -n "__fish_use_subcommand" -f -a "completion" -d 'Generate shell completion file'

completions/powershell/_code-minimap.ps1

@@ -25,6 +25,7 @@ Register-ArgumentCompleter -Native -CommandName 'code-minimap' -ScriptBlock {
             [CompletionResult]::new('-V', 'V', [CompletionResultType]::ParameterName, 'Specify vertical scale factor')
             [CompletionResult]::new('--vertical-scale', 'vertical-scale', [CompletionResultType]::ParameterName, 'Specify vertical scale factor')
             [CompletionResult]::new('--padding', 'padding', [CompletionResultType]::ParameterName, 'Specify padding width')
+            [CompletionResult]::new('--encoding', 'encoding', [CompletionResultType]::ParameterName, 'Specify input encoding')
             [CompletionResult]::new('-h', 'h', [CompletionResultType]::ParameterName, 'Prints help information')
             [CompletionResult]::new('--help', 'help', [CompletionResultType]::ParameterName, 'Prints help information')
             [CompletionResult]::new('--version', 'version', [CompletionResultType]::ParameterName, 'Prints version information')

completions/zsh/_code-minimap

@@ -20,6 +20,7 @@ _code-minimap() {
 '-V+[Specify vertical scale factor]' \
 '--vertical-scale=[Specify vertical scale factor]' \
 '--padding=[Specify padding width]' \
+'--encoding=[Specify input encoding]: :(UTF8 UTF8Lossy)' \
 '-h[Prints help information]' \
 '--help[Prints help information]' \
 '--version[Prints version information]' \

src/bin/code-minimap/cli.rs

@@ -1,6 +1,6 @@
 use std::path::PathBuf;
 
-use structopt::clap::{self, AppSettings};
+use structopt::clap::{self, arg_enum, AppSettings};
 pub use structopt::StructOpt;
 
 #[derive(StructOpt)]
@@ -25,6 +25,10 @@ pub struct Opt {
     #[structopt(long = "padding")]
     pub padding: Option<usize>,
 
+    /// Specify input encoding
+    #[structopt(long = "encoding", default_value = "UTF8Lossy", possible_values = &Encoding::variants(), case_insensitive = true)]
+    pub encoding: Encoding,
+
     /// Subcommand
     #[structopt(subcommand)]
     pub subcommand: Option<Subcommand>,
@@ -42,3 +46,10 @@ pub struct CompletionOpt {
     #[structopt(possible_values = &clap::Shell::variants())]
     pub shell: clap::Shell,
 }
+
+arg_enum! {
+    pub enum Encoding {
+        UTF8,
+        UTF8Lossy,
+    }
+}

src/bin/code-minimap/main.rs

@@ -1,11 +1,11 @@
 mod cli;
 use std::{
     fs::File,
-    io::{self, BufRead},
+    io::{self, BufRead, BufReader, Read},
     process,
 };
 
-use cli::{CompletionOpt, Opt, StructOpt, Subcommand};
+use cli::{CompletionOpt, Encoding, Opt, StructOpt, Subcommand};
 use code_minimap::lossy_reader::LossyReader;
 
 fn main() {
@@ -28,12 +28,19 @@ fn try_main() -> anyhow::Result<()> {
         }
         None => {
             let stdin = io::stdin();
-            let reader: Box<dyn BufRead> = match &opt.file {
-                Some(path) => Box::new(LossyReader::new(File::open(path)?)),
-                None => Box::new(LossyReader::new(stdin)),
+            let reader = match &opt.file {
+                Some(path) => buf_reader(&opt.encoding, File::open(path)?),
+                None => buf_reader(&opt.encoding, stdin),
             };
             code_minimap::print(reader, opt.hscale, opt.vscale, opt.padding)?;
         }
     }
     Ok(())
 }
+
+fn buf_reader<R: 'static + Read>(encoding: &Encoding, reader: R) -> Box<dyn BufRead> {
+    match encoding {
+        Encoding::UTF8 => Box::new(BufReader::new(reader)),
+        Encoding::UTF8Lossy => Box::new(LossyReader::new(reader)),
+    }
+}