biors 0.57.4

Command-line tools for bio-rs biological AI model input workflows.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68

use crate::cli::{MoleculeCommand, MoleculeFormatArg};
use crate::errors::CliError;
use crate::input::open_buffered_input;
use crate::output::print_success;
use biors_core::molecule::{
    parse_mol2_records_reader, parse_sdf_records_reader, parse_smiles_records_reader,
    validate_molecule_records, validate_smiles_reader_with_hash,
};
use std::path::PathBuf;

pub(super) fn run_molecule_command(command: MoleculeCommand) -> Result<(), CliError> {
    match command {
        MoleculeCommand::Validate { format, path } => run_molecule_validation(format, path),
        MoleculeCommand::Inspect { format, path } => run_molecule_inspect(format, path),
    }
}

fn run_molecule_validation(format: MoleculeFormatArg, path: PathBuf) -> Result<(), CliError> {
    match format {
        MoleculeFormatArg::Smiles => {
            let reader = open_buffered_input(&path)?;
            let output = validate_smiles_reader_with_hash(reader)
                .map_err(|error| CliError::from_molecule_read(path, error))?;
            print_success(Some(output.input_hash), output.report)
        }
        MoleculeFormatArg::Sdf => {
            let reader = open_buffered_input(&path)?;
            let output = parse_sdf_records_reader(reader)
                .map_err(|error| CliError::from_molecule_read(path, error))?;
            print_success(
                Some(output.input_hash),
                validate_molecule_records(&output.records),
            )
        }
        MoleculeFormatArg::Mol2 => {
            let reader = open_buffered_input(&path)?;
            let output = parse_mol2_records_reader(reader)
                .map_err(|error| CliError::from_molecule_read(path, error))?;
            print_success(
                Some(output.input_hash),
                validate_molecule_records(&output.records),
            )
        }
    }
}

fn run_molecule_inspect(format: MoleculeFormatArg, path: PathBuf) -> Result<(), CliError> {
    match format {
        MoleculeFormatArg::Smiles => {
            let reader = open_buffered_input(&path)?;
            let output = parse_smiles_records_reader(reader)
                .map_err(|error| CliError::from_molecule_read(path, error))?;
            print_success(Some(output.input_hash), output.records)
        }
        MoleculeFormatArg::Sdf => {
            let reader = open_buffered_input(&path)?;
            let output = parse_sdf_records_reader(reader)
                .map_err(|error| CliError::from_molecule_read(path, error))?;
            print_success(Some(output.input_hash), output.records)
        }
        MoleculeFormatArg::Mol2 => {
            let reader = open_buffered_input(&path)?;
            let output = parse_mol2_records_reader(reader)
                .map_err(|error| CliError::from_molecule_read(path, error))?;
            print_success(Some(output.input_hash), output.records)
        }
    }
}