added lei to isin mapping

src/corporate/storage.rs

@@ -111,8 +111,8 @@ pub async fn load_companies() -> Result<Vec<CompanyMetadata>, anyhow::Error> {
     Ok(companies)
 }
 
-pub fn get_company_dir(isin: &str) -> PathBuf {
-    PathBuf::from("corporate_prices").join(isin)
+pub fn get_company_dir(lei: &str) -> PathBuf {
+    PathBuf::from("corporate_prices").join(lei)
 }
 
 pub async fn ensure_company_dirs(isin: &str) -> anyhow::Result<()> {
@@ -131,13 +131,19 @@ pub async fn ensure_company_dirs(isin: &str) -> anyhow::Result<()> {
 }
 
 pub async fn save_company_metadata(company: &CompanyMetadata) -> anyhow::Result<()> {
-    let dir = get_company_dir(&company.isin);
+    let dir = get_company_dir(&company.lei);
     fs::create_dir_all(&dir).await?;
     let path = dir.join("metadata.json");
-    fs::write(path, serde_json::to_string_pretty(company)?).await?;
+    fs::write(&path, serde_json::to_string_pretty(company)?).await?;
     Ok(())
 }
 
+pub async fn load_company_metadata(lei: &str) -> anyhow::Result<CompanyMetadata> {
+    let path = get_company_dir(lei).join("metadata.json");
+    let content = fs::read_to_string(path).await?;
+    Ok(serde_json::from_str(&content)?)
+}
+
 pub async fn save_available_exchanges(isin: &str, exchanges: Vec<AvailableExchange>) -> anyhow::Result<()> {
     let dir = get_company_dir(isin);
     fs::create_dir_all(&dir).await?;
@@ -146,8 +152,8 @@ pub async fn save_available_exchanges(isin: &str, exchanges: Vec<AvailableExchan
     Ok(())
 }
 
-pub async fn load_available_exchanges(isin: &str) -> anyhow::Result<Vec<AvailableExchange>> {
-    let path = get_company_dir(isin).join("available_exchanges.json");
+pub async fn load_available_exchanges(lei: &str) -> anyhow::Result<Vec<AvailableExchange>> {
+    let path = get_company_dir(lei).join("available_exchanges.json");
     if path.exists() {
         let content = fs::read_to_string(&path).await?;
         Ok(serde_json::from_str(&content)?)
@@ -157,19 +163,17 @@ pub async fn load_available_exchanges(isin: &str) -> anyhow::Result<Vec<Availabl
 }
 
 pub async fn save_prices_by_source(
-    isin: &str,
+    lei: &str,
     source_ticker: &str,
     timeframe: &str,
     prices: Vec<CompanyPrice>,
 ) -> anyhow::Result<()> {
     let source_safe = source_ticker.replace(".", "_").replace("/", "_");
-    let dir = get_company_dir(isin).join(timeframe).join(&source_safe);
+    let dir = get_company_dir(lei).join(timeframe).join(&source_safe);
     fs::create_dir_all(&dir).await?;
     let path = dir.join("prices.json");
-
     let mut prices = prices;
     prices.sort_by_key(|p| (p.date.clone(), p.time.clone()));
-
     fs::write(&path, serde_json::to_string_pretty(&prices)?).await?;
     Ok(())
 }