package main

import (
	"context"
	"ersteller-lib/schema/ent/example/ent"
	"ersteller-lib/schema/ent/example/ent/user"
	"log"
	"time"

	"entgo.io/ent/dialect/sql"
	_ "github.com/mattn/go-sqlite3"
)

func main() {
	//client, err := ent.Open("sqlite3", "file:ent?mode=memory&cache=shared&_fk=1")
	client, err := ent.Open("sqlite3", "/tmp/ersteller_ent_example.db?_fk=1",
		ent.Log(log.Println), ent.Debug())
	if err != nil {
		log.Fatalf("failed opening connection to sqlite: %v", err)
	}
	log.Println("client", client)
	defer client.Close()
	ctx, cancel := context.WithTimeout(context.Background(), time.Minute*5)
	defer cancel()
	if err := client.Schema.Create(ctx); err != nil {
		log.Fatalf("failed creating schema resources: %v", err)
	}
	u, err := client.User.
		Create().
		SetEmail("something@gorlug.de").
		SetPassword("uhoh").Save(ctx)
	if err != nil {
		log.Fatalf("failed creating user: %w", err)
	}
	log.Println("user was created: ", u)

	//time.Sleep(time.Second * 1)
	//u, err = client.User.UpdateOneID(u.ID).SetPassword("wtf").Save(ctx)
	//if err != nil {
	//	log.Fatalf("failed updating user: %w", err)
	//}j
	//log.Println("user was updated", u)
	users, nextId, hasNext, err := ListUsersAfterID(ctx, client, 0, 2)
	if err != nil {
		log.Fatalf("failed listing users: %w", err)
	}
	if hasNext {
		log.Println("next id", nextId)
		for _, u := range users {
			log.Println("user", u.ID, u.Email, u.Password)
		}
	}
	users, nextId, hasNext, err = ListUsersAfterID(ctx, client, nextId, 2)
	if err != nil {
		log.Fatalf("failed listing users: %w", err)
	}
	if hasNext {
		log.Println("2 next id", nextId)
		for _, u := range users {
			log.Println("2 user", u.ID, u.Email, u.Password)
		}
	}

}

// afterID is the last ID from the previous page. Use 0 for the first page.
func ListUsersAfterID(ctx context.Context, client *ent.Client, afterID, limit int) ([]*ent.User, int, bool, error) {
	q := client.User.Query().
		Order(user.ByID(sql.OrderAsc())).
		Limit(limit + 1) // fetch one extra to detect "has next"

	if afterID > 0 {
		q = q.Where(user.IDGT(afterID))
	}

	rows, err := q.All(ctx)
	if err != nil {
		return nil, -1, false, err
	}

	hasNext := len(rows) > limit
	if hasNext {
		rows = rows[:limit]
	}

	var next int
	if len(rows) > 0 {
		last := rows[len(rows)-1].ID
		next = last
	}
	return rows, next, hasNext, nil
}