package web

import (
	"fmt"
	"net/http"
	"time"

	"inventree/internal/assets"
)

func (a *App) handleExportCSV(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()
	filter := assets.ListFilter{
		ItemTypeID:     queryInt64(r, "item_type_id"),
		StatusID:       queryInt64(r, "status_id"),
		AssignedUserID: queryInt64(r, "assigned_user_id"),
		Query:          r.URL.Query().Get("q"),
	}
	if folderID, err := queryOptionalID(r, "folder_id"); err == nil && folderID != nil {
		filter.FolderID = folderID
		filter.FolderSet = true
		filter.IncludeSub = r.URL.Query().Get("recursive") == "1"
	}

	filename := fmt.Sprintf("inventree-assets-%s.csv", time.Now().Format("2006-01-02"))
	w.Header().Set("Content-Type", "text/csv; charset=utf-8")
	w.Header().Set("Content-Disposition", "attachment; filename=\""+filename+"\"")
	if err := a.Assets.Export(ctx, w, filter); err != nil {
		a.Log.Error("csv export failed", "error", err)
	}
}

// handleImportCSV backs the folder-scoped "Import CSV..." context-menu
// action: every row imports directly into the given folder_id, so the CSV
// needs no Folder column at all. It's only ever invoked from within a
// specific folder (never at the top level) and responds with JSON since
// it's called via fetch from the tree/panel context menu.
func (a *App) handleImportCSV(w http.ResponseWriter, r *http.Request) {
	ctx := r.Context()
	if err := r.ParseMultipartForm(32 << 20); err != nil {
		jsonErr(w, http.StatusBadRequest, err)
		return
	}
	file, _, err := r.FormFile("file")
	if err != nil {
		jsonErr(w, http.StatusBadRequest, err)
		return
	}
	defer file.Close()

	opts := assets.ImportOptions{}
	if folderID, err := formOptionalID(r, "folder_id"); err == nil && folderID != nil {
		opts.TargetFolderID = folderID
	}

	itemTypes, err := a.ItemTypes.List(ctx, true)
	if err != nil {
		jsonErr(w, http.StatusInternalServerError, err)
		return
	}
	statusList, err := a.Statuses.List(ctx)
	if err != nil {
		jsonErr(w, http.StatusInternalServerError, err)
		return
	}
	userList, err := a.Users.List(ctx)
	if err != nil {
		jsonErr(w, http.StatusInternalServerError, err)
		return
	}

	result, err := a.Assets.Import(ctx, file, opts, itemTypes, statusList, userList)
	if err != nil {
		jsonErr(w, http.StatusBadRequest, err)
		return
	}
	jsonOK(w, result)
}