main.go

package main

import (
	"fmt"
	"funfunviz/database"
	"funfunviz/sec"
	"log"
)

func main() {
	db := database.InitDB("funfunviz.db")
	defer db.Close()

	fmt.Println("Database initialized.")

	// Clear the database to avoid duplicates
	if err := database.ClearStatements(db); err != nil {
		log.Fatal(err)
	}

	// Fetch company facts for Apple
	facts, err := sec.GetCompanyFacts("320193")
	if err != nil {
		log.Fatal(err)
	}

	// Insert the facts into the database
	if err := database.InsertFacts(db, facts); err != nil {
		log.Fatal(err)
	}
	fmt.Printf("Successfully fetched and stored data for %s\n", facts.EntityName)

	// Query and display the financials
	financials, err := database.QueryFinancials(db)
	if err != nil {
		log.Fatal(err)
	}

	fmt.Println("\n--- Stored Financials ---")
	for _, f := range financials {
		fmt.Printf("Company: %s (CIK: %s)\n", f.Company, f.CIK)
		fmt.Printf("  Revenue: %.2f\n", f.Revenue)
		fmt.Printf("  Net Income: %.2f\n", f.NetIncome)
		fmt.Printf("  Operating Expenses: %.2f\n", f.OperatingExpenses)
		fmt.Printf("  Cost of Goods Sold: %.2f\n", f.COGS)
	}
	fmt.Println("-------------------------")
}