#!/usr/bin/env ruby

require "tty-prompt"
require "tty-screen"

def sep
    puts
    puts "~" * TTY::Screen.width
    puts
end

prompt = TTY::Prompt.new

system "clear"

# welcome
system "figlet -f slant tilde.team"
puts
puts "welcome to tilde.team!!"
puts
puts "we're glad you're here!"
puts "let's walk through some basic questions to get you set up"

prompt.keypress("ready? press enter to continue")
puts

# change your password
sep
puts "step 1:"
puts "  first, let's change your shell password"
puts
puts "  you'll find the temporary password in your welcome email"
puts "  enter the current password once, followed by your new password twice"
system "passwd"

# select your shell
sep
puts "step 2:"
puts "  now, let's pick your default shell"
puts
puts "  a shell is a program that handles commands you type"
puts "  bash is the most common shell and is a good place to start"
puts "  note that the list of shells extends beyond one page"
puts

shells = File.readlines("/etc/shells")
    .select { |line| !line.start_with?("#") }
    .map { |line| line.chomp }
    .map { |line| [File.basename(line), line] }
    .to_h

shell = prompt.select("  which shell would you like to use?", shells, per_page: shells.count)
puts
puts "  great, you've picked #{shell}!"
puts "  in order to change your shell, you'll have to enter your password again"
system "chsh -s #{shell}"

# byobu or not
sep
puts "step 3:"
puts "  we recommend using a terminal multiplexer, which is a tool that allows you"
puts "  to have tabs in your shell and even disconnect while leaving things running"
puts "  as you left them."
puts
puts "  the tool we recommend is byobu: https://superuser.com/a/423397/866501"
puts
puts "  if you're not sure about this, decline for now. you can set it up at any time"
puts "  later on by running 'byobu-enable' from your shell."
puts

enable_byobu = prompt.yes?("  would you like to set byobu to launch automatically when you log in?")

if enable_byobu
    system "byobu-enable"
    puts "our default configs will connect you to chat and open a mail client if you choose to enable it"
end

# tz
sep
puts "step 4:"
puts "  great, let's set up your timezone!"
puts
tz = %x{tzselect}
puts
puts "  you selected #{tz}, adding this to your ~/.profile now"
puts "  it might not take effect until you log out and back in"
open(".profile", "a") { |f| f.puts "export TZ='#{tz}'" }

# pronouns
sep
puts "step 5:"
pronouns = prompt.ask("  what are your preferred pronouns?")
puts "  saving your pronouns to your ~/.pronouns file."
puts "  feel free to update it as needed!"
open(".pronouns", "w") { |f| f.puts pronouns }


# welcome completed
sep
puts "welcome to the ~team!"
puts
puts "please come stop by chat when you get a chance by running the 'chat' command" unless enable_byobu
puts "we're happy to help as needed and get you any information you're looking for"
puts "have a look at our wiki: https://tilde.team/wiki/ (ctrl-click will let you open that from here)"
File.delete(".new_user")

if enable_byobu
    exec "byobu"
end