parcoursmob/handlers/exports/fleets.go

package exports

import (
	"context"
	"fmt"
	"net/http"

	fleets "git.coopgo.io/coopgo-platform/fleets/grpcapi"
	fleetsstorage "git.coopgo.io/coopgo-platform/fleets/storage"
	groupsmanagement "git.coopgo.io/coopgo-platform/groups-management/grpcapi"
	groupsstorage "git.coopgo.io/coopgo-platform/groups-management/storage"
	accounts "git.coopgo.io/coopgo-platform/mobility-accounts/grpcapi"
	accountsstorage "git.coopgo.io/coopgo-platform/mobility-accounts/storage"
	"github.com/xuri/excelize/v2"
)

func (h *ExportsHandler) Bookings(w http.ResponseWriter, r *http.Request) {

	vehicles := map[string]fleetsstorage.Vehicle{}
	bookings := []fleetsstorage.Booking{}

	reequest := &fleets.GetVehiclesRequest{
		Namespaces: []string{"parcoursmob"},
	}
	reesp, err := h.services.GRPC.Fleets.GetVehicles(context.TODO(), reequest)
	if err != nil {
		fmt.Println(err)
		w.WriteHeader(http.StatusInternalServerError)
		return
	}

	beneficiaries_ids := []string{}

	for _, vehicle := range reesp.Vehicles {

		v := vehicle.ToStorageType()
		fmt.Println(v)

		for _, b := range v.Bookings {
			bookings = append(bookings, b)
			beneficiaries_ids = append(beneficiaries_ids, b.Driver)
		}

		vehicles[vehicle.Id] = v

	}

	groups := map[string]groupsstorage.Group{}

	admingroups, err := h.services.GRPC.GroupsManagement.GetGroups(context.TODO(), &groupsmanagement.GetGroupsRequest{
		Namespaces: []string{"parcoursmob_organizations"},
	})
	if err != nil {
		fmt.Println(err)
		w.WriteHeader(http.StatusInternalServerError)
		return
	}

	for _, g := range admingroups.Groups {
		groups[g.Id] = g.ToStorageType()
	}

	beneficiaries, err := h.services.GRPC.MobilityAccounts.GetAccountsBatch(context.TODO(), &accounts.GetAccountsBatchRequest{
		Accountids: beneficiaries_ids,
	})
	if err != nil {
		fmt.Println(err)
		w.WriteHeader(http.StatusInternalServerError)
		return
	}

	beneficiaries_map := map[string]accountsstorage.Account{}
	for _, ben := range beneficiaries.Accounts {
		beneficiaries_map[ben.Id] = ben.ToStorageType()
	}

	/////////////// Generate file

	f := excelize.NewFile()
	defer func() {
		if err := f.Close(); err != nil {
			fmt.Println(err)
		}
	}()

	f.SetCellValue("Sheet1", "A1", "Numéro")
	f.SetCellValue("Sheet1", "B1", "Type")
	f.SetCellValue("Sheet1", "C1", "Gestionnaire")
	f.SetCellValue("Sheet1", "D1", "Prescripteur")
	f.SetCellValue("Sheet1", "E1", "Bénéficiaire")
	f.SetCellValue("Sheet1", "F1", "Numéro allocataire / Pole emploi")
	f.SetCellValue("Sheet1", "G1", "Début de Mise à disposition")
	f.SetCellValue("Sheet1", "H1", "Fin de mise à disposition")
	f.SetCellValue("Sheet1", "I1", "Début indisponibilité")
	f.SetCellValue("Sheet1", "J1", "Fin indisponibilité")
	f.SetCellValue("Sheet1", "K1", "Retrait par le gestionnaire")
	f.SetCellValue("Sheet1", "L1", "Commentaire")

	i := 2
	for _, b := range bookings {
		vehicle := vehicles[b.Vehicleid]
		if len(vehicle.Administrators) == 0 {
			continue
		}
		admin := groups[vehicle.Administrators[0]]

		bookedby := ""
		if v, ok := b.Data["booked_by"].(map[string]any); ok {
			if v2, ok := v["group"].(map[string]any); ok {
				if v3, ok := v2["name"].(string); ok {
					bookedby = v3
				}

			}
		}

		beneficiary := beneficiaries_map[b.Driver]
		adminunavailability := false

		if av, ok := b.Data["administrator_unavailability"].(bool); ok && av {
			adminunavailability = true
		}

		f.SetCellValue("Sheet1", fmt.Sprintf("A%d", i), vehicle.Data["licence_plate"])
		f.SetCellValue("Sheet1", fmt.Sprintf("B%d", i), vehicle.Type)
		f.SetCellValue("Sheet1", fmt.Sprintf("C%d", i), admin.Data["name"])
		f.SetCellValue("Sheet1", fmt.Sprintf("D%d", i), bookedby)
		f.SetCellValue("Sheet1", fmt.Sprintf("E%d", i), fmt.Sprintf("%v %v", beneficiary.Data["first_name"], beneficiary.Data["last_name"]))
		f.SetCellValue("Sheet1", fmt.Sprintf("F%d", i), beneficiary.Data["file_number"])
		f.SetCellValue("Sheet1", fmt.Sprintf("G%d", i), b.Startdate.Format("2006-01-02"))
		f.SetCellValue("Sheet1", fmt.Sprintf("H%d", i), b.Enddate.Format("2006-01-02"))
		f.SetCellValue("Sheet1", fmt.Sprintf("I%d", i), b.Unavailablefrom.Format("2006-01-02"))
		f.SetCellValue("Sheet1", fmt.Sprintf("J%d", i), b.Unavailableto.Format("2006-01-02"))
		f.SetCellValue("Sheet1", fmt.Sprintf("K%d", i), adminunavailability)
		f.SetCellValue("Sheet1", fmt.Sprintf("L%d", i), b.Data["comment"])
		i = i + 1
	}

	w.Header().Set("Content-Type", "application/octet-stream")
	w.Header().Set("Content-Disposition", "attachment; filename="+"Workbook.xlsx")
	w.Header().Set("Content-Transfer-Encoding", "binary")
	w.Header().Set("Expires", "0")
	f.Write(w)

}