modules.go

// Copyright 2020 The jackal 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 main

import (
	"github.com/ortuman/jackal/pkg/module"
	"github.com/ortuman/jackal/pkg/module/offline"
	"github.com/ortuman/jackal/pkg/module/roster"
	"github.com/ortuman/jackal/pkg/module/xep0012"
	"github.com/ortuman/jackal/pkg/module/xep0030"
	"github.com/ortuman/jackal/pkg/module/xep0049"
	"github.com/ortuman/jackal/pkg/module/xep0054"
	"github.com/ortuman/jackal/pkg/module/xep0092"
	"github.com/ortuman/jackal/pkg/module/xep0115"
	"github.com/ortuman/jackal/pkg/module/xep0191"
	"github.com/ortuman/jackal/pkg/module/xep0198"
	"github.com/ortuman/jackal/pkg/module/xep0199"
	"github.com/ortuman/jackal/pkg/module/xep0202"
	"github.com/ortuman/jackal/pkg/module/xep0280"
)

var defaultModules = []string{
	roster.ModuleName,
	offline.ModuleName,
	xep0012.ModuleName,
	xep0030.ModuleName,
	xep0049.ModuleName,
	xep0054.ModuleName,
	xep0092.ModuleName,
	xep0115.ModuleName,
	xep0191.ModuleName,
	xep0198.ModuleName,
	xep0199.ModuleName,
	xep0280.ModuleName,
}

var modFns = map[string]func(a *serverApp, cfg modulesConfig) module.Module{
	// Roster
	// (https://xmpp.org/rfcs/rfc6121.html#roster)
	roster.ModuleName: func(a *serverApp, _ modulesConfig) module.Module {
		return roster.New(a.router, a.hosts, a.resMng, a.rep, a.hk)
	},
	// Offline
	// (https://xmpp.org/extensions/xep-0160.html)
	offline.ModuleName: func(a *serverApp, cfg modulesConfig) module.Module {
		return offline.New(a.router, a.hosts, a.resMng, a.rep, a.locker, a.hk, offline.Config{
			QueueSize: cfg.Offline.QueueSize,
		})
	},
	// XEP-0012: Last Activity
	// (https://xmpp.org/extensions/xep-0012.html)
	xep0012.ModuleName: func(a *serverApp, _ modulesConfig) module.Module {
		return xep0012.New(a.router, a.hosts, a.resMng, a.rep, a.hk)
	},
	// XEP-0030: Service Discovery
	// (https://xmpp.org/extensions/xep-0030.html)
	xep0030.ModuleName: func(a *serverApp, _ modulesConfig) module.Module {
		return xep0030.New(a.router, a.comps, a.rep, a.resMng, a.hk)
	},
	// XEP-0049: Private XML Storage
	// (https://xmpp.org/extensions/xep-0049.html)
	xep0049.ModuleName: func(a *serverApp, _ modulesConfig) module.Module {
		return xep0049.New(a.router, a.rep, a.hk)
	},
	// XEP-0054: vcard-temp
	// (https://xmpp.org/extensions/xep-0054.html)
	xep0054.ModuleName: func(a *serverApp, _ modulesConfig) module.Module {
		return xep0054.New(a.router, a.rep, a.hk)
	},
	// XEP-0092: Software Version
	// (https://xmpp.org/extensions/xep-0092.html)
	xep0092.ModuleName: func(a *serverApp, cfg modulesConfig) module.Module {
		return xep0092.New(a.router, xep0092.Config{
			ShowOS: cfg.Version.ShowOS,
		})
	},
	// XEP-0115: Entity Capabilities
	// (https://xmpp.org/extensions/xep-0115.html)
	xep0115.ModuleName: func(a *serverApp, _ modulesConfig) module.Module {
		return xep0115.New(a.router, a.rep, a.hk)
	},
	// XEP-0191: Blocking Command
	// (https://xmpp.org/extensions/xep-0191.html)
	xep0191.ModuleName: func(a *serverApp, _ modulesConfig) module.Module {
		return xep0191.New(a.router, a.hosts, a.resMng, a.rep, a.hk)
	},
	// XEP-0198: Stream Management
	// (https://xmpp.org/extensions/xep-0198.html)
	xep0198.ModuleName: func(a *serverApp, cfg modulesConfig) module.Module {
		return xep0198.New(a.router, a.hosts, a.resMng, a.hk, xep0198.Config{
			HibernateTime:      cfg.Stream.HibernateTime,
			RequestAckInterval: cfg.Stream.RequestAckInterval,
			WaitForAckTimeout:  cfg.Stream.WaitForAckTimeout,
			MaxQueueSize:       cfg.Stream.MaxQueueSize,
		})
	},
	// XEP-0199: XMPP Ping
	// (https://xmpp.org/extensions/xep-0199.html)
	xep0199.ModuleName: func(a *serverApp, cfg modulesConfig) module.Module {
		return xep0199.New(a.router, a.hk, xep0199.Config{
			AckTimeout:    cfg.Ping.AckTimeout,
			Interval:      cfg.Ping.Interval,
			SendPings:     cfg.Ping.SendPings,
			TimeoutAction: cfg.Ping.TimeoutAction,
		})
	},
	// XEP-0202: Entity Time
	// (https://xmpp.org/extensions/xep-0202.html)
	xep0202.ModuleName: func(a *serverApp, _ modulesConfig) module.Module {
		return xep0202.New(a.router)
	},
	// XEP-0280: Message Carbons
	// (https://xmpp.org/extensions/xep-0280.html)
	xep0280.ModuleName: func(a *serverApp, _ modulesConfig) module.Module {
		return xep0280.New(a.router, a.hosts, a.resMng, a.hk)
	},
}