package main

import (
	"context"
	"flag"
	"fmt"
	"hackernewsletter/db"
	"hackernewsletter/hackernews"
	"hackernewsletter/params"
	"log"

	"github.com/aws/aws-sdk-go-v2/config"
	"github.com/aws/aws-sdk-go-v2/service/dynamodb"
)

func main() {
	conf, err := params.LoadConfig(".")
	if err != nil {
		log.Fatal("Cannot load config:", err)
	}

	table := flag.String("t", "", "The name of the table")
	flag.Parse()

	if *table == "" {
		fmt.Println("Table name not specified. Using default name.")
		table = &conf.TableName
	}

	cfg, err := config.LoadDefaultConfig(context.TODO(), config.WithRegion(conf.AwsRegion))
	if err != nil {
		panic("Unable to load SDK config, " + err.Error())
	}

	client := dynamodb.NewFromConfig(cfg)

	input := &dynamodb.DescribeTableInput{
		TableName: table,
	}

	newTable := db.Table{client, *table}

	resp, err := db.GetTableInfo(context.TODO(), client, input)
	if err != nil {
		println(("Table not found. Creating it..."))
		_, new_err := db.CreateTable(newTable)
		if new_err != nil {
			panic("Failed creating table " + *table)
		}

		resp, err = db.GetTableInfo(context.TODO(), client, input)
	}

	fmt.Printf("Table %v has %v elements\n", *table, resp.Table.ItemCount)

	newsIds := hackernews.GetTopNewsIds(conf.TopNews)

	var newsBatch []db.News

	for i := 0; i < 10; i++ {
		myNews := hackernews.GetNewsById(newsIds[i], conf.SingleNews)
		newsBatch = append(newsBatch, myNews)
	}

	db.AddNewsBatch(newTable, newsBatch, conf.BatchSize)
}