feat: use Anny API and booked entries in YAML just as a fallback

cmd/rooms.go

@@ -0,0 +1,34 @@
+package cmd
+
+import (
+	"fmt"
+
+	"github.com/spf13/cobra"
+)
+
+var (
+	roomsCmd = &cobra.Command{
+		Use:   "rooms [subcommand]",
+		Short: "get rooms",
+		Long: `Get rooms.
+anny		    --- fetch bookings from anny.eu.`,
+		ValidArgs: []string{"anny"},
+		Args:      cobra.MinimumNArgs(1),
+		Run: func(cmd *cobra.Command, args []string) {
+			p := initPlexamsConfig()
+			switch args[0] {
+			case "anny":
+				err := p.FetchFromAnny()
+				if err != nil {
+					panic(err)
+				}
+			default:
+				fmt.Println("info called with unknown sub command")
+			}
+		},
+	}
+)
+
+func init() {
+	rootCmd.AddCommand(roomsCmd)
+}

db/anny_bookings.go

@@ -0,0 +1,69 @@
+package db
+
+import (
+	"context"
+	"strings"
+
+	"github.com/obcode/plexams.go/graph/model"
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
+)
+
+func (db *DB) SaveAnnyBookings(ctx context.Context, bookings []*model.AnnyBooking) error {
+	collection := db.Client.Database(db.databaseName).Collection(collectionAnnyBookings)
+
+	err := collection.Drop(ctx)
+	if err != nil {
+		return err
+	}
+
+	entries := make([]interface{}, 0, len(bookings))
+	for _, booking := range bookings {
+		entries = append(entries, booking)
+	}
+
+	_, err = collection.InsertMany(ctx, entries)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (db *DB) AnnyBookings(ctx context.Context, room *string) ([]*model.AnnyBooking, error) {
+	return db.annyBookings(ctx, room)
+}
+
+func (db *DB) AllAnnyBookings(ctx context.Context) ([]*model.AnnyBooking, error) {
+	return db.annyBookings(ctx, nil)
+}
+
+func (db *DB) annyBookings(ctx context.Context, room *string) ([]*model.AnnyBooking, error) {
+	collection := db.Client.Database(db.databaseName).Collection(collectionAnnyBookings)
+
+	filter := bson.M{}
+	if room != nil && strings.TrimSpace(*room) != "" {
+		filter["room"] = strings.ToUpper(strings.ReplaceAll(strings.TrimSpace(*room), " ", ""))
+	}
+
+	findOptions := options.Find()
+	findOptions.SetSort(bson.D{{Key: "start_date", Value: 1}})
+
+	cur, err := collection.Find(ctx, filter, findOptions)
+	if err != nil {
+		return nil, err
+	}
+	defer cur.Close(ctx) //nolint:errcheck
+
+	bookings := make([]*model.AnnyBooking, 0)
+	if err := cur.All(ctx, &bookings); err != nil {
+		return nil, err
+	}
+
+	return bookings, nil
+}
+
+func (db *DB) AnnyBookingsCollection(ctx context.Context) *mongo.Collection {
+	return db.Client.Database(db.databaseName).Collection(collectionAnnyBookings)
+}

db/collection.go

@@ -16,6 +16,7 @@ const (
 	collectionNameNTAs           = "nta"
 	collectionNamePlan           = "plan"
 	collectionNamePlanBackup     = "plan_backup"
+	collectionAnnyBookings       = "anny_bookings"
 
 	collectionStudentRegsPerStudentPlanned = "studentregs_per_student_planned"
 	collectionZpaStudents                  = "zpastudents"

graph/anny.graphqls

@@ -0,0 +1,29 @@
+extend type Query {
+  annyBookings(room: String): [AnnyBooking!]!
+  allAnnyBookings: [AnnyBooking!]!
+}
+
+type AnnyBooking {
+  number: String!
+  startDate: Time!
+  endDate: Time!
+  blockerStartDate: Time!
+  blockerEndDate: Time!
+  chargedDuration: Int!
+  description: String!
+  createdAt: Time!
+  updatedAt: Time!
+  canceledAt: Time
+  status: String!
+  isBlocker: Boolean!
+  canEdit: Boolean!
+  isEditable: Boolean!
+  manuallyCreated: Boolean!
+  note: String!
+  room: String
+  self: String!
+  personalizationName: String!
+  bookingGroupIdentifier: String
+  cancelableUntil: Time
+  hasCustomDescription: Boolean!
+}