changes to model

internal/shell/company.go

@@ -3,7 +3,6 @@ package shell
 import (
 	"Portifolio/internal/database"
 	"Portifolio/internal/model"
-	"Portifolio/internal/service"
 	"bufio"
 	"database/sql"
 	"fmt"
@@ -16,9 +15,9 @@ import (
 func AddCompany(scanner *bufio.Scanner, db *sql.DB) {
 	input := model.CompanyInput{}
 
-	fmt.Print("  Name: ")
+	fmt.Print("  symbol: ")
 	scanner.Scan()
-	input.Name = strings.TrimSpace(scanner.Text())
+	input.Symbol = strings.TrimSpace(scanner.Text())
 
 	fmt.Print("  Shares outstanding: ")
 	scanner.Scan()
@@ -38,24 +37,19 @@ func AddCompany(scanner *bufio.Scanner, db *sql.DB) {
 	}
 	input.Price = price
 
-	fmt.Print("  Currency ID: ")
+	fmt.Print("  Currency Code: ")
 	scanner.Scan()
-	cid, err := strconv.Atoi(strings.TrimSpace(scanner.Text()))
-	if err != nil {
-		fmt.Println("  Invalid currency ID.")
-		return
-	}
-	input.CurrencyID = cid
+	input.CurrencyCode = strings.TrimSpace(scanner.Text())
 
-	if err := service.AddCompany(input, db); err != nil {
+	if _, err := database.AddCompany(db, input); err != nil {
 		fmt.Println("  Error:", err)
 		return
 	}
-	fmt.Printf("  ✓ Company '%s' added.\n", input.Name)
+	fmt.Printf("  ✓ Company '%s' added.\n", input.Symbol)
 }
 
 func ListCompanies(db *sql.DB) {
-	companies, err := service.GetAllCompanies(db)
+	companies, err := database.GetAllCompanies(db)
 	if err != nil {
 		fmt.Println("  ✗ Error:", err)
 		return
@@ -77,6 +71,6 @@ func ListCompanies(db *sql.DB) {
 		}
 
 		fmt.Printf("  %-5d %-20s %-10s %-15.2f %d\n",
-			c.ID, c.Name, currency, c.Price, c.SharesOutstanding)
+			c.ID, c.Symbol, currency.Code, c.Price, c.SharesOutstanding)
 	}
 }