! Copyright (C) 2006, 2008 Slava Pestov
! See http://factorcode.org/license.txt for BSD license.
USING: alien alien.c-types alien.strings arrays assocs
combinators compiler hashtables kernel libc math namespaces
parser sequences words cocoa.messages cocoa.runtime
compiler.units io.encodings.ascii ;
IN: cocoa.subclassing

: init-method ( method alien -- )
    >r first3 r>
    [ >r execute r> set-objc-method-imp ] keep
    [ >r ascii malloc-string r> set-objc-method-types ] keep
    >r sel_registerName r> set-objc-method-name ;

: <empty-method-list> ( n -- alien )
    "objc-method-list" heap-size
    "objc-method" heap-size pick * + 1 calloc
    [ set-objc-method-list-count ] keep ;

: <method-list> ( methods -- alien )
    dup length dup <empty-method-list> -rot
    [ pick method-list@ objc-method-nth init-method ] 2each ;

: define-objc-methods ( class methods -- )
    <method-list> class_addMethods ;

: <objc-class> ( name info -- class )
    "objc-class" malloc-object
    [ set-objc-class-info ] keep
    [ >r ascii malloc-string r> set-objc-class-name ] keep ;

: <protocol-list> ( name -- protocol-list )
    "objc-protocol-list" malloc-object
    1 over set-objc-protocol-list-count
    swap objc-protocol over set-objc-protocol-list-class ;

! The Objective C object model is a bit funny.
! Every class has a metaclass.

! The superclass of the metaclass of X is the metaclass of the
! superclass of X.

! The metaclass of the metaclass of X is the metaclass of the
! root class of X.
: meta-meta-class ( class -- class ) root-class objc-class-isa ;

: copy-instance-size ( class -- )
    dup objc-class-super-class objc-class-instance-size
    swap set-objc-class-instance-size ;

: <meta-class> ( superclass name -- class )
    CLS_META <objc-class>
    [ >r dup objc-class-isa r> set-objc-class-super-class ] keep
    [ >r meta-meta-class r> set-objc-class-isa ] keep
    dup copy-instance-size ;

: set-protocols ( protocols class -- )
    swap {
        { [ dup empty? ] [ 2drop ] }
        { [ dup length 1 = ] [
            first <protocol-list>
            swap set-objc-class-protocols
        ] }
    } cond ;

: <new-class> ( protocols metaclass superclass name -- class )
    CLS_CLASS <objc-class>
    [ set-objc-class-super-class ] keep
    [ set-objc-class-isa ] keep
    [ set-protocols ] keep
    dup copy-instance-size ;

: (define-objc-class) ( protocols superclass name imeth -- )
    >r
    >r objc-class r>
    [ <meta-class> ] 2keep <new-class> dup objc_addClass
    r> <method-list> class_addMethods ;

: encode-types ( return types -- encoding )
    swap prefix [
        alien>objc-types get at "0" append
    ] map concat ;

: prepare-method ( ret types quot -- type imp )
    >r [ encode-types ] 2keep r> [
        "cdecl" swap 4array % \ alien-callback ,
    ] [ ] make define-temp ;

: prepare-methods ( methods -- methods )
    [
        [ first4 prepare-method 3array ] map
    ] with-compilation-unit ;

: redefine-objc-methods ( imeth name -- )
    dup class-exists? [
        objc_getClass swap define-objc-methods
    ] [
        2drop
    ] if ;

SYMBOL: +name+
SYMBOL: +protocols+
SYMBOL: +superclass+

: define-objc-class ( imeth hash -- )
    clone [
        prepare-methods
        +name+ get "cocoa.classes" create drop
        +name+ get 2dup redefine-objc-methods swap [
            +protocols+ get , +superclass+ get , +name+ get , ,
            \ (define-objc-class) ,
        ] [ ] make import-objc-class
    ] bind ;

: CLASS:
    parse-definition unclip
    >hashtable define-objc-class ; parsing