goran 0.1.0

Goran is a CLI tool for analyzing domains and IP addresses.
Documentation 
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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83

use anyhow::{Context, Result};
use reqwest::Client;
use serde::Deserialize;
use std::fs;

#[derive(Deserialize)]
struct GeminiResponse {
  candidates: Vec<Candidate>,
}
#[derive(Deserialize)]
struct Candidate {
  content: Content,
}
#[derive(Deserialize)]
struct Content {
  parts: Vec<Part>,
}
#[derive(Deserialize)]
struct Part {
  text: Option<String>,
}

/// Generates a report using the Gemini AI model based on the provided analysis findings.
///
/// This function takes the analysis results, serializes them, combines them with a base prompt,
/// sends the combined text to the Gemini API, and returns the generated report text.
///
/// # Arguments
///
/// * `analysis` - A reference to the `Analysis` struct containing the findings to report on.
/// * `api_key` - The API key for accessing the Gemini API.
/// * `client` - A `reqwest::Client` instance to use for the HTTP request.
///
/// # Errors
///
/// This function can return an error in several cases:
/// - If the `analysis` data cannot be serialized to JSON.
/// - If the base prompt file (`src/config/prompt.txt`) cannot be read.
/// - If the HTTP request to the Gemini API fails (e.g., network issues).
/// - If the Gemini API returns an HTTP error status code.
/// - If the response from the Gemini API is not valid JSON or cannot be deserialized.
pub async fn generate_report(
  analysis: &crate::results::Analysis,
  api_key: &str,
  client: &Client,
) -> Result<String> {
  let summary_json = serde_json::to_value(analysis)?;

  let base_prompt_template = fs::read_to_string("src/config/prompt.txt")
    .context("Failed to read prompt file")?;
  let instruction = base_prompt_template
    .replace("__JSON_DATA_PLACEHOLDER__", &summary_json.to_string());

  let url = format!(
        "https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:generateContent?key={api_key}"
    );
  let body = serde_json::json!({
      "contents": [{
          "parts": [{ "text": instruction }],
          "role": "user"
      }]
  });

  let resp = client
    .post(&url)
    .json(&body)
    .send()
    .await
    .context("Failed to call Gemini")?
    .error_for_status()
    .context("Gemini returned HTTP error")?;

  let resp_json: GeminiResponse = resp.json().await.context("Invalid JSON")?;
  let answer = resp_json
    .candidates
    .first()
    .and_then(|c| c.content.parts.first())
    .and_then(|p| p.text.as_ref())
    .cloned()
    .unwrap_or_else(|| "Gemini returned an empty response.".to_string());

  Ok(answer)
}