// Copyright 2012 The Rust Project Developers. See the COPYRIGHT
// file at the top-level directory of this distribution and at
// http://rust-lang.org/COPYRIGHT.
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.

/// Correctness for protocols

/*

This section of code makes sure the protocol is likely to generate
correct code. The correctness criteria include:

  * No protocols transition to states that don't exist.
  * Messages step to states with the right number of type parameters.

In addition, this serves as a lint pass. Lint warns for the following
things.

  * States with no messages, it's better to step to !.

It would also be nice to warn about unreachable states, but the
visitor infrastructure for protocols doesn't currently work well for
that.

*/

use ast;
use codemap::span;
use ext::base::ext_ctxt;
use ext::pipes::proto::{state, protocol, next_state};
use ext::pipes::proto;

impl proto::visitor<(), (), ()> for @ext_ctxt {
    fn visit_proto(&self, _proto: protocol, _states: &[()]) { }

    fn visit_state(&self, state: state, _m: &[()]) {
        let messages = &*state.messages;
        if messages.len() == 0 {
            self.span_warn(
                state.span, // use a real span!
                fmt!("state %s contains no messages, \
                      consider stepping to a terminal state instead",
                      state.name))
        }
    }

    fn visit_message(&self, name: ~str, _span: span, _tys: &[@ast::Ty],
                     this: state, next: Option<next_state>) {
        match next {
          Some(ref next_state) => {
            let proto = this.proto;
            if !proto.has_state(next_state.state) {
                // This should be a span fatal, but then we need to
                // track span information.
                self.span_err(
                    proto.get_state(next_state.state).span,
                    fmt!("message %s steps to undefined state, %s",
                         name, next_state.state));
            }
            else {
                let next = proto.get_state(next_state.state);

                if next.generics.ty_params.len() != next_state.tys.len() {
                    self.span_err(
                        next.span, // use a real span
                        fmt!("message %s target (%s) \
                              needs %u type parameters, but got %u",
                             name, next.name,
                             next.generics.ty_params.len(),
                             next_state.tys.len()));
                }
            }
          }
          None => ()
        }
    }
}