// Copyright 2023 Harness, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package openapi

import (
	"net/http"

	"github.com/harness/gitness/app/api/controller/upload"
	"github.com/harness/gitness/app/api/usererror"

	"github.com/gotidy/ptr"
	"github.com/swaggest/openapi-go/openapi3"
)

func uploadOperations(reflector *openapi3.Reflector) {
	opUpload := openapi3.Operation{}
	opUpload.WithTags("upload")
	opUpload.WithMapOfAnything(map[string]interface{}{"operationId": "repoArtifactUpload"})
	opUpload.WithRequestBody(openapi3.RequestBodyOrRef{
		RequestBody: &openapi3.RequestBody{
			Description: ptr.String("Binary file to upload"),
			Content: map[string]openapi3.MediaType{
				"application/octet-stream": {Schema: &openapi3.SchemaOrRef{}},
			},
			Required: ptr.Bool(true),
		},
	})
	_ = reflector.SetRequest(&opUpload, repoRequest{}, http.MethodPost)
	_ = reflector.SetJSONResponse(&opUpload, new(upload.Result), http.StatusCreated)
	_ = reflector.SetJSONResponse(&opUpload, new(usererror.Error), http.StatusBadRequest)
	_ = reflector.SetJSONResponse(&opUpload, new(usererror.Error), http.StatusNotFound)
	_ = reflector.SetJSONResponse(&opUpload, new(usererror.Error), http.StatusInternalServerError)
	_ = reflector.SetJSONResponse(&opUpload, new(usererror.Error), http.StatusUnauthorized)
	_ = reflector.SetJSONResponse(&opUpload, new(usererror.Error), http.StatusForbidden)

	_ = reflector.Spec.AddOperation(http.MethodPost, "/repos/{repo_ref}/uploads", opUpload)

	downloadOp := openapi3.Operation{}
	downloadOp.WithTags("upload")
	downloadOp.WithMapOfAnything(map[string]interface{}{"operationId": "repoArtifactDownload"})
	_ = reflector.SetRequest(&downloadOp, struct {
		repoRequest
		FilePathRef string `path:"file_ref"`
	}{}, http.MethodGet)
	_ = reflector.SetupResponse(openapi3.OperationContext{
		Operation:  &downloadOp,
		HTTPStatus: http.StatusOK,
	})
	_ = reflector.SetJSONResponse(&downloadOp, nil, http.StatusTemporaryRedirect)
	_ = reflector.SetJSONResponse(&downloadOp, new(usererror.Error), http.StatusNotFound)
	_ = reflector.SetJSONResponse(&downloadOp, new(usererror.Error), http.StatusBadRequest)
	_ = reflector.SetJSONResponse(&downloadOp, new(usererror.Error), http.StatusInternalServerError)
	_ = reflector.SetJSONResponse(&downloadOp, new(usererror.Error), http.StatusUnauthorized)
	_ = reflector.SetJSONResponse(&downloadOp, new(usererror.Error), http.StatusForbidden)

	_ = reflector.Spec.AddOperation(http.MethodGet, "/repos/{repo_ref}/uploads/{file_ref}", downloadOp)
}