module ForemanSalt::StateImporter

def add_to_environment(states, environment)
  environment = SaltEnvironment.where(name: environment).first_or_create
  states.each do |state_name|
    state = SaltModule.where(name: state_name).first_or_create
    state.salt_environments << environment unless state.salt_environments.include? environment
  end
end