/*
 * Copyright 2021 CloudWeGo Authors
 *
 * 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 thrift

import (
	"context"
	"encoding/json"
	"fmt"
	"strconv"
	"strings"

	"github.com/cloudwego/kitex/pkg/generic/descriptor"
)

func assertType(expected, but descriptor.Type) error {
	if expected == but {
		return nil
	}
	return fmt.Errorf("need %s type, but got: %s", expected, but)
}

func splitType(t string) (pkg, name string) {
	idx := strings.LastIndex(t, ".")
	if idx == -1 {
		return "", t
	}
	return t[:idx], t[idx+1:]
}

func requestMappingValue(ctx context.Context, req *descriptor.HTTPRequest, field *descriptor.FieldDescriptor) (interface{}, error) {
	val, ok, err := field.HTTPMapping.Request(ctx, req, field)
	if err != nil {
		return nil, err
	}
	if !ok {
		return nil, nil
	}
	t := field.Type
	switch v := val.(type) {
	case json.Number:
		bt, err := buildinTypeFromString(string(v), t)
		if err != nil {
			return v, nil
		}
		return bt, nil
	case string:
		bt, err := buildinTypeFromString(v, t)
		if err != nil {
			return v, nil
		}
		return bt, err
	}
	return val, nil
}

func buildinTypeFromString(s string, t *descriptor.TypeDescriptor) (interface{}, error) {
	tt := t.Type
	switch tt {
	case descriptor.BOOL:
		switch s {
		case "true":
			return true, nil
		case "false":
			return false, nil
		}
	case descriptor.I08:
		i, err := strconv.ParseInt(s, 10, 8)
		if err != nil {
			return nil, err
		}
		return int8(i), nil
	case descriptor.I16:
		i, err := strconv.ParseInt(s, 10, 16)
		if err != nil {
			return nil, err
		}
		return int16(i), nil
	case descriptor.I32:
		i, err := strconv.ParseInt(s, 10, 32)
		if err != nil {
			return nil, err
		}
		return int32(i), nil
	case descriptor.I64:
		i, err := strconv.ParseInt(s, 10, 64)
		if err != nil {
			return nil, err
		}
		return i, nil
	case descriptor.DOUBLE:
		return strconv.ParseFloat(s, 64)
	case descriptor.STRING:
		return s, nil
	case descriptor.LIST:
		strs := strings.Split(s, ",")
		if len(strs) == 0 {
			return nil, fmt.Errorf("expected list, got:%s", s)
		}
		res := make([]interface{}, len(strs))
		for i, str := range strs {
			val, err := buildinTypeFromString(str, t.Elem)
			if err != nil {
				return nil, err
			}
			res[i] = val
		}
		return res, nil
	}
	return nil, fmt.Errorf("unsupported type:%s for parse buildin types from string", tt)
}

func buildinTypeIntoString(val interface{}) string {
	switch v := val.(type) {
	case bool:
		return strconv.FormatBool(v)
	case int8:
		return strconv.FormatInt(int64(v), 10)
	case int16:
		return strconv.FormatInt(int64(v), 10)
	case int32:
		return strconv.FormatInt(int64(v), 10)
	case int64:
		return strconv.FormatInt(v, 10)
	case float64:
		return strconv.FormatFloat(v, 'f', -1, 64)
	case string:
		return v
	}
	return ""
}
