const algotrader = require('algotrader');

const Robinhood = algotrader.Robinhood;
const Data = algotrader.Data;
const Algorithm = algotrader.Algorithm; // in progress

const robinhood = require('algotrader').Robinhood;
const User = robinhood.User;

const options = {
    doNotSaveToDisk: false, // If the `save` method should not store the user login info to disk (file)
    serializedUserFile: null // File to where the serialized user login info can be saved
};

const myUser = new User("username", "password", options);
myUser.authenticate()
    .then(() => {
        // User was authenticated
    })
    .catch(error => {
        // Either the request failed, or Robinhood responded with an error.
        // (Ex: you don't have internet access or your user credentials were incorrect)
    })

const myUser = new User("username");
// Or with options:
// const myUser = new User("username", null, options);
myUser.authenticate("password")
    .then(() => {
        // User was authenticated
    });

function getMFA() {
    return new Promise((resolve, reject) => {
        // Get the code here
        const mfa = "123456"
        resolve(mfa);
    })
}
// Note: the first parameter here is 'password' and is only required if you are re-authenticating
// an expired user or if you did not provide a password in the User constructor.
myUser.authenticate(null, getMFA)
    .then(() => {
        // User was authenticated
    })

const authenticatedUser;
authenticatedUser.save()
    .then(() => {
        // The user data was saved to:
        // project/node_modules/algotrader/objects/broker/robinhood/User.json
        // This filepath can be configured in the options parameter in `User` constructor
    });

const robinhood = require('algotrader').Robinhood;
const User = robinhood.User;

User.load()
    .then(myUser => {
        myUser.isAuthenticated(); // Boolean - see below
    })
    .catch(error => {
        // Make sure to always catch possible errors. You'll need to re-authenticate here.
        // - Possible errors: user session has expired, a saved user does not exist.
        if (error) {
            // Re-auth here and save if desired.
        }
    });

const options = { doNotSaveToDisk: true };
const authenticatedUser = new User("username", "password", options);
// ...
authenticatedUser.save()
    .then((serializedUser) => {
        // You can store `serializedUser` to where you want, like a Database
    });

// ...
const serializedUser = ''; // Get this value from your storage
User.load(serializedUser)
    .then(myUser => {
        myUser.isAuthenticated(); // Boolean - see below
    })
    .catch(error => {
        // Make sure to always catch possible errors. You'll need to re-authenticate here.
        // - Possible errors: user session has expired, a saved user does not exist.
        if (error) {
            // Re-auth here and save if desired.
        }
    });

myUser.getPortfolio()
    .then(myPortfolio => { // Algotrader retrieved the user's portfolio
        // You can find information on specific symbols
        let myTeslaShares = myPortfolio.getQuantity("TSLA"); // Returns the quantity of shares you own in the given symbol: 10
        let bestDayEver = myPortfolio.getPurchaseDate("SHLD"); // Returns the date (Date object) you purchased the given symbol: 2007-04-17
        // You can find information on the entire portfolio
        let mySymbols = myPortfolio.getSymbols(); // Returns an array of all symbols in the user's portfolio: ['FB', 'AMZN', 'NFLX', 'GOOG']
        let myMoneyMakers = myPortfolio.getQuantityGreaterThan(50); // Returns an array of all positions greater than the given amount: [Object]
        // Along with much more. See the link below to visit the Robinhood portfolio documentation.
    })
    .catch(error => {
        // Either the request failed, or Robinhood responded with an error.
        // (Ex: you don't have internet access or your user credentials were incorrect)
    })

// ES6
Instrument.getBySymbol("TWTR").then(async twtrInstrument => {
    // As of ~01/15/19, Robinhood requires an authenticated user to fetch a quote.
    // Working with Algotrader version > 1.4.3, thanks @Gillinghammer!
    let twtrQuote = await twtrInstrument.getQuote(user);
    const myOrder = new Order(user, {
        instrument: twtrInstrument,
        quote: twtrQuote,
        type: "market",
        timeInForce: "gfd",
        trigger: "immediate",
        quantity: 10,
        side: "buy"
    });
    myOrder.submit().then(res => {
        // Order was successful
    }).catch(error => {
        // Either the request failed, or Robinhood responded with an error.
        // (Ex: you don't have internet access or your balance was insufficient)
    });
});

const Instrument = algotrader.Robinhood.Instrument;
const OptionInstrument = algotrader.Robinhood.OptionInstrument;
const OptionOrder = algotrader.Robinhood.OptionOrder;

async function gains(user) {

    // First, we'll get the instrument that corresponds to the symbol TLRY.
    const tlry = await Instrument.getBySymbol('TLRY');

    // Next, we'll fetch an option chain containing puts for the upcoming expiration date.
    // This will return an array of OptionInstruments. See the example in the next section below.
    const chain = await OptionInstrument.getChain(user, tlry, 'put');

    // Now that we have the option chain, we'll need to find which OptionInstrument to trade
    // based on its strike price and expiration date. See the example below for how to sort them.
    // For now, we'll just take the first option contract in the array.
    const optionToBuy = chain[0];

    // Finally, we can create and place an order like so:
    let order = new OptionOrder(user, {
    side: 'buy',
    type: 'limit', // Note: Robinhood does not allow market buy orders
    price: '',
    timeInForce: 'gtc',
    quantity: 1,
    option: optionToBuy
    });

    order.submit().then(executedOrder => {
        // Success!
        console.log(executedOrder);
    }).catch(error => console.error(error));
}

[ OptionInstrument {
    url: 'https://api.robinhood.com',
    tradability: 'tradable',
    strikePrice: 121,
    state: 'active',
    type: 'put',
    symbol: 'TLRY',
    minTicks: { cutoff_price: '3.00', below_tick: '0.01', above_tick: '0.05' },
    instrumentURL: 'https://api.robinhood.com/options/instruments/28c3224d-3aa3-428c-aa78-7e0f5a4d01a0/',
    ids:
     { chain: 'c49063f0-557b-44b7-aeef-11fbc6a51243',
       option: '28c3224d-3aa3-428c-aa78-7e0f5a4d01a0' },
    dates:
     { expiration: 2019-02-01T00:00:00.000Z,
       created: 2019-01-18T03:08:31.325Z,
       updated: 2019-01-18T03:08:31.325Z } }, ...

const moment = require('moment');

const qqq = await Instrument.getBySymbol('QQQ');
const quote = await qqq.getQuote(user);
const chain = await OptionInstrument.getChain(user, qqq, 'call');
const expirations = await OptionInstrument.getExpirations(user, qqq);
// Returns an array of expiration dates [ 2019-02-01T00:00:00.000Z, 2019-02-08T00:00:00.000Z, ...

const nextExpiration = moment(expirations[0]);

let optionsExpiringNext = [];

chain.forEach(option => {
    let thisExpiration = moment(option.getExpiration());
    if (thisExpiration.isSame(nextExpiration)) {
        optionsExpiringNext.push(option);
    }
});

const Scheduler = algotrader.Algorithm.Scheduler;

Scheduler.onMarketOpen(-5 * 60000, () => {
    // Function to run five minutes before the market opens
});

Scheduler.every(30, false, () => {
    // Function to run every 1/2 hour
});

const Stream = algotrader.Data.Stream;

const myStream = new Stream(["PG", "DPS", "ULTA", "DIN", "ETSY"]);
myStream.start();

myStream
    .on('quote', quote => {
        // Returns a single Quote object. See the link below for documentation on Quotes.
    })
    .on('response', res => {
        // Returns a response object from the request module. Useful for debugging.
    })
    .on('error', error => {
        // Returns an error if the stream failed to start.
    });

const myStreamWithNews = new Stream(["JPM", "COST", "FDX"], {
    news: true, // Tells the streamer to also include news
    allHeadlines: true, // If true, all U.S. headlines will be sent in the stream. If false, only news pertaining to the given symbols will be outputted.
    newsApiKey: "newsApiKeyGoesHere" // Your API key from NewsAPI.org. See the link below for documentation on News.
});
myStreamWithNews.start();

myStreamWithNews
    .on('quote', quote => {
        // Returns a single Quote object. See the link below for documentation on Quotes.
    })
    .on('news', news => {
        // Returns a single News object. See the link below for documentation on News.
    })
    .on('response', res => {
        // Returns a response object from the request module. Useful for debugging.
    })
    .on('error', error => {
        // Returns an error if the stream failed to start.
    });

const streamWithIEX = new Stream(["PG", "DIN", "ULTA"], {
    iex: true,
    iexType: "tops"
});
streamWithIEX.on('iex', iex => {
    // Returns an object described here: https://iextrading.com/developer/docs/#iex-market-data
});

const AlphaVantage = algotrader.Data.AlphaVantage;
const av = new AlphaVantage("myApiKey");

// Get real time intraday price information
av.timeSeriesIntraday("AAPL", "1min").then(array => {
    // Returns an array of Quote objects for every minute since market open
    array.forEach(quote => {
        console.log( quote.getOpen() );   // 174.78
        console.log( quote.getVolume() ); // 13523
    });
});
// Get relative strength index
av.rsi("AAPL", "daily", 14, "close").then(array => {
    // Returns an array of objects representing the RSI on each day
    array.forEach(rsi => {
        // { date: 2017-11-17T00:00:00.000Z, RSI: '57.3707' }
    });
});

const IEX = algotrader.Data.IEX;

// Returns a quote object
IEX.getQuote("CSX").then(quote => {
    // Quote {
    // symbol: 'CSX',
    //     date: 2018-05-04T20:00:00.251Z,
    //     source: 'IEX',
    //     price:
    // { last: 59.97,
    //     open: 58.69,
    //     high: 60.41,
    //     low: 58.575,
    //     close: 59.97,
    //     volume: 4186069 },
    // dom: { bids: [], asks: [] }, - This was written while the market was closed. While the market is open, DOM elements are supported in the Quote object.
    // meta: undefined,
    //     original: undefined }
});

// Returns an array of fiscal reports dating back 5 years
IEX.getFinancials("AVGO").then(financials => {
    // [ { reportDate: '2018-01-31',
    // 	grossProfit: 2643000000,
    // 	costOfRevenue: 2684000000,
    // 	operatingRevenue: 5327000000,
    // 	totalRevenue: 5327000000,
    // 	operatingIncome: 1088000000,
    // 	netIncome: 6230000000,
    // 	researchAndDevelopment: 925000000,
    // 	operatingExpense: 1555000000,
    // 	currentAssets: 11220000000,
    // 	totalAssets: 54544000000,
    // 	totalLiabilities: 28643000000,
    // 	currentCash: 7076000000,
    // 	currentDebt: 117000000,
    // 	totalCash: 7076000000,
    // 	totalDebt: 17592000000,
    // 	shareholderEquity: 25901000000,
    // 	cashChange: -4128000000,
    // 	cashFlow: 1685000000,
    // 	operatingGainsLosses: null },
});

// Returns company name, EPS, divided, short interest, 52-week high/low, percent change, EBITDA, and more.
IEX.getStats("HD").then(stats => {
    // For full output see https://iextrading.com/developer/docs/#key-stats
});

IEX.getLogo("MMM").then(logoURL => {
    // https://storage.googleapis.com/iex/api/logos/MMM.png
});

const Nasdaq = algotrader.Data.Nasdaq;

Nasdaq.getListings().then(array => {
    // [
    //     {
    //         Symbol: 'AABA',
    //         'Security Name': 'Altaba Inc. - Common Stock',
    //         'Market Category': 'Q',
    //         'Test Issue': 'N',
    //         'Financial Status': 'N',
    //         'Round Lot Size': '100',
    //         ETF: 'N',
    //         NextShares: 'N'
    //     },
    //     {
    //         Symbol: 'AAL',
    //         'Security Name': 'American Airlines Group, Inc. - Common Stock',
    //         'Market Category': 'Q',
    //         'Test Issue': 'N',
    //         'Financial Status': 'N',
    //         'Round Lot Size': '100',
    //         ETF: 'N',
    //         NextShares: 'N'
    //     },
    // ... and thousands more
});

const Query = algotrader.Data.Query;

Query.getEarnings(1).then(array => {
    // Returns an array of companies that are reporting their earnings within the next '1' day.
    // [
    //     {
    //         symbol: 'NFLX',
    //         instrument: 'https://api.robinhood.com/instruments/81733743-965a-4d93-b87a-6973cb9efd34/',
    //         year: 2018,
    //         quarter: 1,
    //         eps: { estimate: '0.6400', actual: null },
    //         report: { date: '2018-04-16', timing: 'pm', verified: true },
    //         call:
    //             { datetime: '2018-04-16T22:00:00Z',
    //             broadcast_url: 'http://mmm.wallstreethorizon.com/u.asp?u=152320',
    //             replay_url: null }
    //     },
    // ... and more
});

Query.search("Nordstrom").then(array => {
    // Returns an array of matching stocks, options, ETF's, and others.
    // [
    //     { symbol: 'JWN',
    //         name: 'Nordstrom, Inc.',
    //         exch: 'NYQ',
    //         type: 'S',
    //         exchDisp: 'NYSE',
    //         typeDisp: 'Equity' },
    //     { symbol: 'JWN180420C00045000',
    //         name: 'JWN Apr 2018 call 45.000',
    //         exch: 'OPR',
    //         type: 'O',
    //         exchDisp: 'OPR',
    //         typeDisp: 'Option' },
    //  ... and more
});

Query.getTopGainers(5).then(array => {
    // Returns an array of objects each containing information on the five best performing stocks.
    // You can also use .getTopLosers(count)
});

Query.getHighestVolume(5).then(array => {
    // Returns an array of objects each containing information on five of today's stocks with the highest volume.
});

News.getFromYahoo("AAPL").then(array => {
    // Returns an array of news articles for the given symbol
    // [
    //     News {
    //         title: 'Amazon and Walmart Battle for Control of Flipkart',
    //         description: 'The world's largest retailer and its online counterpart compete on many fronts, but the fight for India's largest online retailer may be one of the most important.',
    //         date: 2018-04-15T15:45:00.000Z,
    //         source: undefined,
    //         author: undefined,
    //         url: 'https://finance.yahoo.com/news/amazon-walmart-battle-control-flipkart-154500760.html?.tsrc=rss' },
    //     News {
    //         title: 'President Trump is considering rejoining the Trans Pacific Partnership trade deal',
    //             description: 'President Trump is opening the door to rejoining the Trans Pacific Partnership trade deal. Yahoo Finance’s Jen Rogers and Rick Newman look at the implications.',
    //             date: 2018-04-13T14:40:56.000Z,
    //             source: undefined,
    //             author: undefined,
    //             url: 'https://finance.yahoo.com/video/president-trump-considering-rejoining-trans-144056381.html?.tsrc=rss' },
    // ... and more
});

News.getHeadlines("myApiKey", {
    country: "us",
    category: "business"
}).then(array => {
    // Returns an array of News objects related to U.S. business.
    // [
    //     News {
    //         title: 'Two black men were arrested waiting at a Starbucks. Now the company, police are on the defensive.',
    //         description: 'The backlash is a dramatic turn from efforts to craft the company as a progressive 	corporate leader that values “diversity and inclusion.”',
    //         date: 2018-04-15T15:22:22.000Z,
    //         source: 'The Washington Post',
    //         author: null,
    //         url: 'https://www.washingtonpost.com/news/business/wp/2018/04/15/two-black-men-were-arrested-waiting-at-a-starbucks-now-the-company-police-are-on-the-defensive/' },
    //     News {
    //         title: 'Zillow surprises investors by buying up homes',
    //         description: 'Real estate platform Zillow changed up its business model this week, announcing that it plans to purchase and sell homes in Las Vegas and Phoenix. Zillow will be working with Berkshire Hathaway and Coldwell Banker to make offers on homes before it finds a buy…',
    //         date: 2018-04-15T00:27:56.000Z,
    //         source: 'TechCrunch',
    //         author: 'Katie Roof',
    //         url: 'https://techcrunch.com/2018/04/14/zillow-surprises-investors-by-buying-up-homes/' }
    // ... and more
});

news.getArticle().then(html => {
    // This will return raw HTML from the source. You'll have to parse it yourself if you want to read the entire article, but typically the description - news.getDescription() - is sufficient.
});

myOrder.submit().then(res => {
    // Order was successful
}).catch(error => { console.error(error.toString()) });