#! /usr/bin/python3
# Copyright (C) 2018-2020 Merspieler, merspieler _at_ airmail.cc
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License as
# published by the Free Software Foundation; either version 3 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

import socket
import re
import sys
from common import send_status, get_status

host = socket.gethostname()
port = 12345
tile = ""
status = ""
verbose = False

argc = len(sys.argv)
first = True
i = 1
while i < argc:
	if sys.argv[i] == "--port":
		i += 1
		port = int(sys.argv[i])
	elif sys.argv[i] == "--host":
		i += 1
		host = sys.argv[i]
	elif sys.argv[i] == "-v" or sys.argv[i] == "--verbose":
		verbose = True
	elif sys.argv[i] == "-h" or sys.argv[i] == "--help":
		print("usage: flag-rebuild.py <file> [OPTIONS]")
		print("Flags tiles for rebuild based on terrasync log file")
		print("")
		print("  <file>            Terrasync log file")
		print("OPTIONS")
		print("    , --host        Manager host")
		print("    , --port        Manager port")
		print("  -v, --verbose     Verbose printouts")
		print("  -h, --help        Shows this help and exit")
		sys.exit(0)
	else:
		if first:
			lfile = sys.argv[i]
		else:
			print("Unknown option " + sys.argv[i])
			sys.exit(1)
	i += 1

if lfile == "":
	print("ERROR: No file given")
	sys.exit(1)

try:
	tiles = []
	with open(lfile) as f:
		for line in f:
			match = re.match(r".*[ew]\d{3}[ns]\d{2}/[ew]\d{3}[ns]\d{2}/([0-9]{1,7})\.[bs]tg", line)
			if match != None:
				tiles.append(match.group(1))
except IOError:
	print("ERROR: Failed to read file")
	sys.exit(1)

tiles = set(tiles)

for tile in tiles:
	status = get_status(tile, host, port)
	if status == "done" or status == "packaged":
		if verbose:
			print("Flagging " + tile)
		send_status(tile, "rebuild", host, port)
	elif verbose:
		print("Skipping " + tile + ". Not yet build")