// 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.

pub enum Fold<T> = Fold_<T>;

impl<T: Clone> Fold<T>: Clone {
    fn clone(&self) -> Fold<T> {
        Fold({
            ctxt: self.ctxt.clone(),
            fold_doc: copy self.fold_doc,
            fold_crate: copy self.fold_crate,
            fold_item: copy self.fold_item,
            fold_mod: copy self.fold_mod,
            fold_nmod: copy self.fold_nmod,
            fold_fn: copy self.fold_fn,
            fold_const: copy self.fold_const,
            fold_enum: copy self.fold_enum,
            fold_trait: copy self.fold_trait,
            fold_impl: copy self.fold_impl,
            fold_type: copy self.fold_type,
            fold_struct: copy self.fold_struct
        })
    }
}

type FoldDoc<T> = fn~(fold: &Fold<T>, +doc: doc::Doc) -> doc::Doc;
type FoldCrate<T> = fn~(fold: &Fold<T>, +doc: doc::CrateDoc) -> doc::CrateDoc;
type FoldItem<T> = fn~(fold: &Fold<T>, +doc: doc::ItemDoc) -> doc::ItemDoc;
type FoldMod<T> = fn~(fold: &Fold<T>, +doc: doc::ModDoc) -> doc::ModDoc;
type FoldNmod<T> = fn~(fold: &Fold<T>, +doc: doc::NmodDoc) -> doc::NmodDoc;
type FoldFn<T> = fn~(fold: &Fold<T>, +doc: doc::FnDoc) -> doc::FnDoc;
type FoldConst<T> = fn~(fold: &Fold<T>, +doc: doc::ConstDoc) -> doc::ConstDoc;
type FoldEnum<T> = fn~(fold: &Fold<T>, +doc: doc::EnumDoc) -> doc::EnumDoc;
type FoldTrait<T> = fn~(fold: &Fold<T>, +doc: doc::TraitDoc) -> doc::TraitDoc;
type FoldImpl<T> = fn~(fold: &Fold<T>, +doc: doc::ImplDoc) -> doc::ImplDoc;
type FoldType<T> = fn~(fold: &Fold<T>, +doc: doc::TyDoc) -> doc::TyDoc;
type FoldStruct<T> = fn~(fold: &Fold<T>,
                         +doc: doc::StructDoc) -> doc::StructDoc;

type Fold_<T> = {
    ctxt: T,
    fold_doc: FoldDoc<T>,
    fold_crate: FoldCrate<T>,
    fold_item: FoldItem<T>,
    fold_mod: FoldMod<T>,
    fold_nmod: FoldNmod<T>,
    fold_fn: FoldFn<T>,
    fold_const: FoldConst<T>,
    fold_enum: FoldEnum<T>,
    fold_trait: FoldTrait<T>,
    fold_impl: FoldImpl<T>,
    fold_type: FoldType<T>,
    fold_struct: FoldStruct<T>
};


// This exists because fn types don't infer correctly as record
// initializers, but they do as function arguments
fn mk_fold<T:Clone>(
    +ctxt: T,
    +fold_doc: FoldDoc<T>,
    +fold_crate: FoldCrate<T>,
    +fold_item: FoldItem<T>,
    +fold_mod: FoldMod<T>,
    +fold_nmod: FoldNmod<T>,
    +fold_fn: FoldFn<T>,
    +fold_const: FoldConst<T>,
    +fold_enum: FoldEnum<T>,
    +fold_trait: FoldTrait<T>,
    +fold_impl: FoldImpl<T>,
    +fold_type: FoldType<T>,
    +fold_struct: FoldStruct<T>
) -> Fold<T> {
    Fold({
        ctxt: move ctxt,
        fold_doc: move fold_doc,
        fold_crate: move fold_crate,
        fold_item: move fold_item,
        fold_mod: move fold_mod,
        fold_nmod: move fold_nmod,
        fold_fn: move fold_fn,
        fold_const: move fold_const,
        fold_enum: move fold_enum,
        fold_trait: move fold_trait,
        fold_impl: move fold_impl,
        fold_type: move fold_type,
        fold_struct: move fold_struct
    })
}

pub fn default_any_fold<T:Send Clone>(+ctxt: T) -> Fold<T> {
    mk_fold(
        move ctxt,
        |f, d| default_seq_fold_doc(f, d),
        |f, d| default_seq_fold_crate(f, d),
        |f, d| default_seq_fold_item(f, d),
        |f, d| default_any_fold_mod(f, d),
        |f, d| default_any_fold_nmod(f, d),
        |f, d| default_seq_fold_fn(f, d),
        |f, d| default_seq_fold_const(f, d),
        |f, d| default_seq_fold_enum(f, d),
        |f, d| default_seq_fold_trait(f, d),
        |f, d| default_seq_fold_impl(f, d),
        |f, d| default_seq_fold_type(f, d),
        |f, d| default_seq_fold_struct(f, d)
    )
}

pub fn default_seq_fold<T:Clone>(+ctxt: T) -> Fold<T> {
    mk_fold(
        move ctxt,
        |f, d| default_seq_fold_doc(f, d),
        |f, d| default_seq_fold_crate(f, d),
        |f, d| default_seq_fold_item(f, d),
        |f, d| default_seq_fold_mod(f, d),
        |f, d| default_seq_fold_nmod(f, d),
        |f, d| default_seq_fold_fn(f, d),
        |f, d| default_seq_fold_const(f, d),
        |f, d| default_seq_fold_enum(f, d),
        |f, d| default_seq_fold_trait(f, d),
        |f, d| default_seq_fold_impl(f, d),
        |f, d| default_seq_fold_type(f, d),
        |f, d| default_seq_fold_struct(f, d)
    )
}

pub fn default_par_fold<T:Send Clone>(+ctxt: T) -> Fold<T> {
    mk_fold(
        move ctxt,
        |f, d| default_seq_fold_doc(f, d),
        |f, d| default_seq_fold_crate(f, d),
        |f, d| default_seq_fold_item(f, d),
        |f, d| default_par_fold_mod(f, d),
        |f, d| default_par_fold_nmod(f, d),
        |f, d| default_seq_fold_fn(f, d),
        |f, d| default_seq_fold_const(f, d),
        |f, d| default_seq_fold_enum(f, d),
        |f, d| default_seq_fold_trait(f, d),
        |f, d| default_seq_fold_impl(f, d),
        |f, d| default_seq_fold_type(f, d),
        |f, d| default_seq_fold_struct(f, d)
    )
}

pub fn default_seq_fold_doc<T>(fold: &Fold<T>, +doc: doc::Doc) -> doc::Doc {
    doc::Doc_({
        pages: do vec::map(doc.pages) |page| {
            match *page {
              doc::CratePage(doc) => {
                doc::CratePage((fold.fold_crate)(fold, doc))
              }
              doc::ItemPage(doc) => {
                doc::ItemPage(fold_ItemTag(fold, doc))
              }
            }
        },
        .. *doc
    })
}

pub fn default_seq_fold_crate<T>(
    fold: &Fold<T>,
    +doc: doc::CrateDoc
) -> doc::CrateDoc {
    {
        topmod: (fold.fold_mod)(fold, doc.topmod)
    }
}

pub fn default_seq_fold_item<T>(
    _fold: &Fold<T>,
    +doc: doc::ItemDoc
) -> doc::ItemDoc {
    doc
}

pub fn default_any_fold_mod<T:Send Clone>(
    fold: &Fold<T>,
    +doc: doc::ModDoc
) -> doc::ModDoc {
    let fold_copy = fold.clone();
    doc::ModDoc_({
        item: (fold.fold_item)(fold, doc.item),
        items: par::map(doc.items, |ItemTag, move fold_copy| {
            fold_ItemTag(&fold_copy, *ItemTag)
        }),
        .. *doc
    })
}

pub fn default_seq_fold_mod<T>(
    fold: &Fold<T>,
    +doc: doc::ModDoc
) -> doc::ModDoc {
    doc::ModDoc_({
        item: (fold.fold_item)(fold, doc.item),
        items: vec::map(doc.items, |ItemTag| {
            fold_ItemTag(fold, *ItemTag)
        }),
        .. *doc
    })
}

pub fn default_par_fold_mod<T:Send Clone>(
    fold: &Fold<T>,
    +doc: doc::ModDoc
) -> doc::ModDoc {
    let fold_copy = fold.clone();
    doc::ModDoc_({
        item: (fold.fold_item)(fold, doc.item),
        items: par::map(doc.items, |ItemTag, move fold_copy| {
            fold_ItemTag(&fold_copy, *ItemTag)
        }),
        .. *doc
    })
}

pub fn default_any_fold_nmod<T:Send Clone>(
    fold: &Fold<T>,
    +doc: doc::NmodDoc
) -> doc::NmodDoc {
    let fold_copy = fold.clone();
    {
        item: (fold.fold_item)(fold, doc.item),
        fns: par::map(doc.fns, |FnDoc, move fold_copy| {
            (fold_copy.fold_fn)(&fold_copy, *FnDoc)
        }),
        .. doc
    }
}

pub fn default_seq_fold_nmod<T>(
    fold: &Fold<T>,
    +doc: doc::NmodDoc
) -> doc::NmodDoc {
    {
        item: (fold.fold_item)(fold, doc.item),
        fns: vec::map(doc.fns, |FnDoc| {
            (fold.fold_fn)(fold, *FnDoc)
        }),
        .. doc
    }
}

pub fn default_par_fold_nmod<T:Send Clone>(
    fold: &Fold<T>,
    +doc: doc::NmodDoc
) -> doc::NmodDoc {
    let fold_copy = fold.clone();
    {
        item: (fold.fold_item)(fold, doc.item),
        fns: par::map(doc.fns, |FnDoc, move fold_copy| {
            (fold_copy.fold_fn)(&fold_copy, *FnDoc)
        }),
        .. doc
    }
}

pub fn fold_ItemTag<T>(fold: &Fold<T>, +doc: doc::ItemTag) -> doc::ItemTag {
    match doc {
      doc::ModTag(ModDoc) => {
        doc::ModTag((fold.fold_mod)(fold, ModDoc))
      }
      doc::NmodTag(nModDoc) => {
        doc::NmodTag((fold.fold_nmod)(fold, nModDoc))
      }
      doc::FnTag(FnDoc) => {
        doc::FnTag((fold.fold_fn)(fold, FnDoc))
      }
      doc::ConstTag(ConstDoc) => {
        doc::ConstTag((fold.fold_const)(fold, ConstDoc))
      }
      doc::EnumTag(EnumDoc) => {
        doc::EnumTag((fold.fold_enum)(fold, EnumDoc))
      }
      doc::TraitTag(TraitDoc) => {
        doc::TraitTag((fold.fold_trait)(fold, TraitDoc))
      }
      doc::ImplTag(ImplDoc) => {
        doc::ImplTag((fold.fold_impl)(fold, ImplDoc))
      }
      doc::TyTag(TyDoc) => {
        doc::TyTag((fold.fold_type)(fold, TyDoc))
      }
      doc::StructTag(StructDoc) => {
        doc::StructTag((fold.fold_struct)(fold, StructDoc))
      }
    }
}

pub fn default_seq_fold_fn<T>(
    fold: &Fold<T>,
    +doc: doc::FnDoc
) -> doc::FnDoc {
    {
        item: (fold.fold_item)(fold, doc.item),
        .. doc
    }
}

pub fn default_seq_fold_const<T>(
    fold: &Fold<T>,
    +doc: doc::ConstDoc
) -> doc::ConstDoc {
    {
        item: (fold.fold_item)(fold, doc.item),
        .. doc
    }
}

pub fn default_seq_fold_enum<T>(
    fold: &Fold<T>,
    +doc: doc::EnumDoc
) -> doc::EnumDoc {
    {
        item: (fold.fold_item)(fold, doc.item),
        .. doc
    }
}

pub fn default_seq_fold_trait<T>(
    fold: &Fold<T>,
    +doc: doc::TraitDoc
) -> doc::TraitDoc {
    {
        item: (fold.fold_item)(fold, doc.item),
        .. doc
    }
}

pub fn default_seq_fold_impl<T>(
    fold: &Fold<T>,
    +doc: doc::ImplDoc
) -> doc::ImplDoc {
    {
        item: (fold.fold_item)(fold, doc.item),
        .. doc
    }
}

pub fn default_seq_fold_type<T>(
    fold: &Fold<T>,
    +doc: doc::TyDoc
) -> doc::TyDoc {
    {
        item: (fold.fold_item)(fold, doc.item),
        .. doc
    }
}

pub fn default_seq_fold_struct<T>(
    fold: &Fold<T>,
    +doc: doc::StructDoc
) -> doc::StructDoc {
    {
        item: (fold.fold_item)(fold, doc.item),
        .. doc
    }
}

#[test]
fn default_fold_should_produce_same_doc() {
    let source = ~"mod a { fn b() { } mod c { fn d() { } } }";
    let ast = parse::from_str(source);
    let doc = extract::extract(ast, ~"");
    let fld = default_seq_fold(());
    let folded = (fld.fold_doc)(&fld, doc);
    assert doc == folded;
}

#[test]
fn default_fold_should_produce_same_consts() {
    let source = ~"const a: int = 0;";
    let ast = parse::from_str(source);
    let doc = extract::extract(ast, ~"");
    let fld = default_seq_fold(());
    let folded = (fld.fold_doc)(&fld, doc);
    assert doc == folded;
}

#[test]
fn default_fold_should_produce_same_enums() {
    let source = ~"enum a { b }";
    let ast = parse::from_str(source);
    let doc = extract::extract(ast, ~"");
    let fld = default_seq_fold(());
    let folded = (fld.fold_doc)(&fld, doc);
    assert doc == folded;
}

#[test]
fn default_parallel_fold_should_produce_same_doc() {
    let source = ~"mod a { fn b() { } mod c { fn d() { } } }";
    let ast = parse::from_str(source);
    let doc = extract::extract(ast, ~"");
    let fld = default_par_fold(());
    let folded = (fld.fold_doc)(&fld, doc);
    assert doc == folded;
}