"""Main module for the game"""

import sys
import pygame
from pygame.locals import SWSURFACE

from mamba.constants import SCREEN
from mamba.options import options, parse_args, check_args
from mamba.engine import Engine
from mamba.sound import SoundSystem
from mamba.habitats.mainmenu import MainMenu
from mamba.habitats.level import LevelHabitat
from mamba.habitats.editor import EditorHabitat


def main():
    """Launch the currently unnamed mamab game"""
    parse_args(sys.argv)
    if not check_args():
        sys.exit(1)

    SoundSystem(options.sound)
    pygame.display.init()
    pygame.font.init()
    pygame.display.set_mode(SCREEN, SWSURFACE)
    pygame.display.set_caption('Mamba')

    if options.edit:
        start = EditorHabitat(options.level)
    elif options.level is not None:
        start = LevelHabitat(options.level)
    else:
        start = MainMenu()

    engine = Engine()
    engine.set_habitat(start)
    engine.run()