mirror of https://github.com/fluffle/goirc
120 lines
2.3 KiB
Go
120 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
"irc"
|
|
"github.com/kless/goconfig/config"
|
|
)
|
|
|
|
const authFile = "auth.conf"
|
|
var auth *config.Config
|
|
|
|
func readAuth() {
|
|
var err os.Error;
|
|
auth, err = config.ReadDefault(authFile)
|
|
if (err != nil) {
|
|
panic(fmt.Sprintf("Auth config error: %s", err))
|
|
}
|
|
}
|
|
|
|
func addAccess(conn *irc.Conn, channel, nick, flags string) (string, string) {
|
|
n := conn.GetNick(nick)
|
|
if n == nil {
|
|
return "", ""
|
|
}
|
|
|
|
section := conn.Network + " " + channel
|
|
cflags, _ := auth.String(section, n.Host)
|
|
|
|
nflags := cflags
|
|
for _, flag := range flags {
|
|
if strings.IndexRune(cflags, flag) > -1 {
|
|
// already has the flag
|
|
continue
|
|
}
|
|
nflags += string(flag)
|
|
}
|
|
|
|
auth.AddOption(section, n.Host, nflags)
|
|
if updateAuth() != nil {
|
|
say(conn, channel, "Error while writing to %s", authFile)
|
|
}
|
|
|
|
return n.Host, nflags
|
|
}
|
|
|
|
func removeAccess(conn *irc.Conn, channel, nick, flags string) (string, string) {
|
|
n := conn.GetNick(nick)
|
|
if n == nil {
|
|
return "", ""
|
|
}
|
|
|
|
section := conn.Network + " " + channel
|
|
cflags, _ := auth.String(section, n.Host)
|
|
|
|
nflags := ""
|
|
for _, flag := range cflags {
|
|
if strings.IndexRune(flags, flag) < 0 {
|
|
// we're not removing this flag
|
|
nflags += string(flag)
|
|
}
|
|
}
|
|
|
|
auth.AddOption(section, n.Host, nflags)
|
|
if updateAuth() != nil {
|
|
say(conn, channel, "Error while writing to %s", authFile)
|
|
}
|
|
|
|
return n.Host, nflags
|
|
}
|
|
|
|
func removeUser(conn *irc.Conn, channel, nick string) (string, bool) {
|
|
n := conn.GetNick(nick)
|
|
if n == nil {
|
|
return "", false
|
|
}
|
|
|
|
section := conn.Network + " " + channel
|
|
|
|
if !auth.RemoveOption(section, n.Host) {
|
|
return n.Host, false
|
|
}
|
|
if updateAuth() != nil {
|
|
say(conn, channel, "Error while writing to %s", authFile)
|
|
}
|
|
|
|
return n.Host, true
|
|
}
|
|
|
|
// passing a flag of "" will check if the user has any access
|
|
func hasAccess(conn *irc.Conn, channel, nick, flag string) bool {
|
|
n := conn.GetNick(nick)
|
|
if n == nil {
|
|
return false
|
|
}
|
|
|
|
if owner, _ := auth.String(conn.Network, "owner"); owner == n.Host {
|
|
return true
|
|
}
|
|
|
|
flags, err := auth.String(conn.Network + " " + channel, n.Host)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
if strings.Index(flags, flag) > -1 {
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
func updateAuth() os.Error {
|
|
if err := auth.WriteFile(authFile, 0644, ""); err != nil {
|
|
return err
|
|
}
|
|
// config.WriteFile destroys the config, so
|
|
readAuth()
|
|
return nil
|
|
}
|