config.go


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39

package main

import "fmt"

// Config represents the global Portgate config.
type Config struct {
	// Where Portgate will be running at.
	portgatePort int
	portgateHost string

	// Where the requests will be proxied to.
	targetHost string

	allowedPorts   []int
	forbiddenPorts []int
}

// GetConfig creates the Portgate config from outside sources such as
// the environment variables and the portgate.yml file.
func GetConfig() (Config, error) {
	// TODO: Read config from environment/file
	return Config{
		portgatePort: 8080,
		targetHost:   "localhost",

		allowedPorts:   []int{80},
		forbiddenPorts: []int{},
	}, nil
}

// PortgateAddress is the address on which Portgate will run.
func (c *Config) PortgateAddress() string {
	return fmt.Sprintf("%s:%d", c.portgateHost, c.portgatePort)
}

// TargetAddress is the address of the destination server.
func (c *Config) TargetAddress(port int) string {
	return fmt.Sprintf("%s:%d", c.targetHost, port)
}