workshop-linux/Vagrantfile at main · opensourcecorp/workshop-linux · GitHub

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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
# Allows for setting an env var for fewer (or more) team servers, for different
# testing scenarios & hardware
num_team_servers = Integer(ENV["num_team_servers"].nil? ? 2 : ENV["num_team_servers"])

Vagrant.configure("2") do |config|
  box = "debian/bookworm64" # Debian 12

  cpus   = 1
  memory = 768

  config.vm.provider "virtualbox" do |vb|
    vb.cpus   = cpus
    vb.memory = memory
  end

  config.vm.provider "libvirt" do |lv|
    lv.cpus   = cpus
    lv.memory = memory
  end

  if Vagrant.has_plugin?("vagrant-vbguest")
    config.vbguest.auto_update = false
  end

  # DB server static IP
  db_addr = "10.0.1.10"
  ip_bytes = db_addr.split(".")

  config.vm.define "db" do |db|
    db.vm.box = box

    db.vm.network "private_network", ip: db_addr
    db.vm.network "forwarded_port", guest: 5432, host: 5432, protocol: "tcp" # DB
    db.vm.network "forwarded_port", guest: 8000, host: 8000, protocol: "tcp" # Dummy web app
    db.vm.network "forwarded_port", guest: 8080, host: 8080, protocol: "tcp" # Score dashboard

    db.vm.synced_folder ".", "/vagrant", disabled: true

    db.vm.provision "file", source: "./scripts", destination: "/tmp/scripts"
    db.vm.provision "file", source: "./services", destination: "/tmp/services"
    db.vm.provision "file", source: "./score-server", destination: "/tmp/score-server"
    db.vm.provision "file", source: "./dummy-web-app", destination: "/tmp/dummy-web-app"

    db.vm.provision "shell",
      inline: <<-SCRIPT
        #!/usr/bin/env bash
        set -euo pipefail

        # Need both running here for Vagrant -- other platforms should ONLY have 2332
        sudo sh -c 'grep -q 2332 /etc/ssh/sshd_config || printf "Port 2332\nPort 22\n" >> /etc/ssh/sshd_config'
        sudo systemctl restart ssh

        rm -rf /root/{score-server,services,dummy-web-app}
        sudo cp -r /tmp/{score-server,services,dummy-web-app} /root/
        bash /tmp/scripts/init-db.sh
      SCRIPT
  end

  (1..num_team_servers).each do |i|
    config.vm.define "team#{i}" do |team|
      team.vm.box = box

      # TODO: this looks as gross as it does b/c maybe we'll loop over it later
      # when we're simulating more teams
      team.vm.network "private_network", ip: ip_bytes[0..2].join(".") + "." + String(Integer(ip_bytes[-1]) + i)

      team.vm.synced_folder ".", "/vagrant", disabled: true

      team.vm.provision "file", source: "./scripts", destination: "/tmp/scripts"
      team.vm.provision "file", source: "./services", destination: "/tmp/services"
      team.vm.provision "file", source: "./instructions", destination: "/tmp/instructions"
      team.vm.provision "file", source: "./dummy-app-src", destination: "/tmp/dummy-app-src"

      team.vm.provision "shell",
        inline: <<-SCRIPT
          #!/usr/bin/env bash
          set -euo pipefail

          # Need both running here for Vagrant -- other platforms should ONLY have 2332
          sudo sh -c 'grep -q 2332 /etc/ssh/sshd_config || printf "Port 2332\nPort 22\n" >> /etc/ssh/sshd_config'
          sudo systemctl restart ssh

          export team_name="Team-#{i}"
          export db_addr='#{db_addr}'
          bash /tmp/scripts/init.sh
          bats -F pretty /.ws/scripts/test.bats
        SCRIPT
    end
  end
end