// Copyright 2019 Yunion
//
// 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 shell

import (
	"yunion.io/x/jsonutils"

	"yunion.io/x/onecloud/pkg/mcclient"
	"yunion.io/x/onecloud/pkg/mcclient/modulebase"
	"yunion.io/x/onecloud/pkg/mcclient/modules"
	"yunion.io/x/onecloud/pkg/mcclient/options"
)

func init() {

	/**
	 * 操作用户的通信地址（如果用户的通信地址不存在则进行添加；如果已存在则进行修改；如果设置空则进行删除。）
	 */
	type ContactsUpdateOptions struct {
		UID         string   `help:"The user you wanna add contact to (Keystone User ID)"`
		CONTACTTYPE string   `help:"The contact type email|mobile" choices:"email|mobile|dingtalk"`
		CONTACT     string   `help:"The contacts details mobile number or email address or dingtalk's userid, if set it the empty str means delete"`
		Status      string   `help:"Enabled or disabled contact status" choices:"enable|disable"`
		Pull        []string `help:"pull some subcontacts(e.g., dingtalk, feishu, etc) related to mobile"`
	}
	R(&ContactsUpdateOptions{}, "contact-update", "Create, delete or update contact for user", func(s *mcclient.ClientSession, args *ContactsUpdateOptions) error {
		arr := jsonutils.NewArray()
		tmpObj := jsonutils.NewDict()
		tmpObj.Add(jsonutils.NewString(args.CONTACTTYPE), "contact_type")
		tmpObj.Add(jsonutils.NewString(args.CONTACT), "contact")
		if len(args.Status) > 0 {
			if args.Status == "disable" {
				tmpObj.Add(jsonutils.NewInt(0), "enabled")
			} else {
				tmpObj.Add(jsonutils.NewInt(1), "enabled")
			}
		}

		arr.Add(tmpObj)

		params := jsonutils.NewDict()
		params.Add(arr, "contacts")

		pulls := jsonutils.NewArray()
		for _, pull := range args.Pull {
			pulls.Add(jsonutils.NewString(pull))
		}
		params.Add(pulls, "pull")

		contact, err := modules.Contacts.CustomizedPerformAction(s, args.UID, "update-contact", params)
		if err != nil {
			return err
		}

		printObject(contact)
		return nil
	})

	type ContactsPullOptions struct {
		UID         string `help:"The user you wanna pull contact"`
		CONTACTTYPE string `help:"The contact type"`
	}
	R(&ContactsPullOptions{}, "contact-pull", "Pull contact", func(s *mcclient.ClientSession, args *ContactsPullOptions) error {
		params := jsonutils.NewDict()
		params.Set("contacts", jsonutils.NewArray())
		params.Set("pull", jsonutils.NewArray(jsonutils.NewString(args.CONTACTTYPE)))
		contact, err := modules.Contacts.CustomizedPerformAction(s, args.UID, "update-contact", params)
		if err != nil {
			return err
		}
		printObject(contact)
		return nil
	})

	type ContactsDeleteOptions struct {
		UID         string `help:"The user you wanna add contact to (Keystone User ID)"`
		CONTACTTYPE string `help:"The contact type email|mobile|dingtalk" choices:"email|mobile|dingtalk"`
	}
	R(&ContactsDeleteOptions{}, "contact-delete", "Delete contact for user", func(s *mcclient.ClientSession, args *ContactsDeleteOptions) error {
		arr := jsonutils.NewArray()
		tmpObj := jsonutils.NewDict()
		tmpObj.Add(jsonutils.NewString(args.CONTACTTYPE), "contact_type")
		tmpObj.Add(jsonutils.NewString(""), "contact")
		arr.Add(tmpObj)
		params := jsonutils.NewDict()
		params.Add(arr, "contacts")
		contact, err := modules.Contacts.CustomizedPerformAction(s, args.UID, "update-contact", params)
		if err != nil {
			return err
		}
		printObject(contact)
		return nil
	})

	/**
	 * 获得所有用户的所有通信地址列表
	 */
	type ContactsListOptions struct {
		options.BaseListOptions
	}
	R(&ContactsListOptions{}, "contact-list", "List all contacts for all users", func(s *mcclient.ClientSession, args *ContactsListOptions) error {
		var params *jsonutils.JSONDict
		{
			var err error
			params, err = args.BaseListOptions.Params()
			if err != nil {
				return err

			}
		}
		params.Add(jsonutils.JSONTrue, "details")

		result, err := modules.Contacts.List(s, params)
		if err != nil {
			return err
		}

		printList(result, modules.Contacts.GetColumns(s))
		return nil
	})

	/**
	 * 获得一个用户全部通信地址
	 */
	type ContactsListForUserOptions struct {
		options.BaseListOptions
		UID string `help:"The user you wanna find contact from (Keystone User ID)"`
	}
	R(&ContactsListForUserOptions{}, "contact-show", "List all contacts for the users", func(s *mcclient.ClientSession, args *ContactsListForUserOptions) error {
		var params *jsonutils.JSONDict
		{
			var err error
			params, err = args.BaseListOptions.Params()
			if err != nil {
				return err

			}
		}

		params.Add(jsonutils.JSONTrue, "details")

		result, err := modules.Contacts.CustomizedGet(s, args.UID, params)
		if err != nil {
			return err
		}

		contactsStr, err := result.GetString("details")
		if err != nil {
			return nil
		}

		contactsJson, err := jsonutils.ParseString(contactsStr)
		if err != nil {
			return err
		}

		contacts, err := contactsJson.GetArray()
		if err != nil {
			return err
		}

		printList(&modulebase.ListResult{Data: contacts}, nil)
		return nil
	})

	/**
	 * 触发验证通信地址操作
	 */
	type ContactsVerifyOptions struct {
		UID          string `help:"The user you wanna verify contact for (Keystone User ID)"`
		CONTACT_TYPE string `help:"The contact type email|mobile"`
		CONTACT      string `help:"The contacts details mobile number or email address"`
	}
	R(&ContactsVerifyOptions{}, "contact-verify-trigger", "Trigger contact verify", func(s *mcclient.ClientSession, args *ContactsVerifyOptions) error {
		tmpDict := jsonutils.NewDict()
		tmpDict.Add(jsonutils.NewString(args.CONTACT_TYPE), "contact_type")
		tmpDict.Add(jsonutils.NewString(args.CONTACT), "contact")
		_, err := modules.Contacts.CustomizedPerformAction(s, args.UID, "verify", tmpDict)
		if err != nil {
			return err
		}
		return nil
	})

	type ContactsBatchDeleteOptions struct {
		UIDS []string `help:"All user'id you wanna to delete contacts (Keystone User ID)"`
	}
	R(&ContactsBatchDeleteOptions{}, "contact-delete", "Delete all contacts for the user", func(s *mcclient.ClientSession, args *ContactsBatchDeleteOptions) error {
		arr := jsonutils.NewArray()
		for _, f := range args.UIDS {
			arr.Add(jsonutils.NewString(f))
		}

		params := jsonutils.NewDict()
		params.Add(arr, "contacts")

		contact, err := modules.Contacts.DoBatchDeleteContacts(s, params)

		if err != nil {
			return err
		}

		printObject(contact)
		return nil
	})
}