// Copyright (c) 2015 Mattermost, Inc. All Rights Reserved.
// See License.txt for license information.

package server

import (
	"encoding/json"
	"fmt"
	"io"
	"os"
	"path/filepath"
)

type ConfigPushProxy struct {
	AndroidPushSettings     []AndroidPushSettings
	ListenAddress           string
	ThrottleVaryByHeader    string
	LogFileLocation         string
	SendTimeoutSec          int
	RetryTimeoutSec         int
	ApplePushSettings       []ApplePushSettings
	EnableMetrics           bool
	EnableConsoleLog        bool
	EnableFileLog           bool
	LogFormat               string // json or plain
	ThrottlePerSec          int
	ThrottleMemoryStoreSize int
}

type ApplePushSettings struct {
	Type                    string
	ApplePushCertPrivate    string
	ApplePushCertPassword   string
	ApplePushTopic          string
	AppleAuthKeyFile        string
	AppleAuthKeyID          string
	AppleTeamID             string
	ApplePushUseDevelopment bool
}

type AndroidPushSettings struct {
	Type                string
	AndroidAPIKey       string `json:"AndroidApiKey"`
	ServiceFileLocation string `json:"ServiceFileLocation"`
}

// FindConfigFile searches for the filepath in a list of directories
// and then returns the absolute path to that file.
func FindConfigFile(fileName string) string {
	if _, err := os.Stat("/tmp/" + fileName); err == nil {
		fileName, _ = filepath.Abs("/tmp/" + fileName)
	} else if _, err := os.Stat("./config/" + fileName); err == nil {
		fileName, _ = filepath.Abs("./config/" + fileName)
	} else if _, err := os.Stat("../config/" + fileName); err == nil {
		fileName, _ = filepath.Abs("../config/" + fileName)
	} else if _, err := os.Stat(fileName); err == nil {
		fileName, _ = filepath.Abs(fileName)
	}

	return fileName
}

// LoadConfig loads the config from the given file path.
func LoadConfig(fileName string) (*ConfigPushProxy, error) {
	file, err := os.Open(fileName)
	if err != nil {
		return nil, err
	}
	defer file.Close()

	buf, err := io.ReadAll(file)
	if err != nil {
		return nil, err
	}

	var cfg *ConfigPushProxy
	err = json.Unmarshal(buf, &cfg)
	if err != nil {
		fmt.Println(buf, err)
		return nil, err
	}

	if !cfg.EnableConsoleLog && !cfg.EnableFileLog {
		cfg.EnableConsoleLog = true
	}

	// Set timeout defaults
	if cfg.SendTimeoutSec == 0 {
		cfg.SendTimeoutSec = 30
	}

	if cfg.RetryTimeoutSec == 0 {
		cfg.RetryTimeoutSec = 8
	}

	if cfg.RetryTimeoutSec > cfg.SendTimeoutSec {
		cfg.RetryTimeoutSec = cfg.SendTimeoutSec
	}

	if cfg.EnableFileLog {
		if cfg.LogFileLocation == "" {
			// We just do an mkdir -p equivalent.
			// Otherwise, it would need 2 steps of statting and creating.
			err := os.MkdirAll("./logs", 0755)
			if err != nil {
				// If it fails, we log in the current directory itself
				cfg.LogFileLocation = "./push_proxy.log"
			} else {
				cfg.LogFileLocation = "./logs/push_proxy.log"
			}
		}
		// if file does not exist, create it.
		if _, err := os.Stat(cfg.LogFileLocation); os.IsNotExist(err) {
			f, err := os.Create(cfg.LogFileLocation)
			if err != nil {
				return nil, err
			}
			if err := f.Close(); err != nil {
				return nil, err
			}
		}
	}

	return cfg, nil
}