#lang racket/base

(require racket/list

racket/stream) ; needed for gen:stream

(provide (struct-out dv)

(struct-out dohlc)

(struct-out test)

test-timeframe-minus-1

(struct-out trade)

(struct-out position)

(struct-out history))

(struct dv (date value)

#:transparent

#:methods gen:stream

[(define (stream-empty? stream)

(cond

[(dv? stream) #f]

[else (empty? stream)]))

(define (stream-first stream)

(cond

[(dv? stream) (dv-date stream)]

[else (first stream)]))

(define (stream-rest stream)

(cond

[(dv? stream) (list (dv-value stream))]

[else (rest stream)]))])

(struct dohlc (date open high low close)

#:transparent

#:methods gen:stream

[(define (stream-empty? stream)

(cond

[(dohlc? stream) #f]

[else (empty? stream)]))

(define (stream-first stream)

(cond

[(dohlc? stream) (dohlc-date stream)]

[else (first stream)]))

(define (stream-rest stream)

(cond

[(dohlc? stream) (list (dohlc-open stream)

(dohlc-high stream)

(dohlc-low stream)

(dohlc-close stream))]

[else (rest stream)]))])

(struct test (timeframe entry stop target)

#:transparent)

(define (test-timeframe-minus-1 t)

(test (- (test-timeframe t) 1)

(test-entry t)

(test-stop t)

(test-target t)))

(struct trade (date price amount test)

#:transparent)

(struct position (price amount)

#:transparent)

(struct history (test trade)

#:transparent)