using Amazon;

using Amazon.S3;

using Funq;

using ServiceStack;

using ServiceStack.Auth;

using ServiceStack.Aws.DynamoDb;

using ServiceStack.Aws.Sqs;

using ServiceStack.Caching;

using ServiceStack.IO;

using ServiceStack.Logging;

using ServiceStack.Messaging;

using ServiceStack.Razor;

using ServiceStack.Validation;

namespace AwsApps

{

public class AppHost : AppHostBase

{

public AppHost() : base("AWS Examples", typeof(AppHost).Assembly)

{

#if DEBUG

LogManager.LogFactory = new StringBuilderLogFactory(); //View logs at /logs

#else

//Deployed RELEASE build uses Config settings in DynamoDb

AppSettings = new MultiAppSettings(

new DynamoDbAppSettings(new PocoDynamo(AwsConfig.CreateAmazonDynamoDb()), initSchema:true),

new AppSettings());

#endif

}

public override void Configure(Container container)

{

SetConfig(new HostConfig

{

UseCamelCase = true,

});

Plugins.Add(new RazorFormat());

//Comment out 2 lines below to change to use local FileSystem instead of S3

var s3Client = new AmazonS3Client(AwsConfig.AwsAccessKey, AwsConfig.AwsSecretKey, RegionEndpoint.USEast1);

VirtualFiles = new S3VirtualFiles(s3Client, AwsConfig.S3BucketName);

AddVirtualFileSources.Add(VirtualFiles);

container.Register<IPocoDynamo>(c => new PocoDynamo(AwsConfig.CreateAmazonDynamoDb()));

var db = container.Resolve<IPocoDynamo>();

db.RegisterTable<Todos.Todo>();

db.RegisterTable<EmailContacts.Email>();

db.RegisterTable<EmailContacts.Contact>();

db.InitSchema();

//AWS Auth

container.Register<ICacheClient>(new DynamoDbCacheClient(db, initSchema:true));

container.Register<IAuthRepository>(new DynamoDbAuthRepository(db, initSchema:true));

Plugins.Add(CreateAuthFeature());

//EmailContacts

ConfigureSqsMqServer(container);

ConfigureEmailer(container);

Plugins.Add(new ValidationFeature());

container.RegisterValidators(typeof(EmailContacts.CreateContact).Assembly);

}

public AuthFeature CreateAuthFeature()

{

return new AuthFeature(() => new AuthUserSession(),

new IAuthProvider[]

{

new CredentialsAuthProvider(), //HTML Form post of UserName/Password credentials

new BasicAuthProvider(), //Sign-in with HTTP Basic Auth

new DigestAuthProvider(AppSettings), //Sign-in with HTTP Digest Auth

new TwitterAuthProvider(AppSettings), //Sign-in with Twitter

new FacebookAuthProvider(AppSettings), //Sign-in with Facebook

new GoogleAuthProvider(AppSettings), //Sign-in with Google OAuth2 Provider

new LinkedInAuthProvider(AppSettings), //Sign-in with LinkedIn OAuth2 Provider

new GithubAuthProvider(AppSettings), //Sign-in with GitHub OAuth Provider

})

{

HtmlRedirect = "/awsauth/",

IncludeRegistrationService = true,

};

}

private void ConfigureSqsMqServer(Container container)

{

container.Register<IMessageService>(c => new SqsMqServer(

AwsConfig.AwsAccessKey, AwsConfig.AwsSecretKey, RegionEndpoint.USEast1) {

DisableBuffering = true,

});

var mqServer = container.Resolve<IMessageService>();

mqServer.RegisterHandler<EmailContacts.EmailContact>(ExecuteMessage);

mqServer.Start();

}

private void ConfigureEmailer(Container container)

{

//If SmtpConfig exists, use real SMTP Emailer otherwise use simulated DbEmailer

var smtpConfig = AppSettings.Get<EmailContacts.SmtpConfig>("SmtpConfig");

if (smtpConfig != null)

{

container.Register(smtpConfig);

container.RegisterAs<EmailContacts.SmtpEmailer, EmailContacts.IEmailer>().ReusedWithin(ReuseScope.Request);

}

else

{

container.RegisterAs<EmailContacts.DbEmailer, EmailContacts.IEmailer>().ReusedWithin(ReuseScope.Request);

}

}

}

#if DEBUG

[Route("/logs")]

public class GetLogs { } //Useful service to dump Debug logs at /logs

public class LogService : Service

{

[AddHeader(ContentType = MimeTypes.PlainText)]

public object Any(GetLogs request)

{

return ((StringBuilderLogFactory)LogManager.LogFactory).GetLogs();

}

}

#endif

}