account.go

// Copyright 2014 Eryx <evorui аt gmаil dοt cοm>, All rights reserved.
//
// 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 iamapi

import (
	"encoding/binary"
	"encoding/hex"
	"errors"
	"fmt"
	"regexp"
	"time"

	"github.com/lessos/lessgo/crypto/idhash"
	"github.com/lessos/lessgo/types"
)

var (
	AccGenIdReg           = regexp.MustCompile("^[0-9a-f]{16,24}$")
	accPayTimeZero uint32 = 1500000000
)

const (
	AccountCurrencyTypeCash    uint8 = 1
	AccountCurrencyTypeVirtual uint8 = 32
	AccountCurrencyTypeCard    uint8 = 33
)

func AccountCurrencyTypeValid(v uint8) bool {
	if v == AccountCurrencyTypeCash ||
		v == AccountCurrencyTypeVirtual ||
		v == AccountCurrencyTypeCard {
		return true
	}
	return false
}

// iam/acc_user/user-id
type AccountUser struct {
	User    string         `json:"user" toml:"user"`
	Balance float64        `json:"balance" toml:"balance"`
	Prepay  float64        `json:"prepay" toml:"prepay"`
	Updated types.MetaTime `json:"updated" toml:"updated"`
}

type AccountCurrencyOption struct {
	Name  string       `json:"name" toml:"name"`
	Items types.Labels `json:"items,omitempty" toml:"items,omitempty"`
}

// iam/acc_fund/user-id/rand-id
// iam/acc_fund_mgr/rand-id
type AccountFund struct {
	Id               string                    `json:"id" toml:"id"`
	Type             uint8                     `json:"type" toml:"type"`
	User             string                    `json:"user" toml:"user"`
	Operator         string                    `json:"operator,omitempty" toml:"operator,omitempty"`
	CashType         uint16                    `json:"cash_type,omitempty" toml:"cash_type,omitempty"`
	CashAmount       float32                   `json:"cash_amount,omitempty" toml:"cash_amount,omitempty"`
	Amount           float64                   `json:"amount" toml:"amount"`
	Prepay           float64                   `json:"prepay" toml:"prepay"`
	Payout           float64                   `json:"payout" toml:"payout"`
	Priority         uint8                     `json:"priority" toml:"priority"`
	Options          types.Labels              `json:"options,emitempty" toml:"options,emitempty"`
	Created          types.MetaTime            `json:"created" toml:"created"`
	Updated          types.MetaTime            `json:"updated" toml:"updated"`
	Comment          string                    `json:"comment,omitempty" toml:"comment,omitempty"`
	ExpProductLimits types.ArrayNameIdentifier `json:"exp_product_limits,omitempty" toml:"exp_product_limits,omitempty"`
	ExpProductMax    int                       `json:"exp_product_max,omitempty" toml:"exp_product_max,omitempty"`
	ExpProductInpay  types.ArrayNameIdentifier `json:"exp_product_inpay,omitempty" toml:"exp_product_inpay,omitempty"`
}

// iam/acc_charge/user-id/hash-id
// iam/acc_charge_mgr/hash-id
type AccountCharge struct {
	types.TypeMeta `json:",inline" toml:",inline"`
	Id             string               `json:"id" toml:"id"`
	Fund           string               `json:"fund" toml:"fund"`
	User           string               `json:"user" toml:"user"`
	Product        types.NameIdentifier `json:"product" toml:"product"`
	Prepay         float64              `json:"prepay" toml:"prepay"`
	Payout         float64              `json:"payout" toml:"payout"`
	TimeStart      uint32               `json:"time_start" toml:"time_start"`
	TimeClose      uint32               `json:"time_close" toml:"time_close"`
	Created        types.MetaTime       `json:"created" toml:"created"`
	Updated        types.MetaTime       `json:"updated" toml:"updated"`
	Comment        string               `json:"comment,omitempty" toml:"comment,omitempty"`
}

func AccountChargeId(prod types.NameIdentifier, start uint32) ([]byte, string) {

	bs := make([]byte, 4)
	binary.BigEndian.PutUint32(bs, start)

	hk := idhash.Hash([]byte(fmt.Sprintf(
		"acc.charge.%s.%d",
		prod.String(),
		start,
	)), 8)

	rs := append(bs, hk...)

	return rs, hex.EncodeToString(rs)
}

const (
	PayTypeLease uint8 = 1
	PayTypeOrder uint8 = 2
)

type AccountChargePrepay struct {
	types.TypeMeta `json:",inline" toml:",inline"`
	User           string               `json:"user" toml:"user"`
	Product        types.NameIdentifier `json:"product" toml:"product"`
	Prepay         float64              `json:"prepay" toml:"prepay"`
	TimeStart      uint32               `json:"time_start" toml:"time_start"`
	TimeClose      uint32               `json:"time_close" toml:"time_close"`
	Comment        string               `json:"comment,omitempty" toml:"comment,omitempty"`
}

func (this *AccountChargePrepay) Valid() error {

	if !UsernameRE.MatchString(this.User) {
		return errors.New("Invalid User")
	}

	if err := this.Product.Valid(); err != nil {
		return errors.New("Invalid Product : " + err.Error())
	}

	if this.Prepay <= 0 {
		return errors.New("Invalid Prepay")
	}

	if this.TimeStart < accPayTimeZero {
		return errors.New("Invalid TimeStart")
	}

	if this.TimeClose < accPayTimeZero {
		return errors.New("Invalid TimeClose")
	}

	if this.TimeStart >= this.TimeClose {
		return errors.New("Invalid TimeStart or TimeClose")
	}

	return nil
}

type AccountChargePayout struct {
	types.TypeMeta `json:",inline" toml:",inline"`
	Id             string               `json:"id" toml:"id"`
	User           string               `json:"user" toml:"user"`
	Product        types.NameIdentifier `json:"product" toml:"product"`
	Payout         float64              `json:"payout" toml:"payout"`
	TimeStart      uint32               `json:"time_start" toml:"time_start"`
	TimeClose      uint32               `json:"time_close" toml:"time_close"`
	Comment        string               `json:"comment,omitempty" toml:"comment,omitempty"`
}

func (this *AccountChargePayout) Valid() error {

	if !UsernameRE.MatchString(this.User) {
		return errors.New("Invalid User")
	}

	if err := this.Product.Valid(); err != nil {
		return errors.New("Invalid Product : " + err.Error())
	}

	if this.Payout <= 0 {
		return errors.New("Invalid Payout")
	}

	if this.TimeStart < accPayTimeZero {
		return errors.New("Invalid TimeStart")
	}

	if this.TimeClose < accPayTimeZero {
		return errors.New("Invalid TimeClose")
	}

	if this.TimeStart >= this.TimeClose {
		return errors.New("Invalid TimeStart or TimeClose")
	}

	return nil
}

const (
	AccountChargeTypePrepay uint8 = 1
	AccountChargeTypePayout uint8 = 2

	AccountChargeCycleHour  uint32 = 3600
	AccountChargeCycleDay   uint32 = 86400
	AccountChargeCycleMonth uint32 = 2592000
)

func account_charge_cycle_fix(cycle uint32) uint32 {
	if cycle < AccountChargeCycleHour {
		cycle = AccountChargeCycleHour
	} else if cycle > AccountChargeCycleMonth {
		cycle = AccountChargeCycleMonth
	}
	if fix := cycle % 3600; fix > 0 {
		cycle -= fix
	}
	return cycle
}

func AccountChargeCycleTimeCloseNow(cycle uint32) uint32 {

	cycle = account_charge_cycle_fix(cycle)
	var (
		tm  = time.Now()
		ctm = uint32(tm.Unix())
	)

	if cycle >= AccountChargeCycleMonth {

		if tm.Month() == 12 {
			ctm = uint32(time.Date(tm.Year()+1, 1, 1,
				0, 0, 0, 0, time.Local).Unix())
		} else {
			ctm = uint32(time.Date(tm.Year(), tm.Month()+1, 1,
				0, 0, 0, 0, time.Local).Unix())
		}

	} else if cycle >= AccountChargeCycleDay {

		ctm = uint32(time.Date(tm.Year(), tm.Month(), tm.Day(),
			0, 0, 0, 0, time.Local).AddDate(0, 0, 1).Unix())

	} else {

		offset := uint32(tm.Hour()*3600 + tm.Minute()*60 + tm.Second())
		if fix := offset % cycle; fix > 0 {
			ctm -= fix
		}

		ctm += cycle
	}

	return ctm
}

func AccountChargeCycleTimeClose(cycle, ctc uint32) uint32 {

	cycle = account_charge_cycle_fix(cycle)
	var (
		tm  = time.Unix(int64(ctc), 0)
		ctm = ctc
	)

	if cycle >= AccountChargeCycleMonth {

		ctm = uint32(time.Date(tm.Year(), tm.Month(), 1,
			0, 0, 0, 0, time.Local).Unix())

		if ctm < ctc {

			if tm.Month() == 12 {
				ctm = uint32(time.Date(tm.Year()+1, 1, 1,
					0, 0, 0, 0, time.Local).Unix())
			} else {
				ctm = uint32(time.Date(tm.Year(), tm.Month()+1, 1,
					0, 0, 0, 0, time.Local).Unix())
			}
		}

	} else if cycle >= AccountChargeCycleDay {

		ctm = uint32(time.Date(tm.Year(), tm.Month(), tm.Day(),
			0, 0, 0, 0, time.Local).Unix())

		if ctm < ctc {
			ctm = uint32(time.Date(tm.Year(), tm.Month(), tm.Day(),
				0, 0, 0, 0, time.Local).AddDate(0, 0, 1).Unix())
		}

	} else {

		offset := uint32(tm.Hour()*3600 + tm.Minute()*60 + tm.Second())
		if fix := offset % cycle; fix > 0 {
			ctm = ctm - fix + cycle
		}
	}

	return ctm
}

func AccountChargeTimeNow() uint32 {
	return uint32(time.Now().Unix())
}

func AccountFloat64Round(f float64, pa_num int64) float64 {
	pa_fix := float64(1e4)
	switch pa_num {
	case 2:
		pa_fix = 1e2
	case 3:
		pa_fix = 1e3
	default:
		pa_fix = 1e4
	}
	return float64(int64(f*pa_fix+0.5)) / pa_fix
}