use std::fmt::Debug;
use crate::batbelt::metadata::{
BatMetadataParser, BatMetadataType, MetadataId, SourceCodeMetadata,
};
use crate::batbelt::parser::syn_struct_classifier;
use crate::batbelt::sonar::{BatSonar, SonarResultType};
use crate::batbelt::BatEnumerator;
use error_stack::{Result, ResultExt};
use super::MetadataError;
use crate::batbelt::bat_dialoguer::BatDialoguer;
use crate::batbelt::parser::parse_formatted_path;
use colored::Colorize;
use serde::{Deserialize, Serialize};
use std::{fs, vec};
use walkdir::DirEntry;
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct StructSourceCodeMetadata {
pub path: String,
pub name: String,
pub struct_type: StructMetadataType,
pub metadata_id: String,
pub start_line_index: usize,
pub end_line_index: usize,
#[serde(default)]
pub program_name: String,
}
impl BatMetadataParser<StructMetadataType> for StructSourceCodeMetadata {
fn name(&self) -> String {
self.name.clone()
}
fn path(&self) -> String {
self.path.clone()
}
fn metadata_id(&self) -> MetadataId {
self.metadata_id.clone()
}
fn start_line_index(&self) -> usize {
self.start_line_index
}
fn end_line_index(&self) -> usize {
self.end_line_index
}
fn metadata_sub_type(&self) -> StructMetadataType {
self.struct_type
}
fn get_bat_metadata_type() -> BatMetadataType {
BatMetadataType::Struct
}
fn metadata_name() -> String {
"Struct".to_string()
}
fn new(
path: String,
name: String,
metadata_sub_type: StructMetadataType,
start_line_index: usize,
end_line_index: usize,
metadata_id: MetadataId,
) -> Self {
use crate::batbelt::metadata::derive_program_name_from_path;
let program_name = derive_program_name_from_path(&path);
StructSourceCodeMetadata {
path,
name,
metadata_id,
struct_type: metadata_sub_type,
start_line_index,
end_line_index,
program_name,
}
}
fn create_metadata_from_dir_entry(entry: DirEntry) -> Result<Vec<Self>, MetadataError> {
let entry_path = entry.path().to_str().unwrap().to_string();
let file_content = fs::read_to_string(entry.path()).unwrap();
let classification = syn_struct_classifier::classify_file(&file_content);
let bat_sonar = BatSonar::new_scanned(&file_content, SonarResultType::Struct);
let mut metadata_result = vec![];
for result in bat_sonar.results {
let struct_type = if classification.solana_account_names.contains(&result.name) {
StructMetadataType::SolanaAccount
} else if classification.context_accounts_names.contains(&result.name) {
StructMetadataType::ContextAccounts
} else if classification
.pinocchio_context_accounts_names
.contains(&result.name)
{
StructMetadataType::ContextAccounts
} else {
StructMetadataType::Other
};
let struct_metadata = StructSourceCodeMetadata::new(
entry_path.clone(),
result.name.to_string(),
struct_type,
result.start_line_index + 1,
result.end_line_index + 1,
Self::create_metadata_id(),
);
metadata_result.push(struct_metadata);
}
Ok(metadata_result)
}
}
impl StructSourceCodeMetadata {
pub fn create_metadata_from_content(
entry_path: &str,
file_content: &str,
) -> Result<Vec<Self>, MetadataError> {
let classification = syn_struct_classifier::classify_file(file_content);
let bat_sonar = BatSonar::new_scanned(file_content, SonarResultType::Struct);
let mut metadata_result = vec![];
for result in bat_sonar.results {
let struct_type = if classification.solana_account_names.contains(&result.name) {
StructMetadataType::SolanaAccount
} else if classification.context_accounts_names.contains(&result.name) {
StructMetadataType::ContextAccounts
} else if classification
.pinocchio_context_accounts_names
.contains(&result.name)
{
StructMetadataType::ContextAccounts
} else {
StructMetadataType::Other
};
let struct_metadata = StructSourceCodeMetadata::new(
entry_path.to_string(),
result.name.to_string(),
struct_type,
result.start_line_index + 1,
result.end_line_index + 1,
Self::create_metadata_id(),
);
metadata_result.push(struct_metadata);
}
Ok(metadata_result)
}
pub fn prompt_selection() -> Result<Self, MetadataError> {
let (metadata_vec, metadata_names) = Self::prompt_types()?;
let prompt_text = format!("Please select the {}:", Self::metadata_name().blue());
let selection = BatDialoguer::select(prompt_text, metadata_names, None)
.change_context(MetadataError)?;
Ok(metadata_vec[selection].clone())
}
pub fn prompt_multiselection(
select_all: bool,
force_select: bool,
) -> Result<Vec<Self>, MetadataError> {
let (metadata_vec, metadata_names) = Self::prompt_types()?;
let prompt_text = format!("Please select the {}:", Self::metadata_name().blue());
let selections = BatDialoguer::multiselect(
prompt_text,
metadata_names.clone(),
Some(&vec![select_all; metadata_names.len()]),
force_select,
)
.change_context(MetadataError)?;
let filtered_vec = metadata_vec
.into_iter()
.enumerate()
.filter_map(|(sc_index, sc_metadata)| {
if selections.iter().any(|selection| &sc_index == selection) {
Some(sc_metadata)
} else {
None
}
})
.collect::<Vec<_>>();
Ok(filtered_vec)
}
pub fn prompt_types() -> Result<(Vec<Self>, Vec<String>), MetadataError> {
let prompt_text = format!(
"Please select the {} {}:",
Self::metadata_name().blue(),
"type".blue()
);
let selection = BatDialoguer::select(
prompt_text,
StructMetadataType::get_colorized_type_vec(true),
None,
)
.change_context(MetadataError)?;
let selected_sub_type = StructMetadataType::get_type_vec()[selection];
let metadata_vec_filtered =
SourceCodeMetadata::get_filtered_structs(None, Some(selected_sub_type))
.change_context(MetadataError)?;
let metadata_names = metadata_vec_filtered
.iter()
.map(|metadata| {
parse_formatted_path(
metadata.name(),
metadata.path(),
metadata.start_line_index(),
)
})
.collect::<Vec<_>>();
Ok((metadata_vec_filtered, metadata_names))
}
}
#[derive(
Debug,
PartialEq,
Clone,
Copy,
strum_macros::Display,
strum_macros::EnumIter,
Serialize,
Deserialize,
)]
pub enum StructMetadataType {
ContextAccounts,
SolanaAccount,
Other,
}
impl BatEnumerator for StructMetadataType {}