antiroachlol/cmd/swarm/main.go

package main

import (
	"context"
	"fmt"
	"io/ioutil"
	"log"
	"math/rand"
	"net/http"
	"os"
	"strconv"
	"strings"
	"time"

	"golang.org/x/net/proxy"
	irc "gopkg.in/irc.v3"
)

var insults = []string{"Leopard", "SailfinMolly", "NarwhalMan", "BluejayMan", "GiantCaveSwallow", "GreatHornedOwl", "SnappingTurtleMan", "Wolverine", "GiantAardvark", "SeaLamprey", "GrasshopperMan", "WhitebrowedGibbon", "GiantLouse", "RedwingedBlackbird", "Fairy", "LorikeetMan", "GiantSkunk", "GiantAlligator", "CaveFish", "Mosquito", "GiantFox", "SaltwaterCrocodileMan", "GiantOsprey", "Lion", "WeaselMan", "Human", "Swordfish", "ClownLoach", "DragonflyMan", "GiantMosquito", "SpongeMan", "GiantCopperheadSnake", "GiantBeetle", "KestrelMan", "BullShark", "AardvarkMan", "PeachfacedLovebirdMan", "GiantThrips", "Capybara", "Rutherer", "MaskedLovebirdMan", "KnuckleWorm", "OnehumpedCamelMan", "Deer", "Cassowary", "BaskingShark", "Wombat", "GiantSkink", "GiantMongoose", "EchidnaMan", "ParakeetMan", "HareMan", "BlackcrestedGibbon", "VultureMan", "Slug", "BushmasterMan", "GiantBobcat", "GiganticPanda", "CapybaraMan", "MagmaMan", "SquidMan", "GiantCrab", "ChinchillaMan", "GiantOstrich", "GiantWhiteStork", "GiantDeer", "RedPanda", "Siamang", "Dwarf", "GiantLeopard", "Bobcat", "Duck", "TwohumpedCamelMan", "GiantBeaver", "Troglodyte", "GiantRattlesnake", "RoachMan", "Satyr", "Albatross", "CheetahMan", "SpottedRatfish", "AdderMan", "GrayLangur", "CockatielMan", "GiantMuskox", "GiantKea", "CaveBlob", "GiantWren", "SeaNettleJellyfish", "GiantMoth", "GiantDingo", "SpiderMonkeyMan", "Oyster", "GiantMaskedLovebird", "GiantSaltwaterCrocodile", "GiantKoala", "BrownRecluseSpiderMan", "MoonSnailMan", "GiantMagpie", "FoxSquirrel", "GiantLionTamarin", "Halibut", "Clownfish", "BobcatMan", "Groundhog", "DarkGnome", "GiantCapuchin", "GiantPlatypus", "HamsterMan", "BrownRecluseSpider", "Harpy", "BloodGnat", "Molemarian", "GroundhogMan", "GiantGrayLangur", "Beetle", "GiantChinchilla", "Buzzard", "BlackMamba", "BeakDog", "Magpie", "DesertTortoiseMan", "ChameleonMan", "IguanaMan", "TigerMan", "BarkScorpion", "GiantAlbatross", "GiantTiger", "Guppy", "GiantSnappingTurtle", "Mantis", "Raccoon", "Cardinal", "RhesusMacaque", "RatMan", "GiantOcelot", "GiantGrizzlyBear", "MountainGoatMan", "Carp", "Beaver", "Tiger", "GiantOtter", "ToadMan", "GreatBarracuda", "MonarchButterfly", "Thrips", "SlugMan", "GiantCaveToad", "GreenTreeFrog", "Moghopper", "GiantKingsnake", "KingCobraMan", "SpermWhaleMan", "Capuchin", "PondGrabber", "MooseCow", "Vulture", "RhinocerosMan", "BlackMambaMan", "GiantBat", "GiantAxolotl", "WhitetipReefShark", "GiantOrca", "Louse", "GrayLangurMan", "Snail", "Tapir", "Herring", "Nautilus", "SeaOtter", "Anole", "SkunkMan", "TwohumpedCamel", "GiantMooseBull", "RedPandaMan", "Damselfly", "Orca", "FlyingSquirrel", "KoalaMan", "GreenDevourer", "WildBoar", "Ibex", "GiantLion", "SeaSerpent", "ReptileMan", "LizardMan", "GiantNarwhal", "GiantMandrill", "Perch", "Oriole", "Hamster", "GiantRat", "GiantRaven", "Wagon", "GiantLizard", "GiantCougar", "Axolotl", "GiantWeasel", "WrenMan", "Creature", "GiantCardinal", "Kobold", "GiantEarthworm", "Cougar", "SnowyOwl", "BarnOwlMan", "PuffinMan", "GiantCoyote", "Hake", "GiantNautilus", "HornbillMan", "Bilou", "LeopardSealMan", "GreyParrotMan", "Unicorn", "GiantOpossum", "Stingray", "Bat", "Kangaroo", "GiantMooseCow", "PurringMaggot", "Impala", "CaveFloater", "Horse", "SwanMan", "Gazelle", "FoulBlendec", "LouseMan", "Elk", "Llama", "PorcupineMan", "PeregrineFalcon", "Lorikeet", "NakedMoleDog", "CaveSpider", "MagpieMan", "CrabMan", "GiantGilaMonster", "GiantCapybara", "Lizard", "HoaryMarmotMan", "WolfMan", "Kingsnake", "CapuchinMan", "GiantAnaconda", "GiantHornbill", "GiantToad", "NautilusMan", "GiantGraySquirrel", "TapirMan", "GiantSnowyOwl", "Bushtit", "GiantWalrus", "GiantOriole", "GiantParakeet", "HoneyBadger", "Cyclops", "SpinyDogfish", "Coati", "GiantRedSquirrel", "WhitespottedPuffer", "OspreyMan", "MothMan", "Chimpanzee", "Mink", "Walrus", "Tigerfish", "GiantDesertTortoise", "DeerMan", "BlueJay", "Sole", "Cat", "ElkMan", "Cow", "IceWolf", "Fly", "Crow", "BatRay", "MagmaCrab", "Angelshark", "Anchovy", "MonarchButterflyMan", "Bonobo", "BarnOwl", "GiganticTortoise", "Seahorse", "Badger", "GilaMonsterMan", "Hydra", 
var friends = []string{}
var s = rand.NewSource(time.Now().UnixNano()) // initialize global pseudo random generator
var r = rand.New(s)
var channel string
var victim string
var is_connected = false
var op_mode = false
var greetChance = 100
var suffixLength = 0

func loadFriends() {
	friendsV := os.Getenv("IRC_EXTRAFRIENDS")
	if friendsV == "" {
		return
	}
	for _, name := range strings.Split(strings.TrimSpace(friendsV), ",") {
		name := strings.TrimSpace(name)
		if name != "" {
			log.Printf("loaded extra friend: %s", name)
			friends = append(friends, name)
		}
	}
}

func loadNames(fpath string) {
	content, err := ioutil.ReadFile(fpath)
	if err != nil {
		panic(err)
	}
	insults = []string{}
	for _, line := range strings.Split(string(content), "\n") {
		stripped := strings.TrimSpace(line)
		if stripped != "" {
			log.Printf("loaded name: %s", stripped)
			insults = append(insults, stripped)
		}
	}
	if len(insults) == 0 {
		panic(fmt.Errorf("no names loaded from %s", fpath))
	}
}

func getNumericVar(key string, default_ int) int {
	varS := os.Getenv(key)
	if varS != "" {
		var err error
		default_, err = strconv.Atoi(varS)
		if err != nil {
			panic(fmt.Errorf("invalid %s: '%s'", key, varS))
		}
	}
	return default_
}

func friendlyNick(nick string) bool {
	for _, n := range friends {
		if nick == n {
			return true
		}
	}
	for _, n := range insults {
		if strings.HasPrefix(nick, n) {
			return true
		}
	}
	if strings.HasPrefix(nick, "Op") {
		return friendlyNick(nick[2:])
	}
	return false
}

func genNick() string {
	n := insults[r.Intn(len(insults))] + trailingDigits(suffixLength)
	if op_mode {
		n = "Op" + n
	}
	return n
}

func trailingDigits(length int) string {
	trailing := ""
	for ; length > 0; length-- {
		trailing = fmt.Sprintf("%s%d", trailing, r.Intn(10))
	}
	return trailing
}

func runStatusHandler() {
	statusPort := getNumericVar("HEALTH_PORT", 8000)
	if statusPort < 0 || statusPort > 65535 {
		panic(fmt.Errorf("invalid HEALTH_PORT: %d", statusPort))
	}
	healthAddress := fmt.Sprintf("0.0.0.0:%d", statusPort)
	log.Printf("serving health on %s", healthAddress)
	http.HandleFunc("/health", statusHandler())
	log.Fatal(http.ListenAndServe(healthAddress, nil))
}

func statusHandler() func(http.ResponseWriter, *http.Request) {
	return func(w http.ResponseWriter, r *http.Request) {
		if !is_connected {
			http.Error(w, "Not connected", http.StatusInternalServerError)
			return
		}
		w.Write([]byte("ok"))
	}
}

func HandleRequest(ctx context.Context) (string, error) {
	// So i was too lazy to get rid of the boilerplate
	// Set up IRC connection
	server := os.Getenv("IRC_SERVER")
	if server == "" {
		log.Fatal("Set IRC_SERVER!")
	}
	channel = os.Getenv("IRC_CHANNEL")
	if channel == "" {
		log.Fatal("Set IRC_CHANNEL!")
	}
	victim = os.Getenv("IRC_VICTIM")
	if victim == "" {
		log.Fatal("Set IRC_VICTIM!")
	}
	enable_op_mode := os.Getenv("IRC_OP_MODE")
	if enable_op_mode != "" {
		op_mode = true
	}

	suffixLength = getNumericVar("SUFFIX_DIGITS", 0)
	if suffixLength < 0 || suffixLength > 10 {
		panic(fmt.Errorf("invalid SUFFIX_DIGITS: %d", suffixLength))
	}

	log.Println("dialing...")

	dialCtx, cancelDial := context.WithTimeout(context.Background(), time.Second*10)
	defer cancelDial()

	conn, err := proxy.Dial(dialCtx, "tcp", server)
	if err != nil {
		log.Fatal(err)
	}
	hostname, _ := os.Hostname()
	config := irc.ClientConfig{
		Nick:    genNick(),
		User:    hostname,
		Name:    "dabbing on " + victim,
		Handler: irc.HandlerFunc(handleIRCCon), // See below for callback
	}
	log.Println("starting client...")
	client := irc.NewClient(conn, config)
	err = client.Run()
	return fmt.Sprintf("i died"), err
}

func handleIRCCon(c *irc.Client, m *irc.Message) {
	log.Printf("%+v\n", m)
	sent := 0

	// Only one reader, prob not necessary
	// Handle connection established
	if m.Command == "001" {
		is_connected = true
		// 001 is welcome event
		c.Write("JOIN " + channel)
	} else if m.Command == "PRIVMSG" && c.FromChannel(m) {
		// Handle chat message to print stats
		text := strings.Split(m.Trailing(), " ")
		if text[0] == c.CurrentNick()+":" && text[len(text)-1] == "stats" {
			c.WriteMessage(&irc.Message{
				Command: "PRIVMSG",
				Params: []string{
					m.Params[0],
					"I've sent " + strconv.FormatInt(int64(sent), 10) + " invites to " + victim + " :)",
				},
			})
		} else if text[len(text)-1] == "name" {
			nick := genNick()
			c.Write("NICK " + nick)
		} else if text[len(text)-1] == "invite" {
			c.Write("INVITE " + victim + " " + channel)
			sent++
		} /* else if strings.Contains(m.Trailing(), "greetings") {
			c.WriteMessage(&irc.Message{
				Command: "PRIVMSG",
				Params: []string{
					m.Params[0],
					victim + ": hi",
				},
			})
		}*/
	} else if m.Command == "JOIN" && c.FromChannel(m) {
		if m.Prefix.Name != c.CurrentNick() {
			if greetChance == 1000 || r.Intn(1000) < greetChance {
				c.WriteMessage(&irc.Message{
					Command: "PRIVMSG",
					Params: []string{
						m.Params[0],
						"greetings " + m.Prefix.Name,
					},
				})
			}
			if op_mode && friendlyNick(m.Prefix.Name) {
				log.Printf("opping %s", m.Prefix.Name)
				c.WriteMessage(&irc.Message{
					Command: "MODE",
					Params: []string{
						m.Params[0],
						"+o",
						m.Prefix.Name,
					},
				})
			}
		}
	}
}

func main() {
	greetChance = getNumericVar("GREET_CHANCE", 100)
	if greetChance > 100 || greetChance < 0 {
		log.Fatalf("invalid GREET_CHANCE: '%s'", greetChance)
	}

	namesPath := os.Getenv("NAMES_FPATH")
	if namesPath != "" {
		loadNames(namesPath)
	}
	loadFriends()

	go runStatusHandler()
	ctx, _ := context.WithCancel(context.Background())
	ret, err := HandleRequest(ctx)
	if err != nil {
		log.Fatalln(err)
	}
	log.Println(ret)
}