-#include <Cocoa/Cocoa.h>

-

-class Slimber;

-

-@interface MainController : NSObject {

- Slimber* slimber;

-}

-

-@end

-#include "MainController.h"

-#include "Slimber.h"

-

-@implementation MainController

-

-- (void) dealloc {

- delete slimber;

- [super dealloc];

-}

-

-- (void) awakeFromNib {

- slimber = new Slimber();

-}

-

-@end

- <string key="NSClassName">MainController</string>

- <string key="className">MainController</string>

- <string key="minorKey">MainController.h</string>

- Slimber/Cocoa/MainController.mm \

- Slimber/Cocoa/Slimber.mm \

- Slimber/Cocoa/Menulet.m

-//DEPS += \

-// $(SLIMBER_COCOA_SOURCES:.m=.dep) \

-$(SLIMBER_COCOA_BINARY): $(SLIMBER_COCOA_OBJECTS) $(SWIFTEN_TARGET) $(SLIMBER_TRGET)

-#include <Cocoa/Cocoa.h>

-

-@interface Menulet : NSObject {

- NSStatusItem* statusItem;

- NSMenu* statusMenu;

- NSImage* menuIcon;

- NSArray* userNames;

- BOOL xmppOnline;

- NSString* xmppStatus;

-}

-

-- (id) init;

-- (void) updateMenu;

-- (void) setUserNames: (NSArray*) names;

-- (void) setXMPPStatus: (NSString*) status online: (BOOL) online;

-

-@end

-#import "Menulet.h"

-

-@implementation Menulet

-

-- (id) init {

- if ([super init]) {

- statusMenu = [[NSMenu alloc] init];

-

- statusItem = [[[NSStatusBar systemStatusBar] statusItemWithLength: NSVariableStatusItemLength] retain];

- [statusItem setHighlightMode: YES];

- [statusItem setEnabled: YES];

- [statusItem setToolTip: @"Slimber"];

- [statusItem setMenu: statusMenu];

-

- xmppStatus = @"";

- xmppOnline = NO;

- userNames = [[NSArray alloc] init];

-

- [self updateMenu];

- }

- return self;

-}

-

-- (void) dealloc {

- [statusItem release];

- [menuIcon release];

- [super dealloc];

-}

-

-- (void) updateIcon: (BOOL) online {

- NSBundle* bundle = [NSBundle bundleForClass: [self class]];

- NSString* path;

- if (online) {

- path = [bundle pathForResource: @"UsersOnline" ofType:@"png"];

- }

- else {

- path = [bundle pathForResource: @"UsersOffline" ofType:@"png"];

- }

- [statusItem setImage: [[NSImage alloc] initWithContentsOfFile: path]];

-}

-

-- (void) updateMenu {

- // Clear the menu

- while ([statusMenu numberOfItems] > 0) {

- [statusMenu removeItemAtIndex: 0];

- }

-

- // User items

- if ([userNames count] > 0) {

- [statusMenu addItem: [[NSMenuItem alloc] initWithTitle: @"Online users:" action: NULL keyEquivalent: @""]];

- int i;

- for (i = 0; i < [userNames count]; ++i) {

- NSString* text = [NSString stringWithFormat: @" %@", [userNames objectAtIndex: i]];

- NSMenuItem* userItem = [[NSMenuItem alloc] initWithTitle: text action: NULL keyEquivalent: @""];

- [statusMenu addItem: userItem];

- [userItem release];

- }

- }

- else {

- [statusMenu addItem: [[NSMenuItem alloc] initWithTitle: @"No online users" action: NULL keyEquivalent: @""]];

- }

- [self updateIcon: [userNames count] > 0];

- [statusMenu addItem: [NSMenuItem separatorItem]];

-

- // Self item

- NSMenuItem* loggedInItem;

- NSBundle* bundle = [NSBundle bundleForClass: [self class]];

- NSString* path;

- loggedInItem = [[NSMenuItem alloc] initWithTitle: xmppStatus action: NULL keyEquivalent: @""];

- if (xmppOnline) {

- path = [bundle pathForResource: @"Online" ofType:@"png"];

- }

- else {

- path = [bundle pathForResource: @"Offline" ofType:@"png"];

- }

- [loggedInItem setImage: [[NSImage alloc] initWithContentsOfFile: path]];

- [statusMenu addItem: loggedInItem];

- [statusMenu addItem: [NSMenuItem separatorItem]];

-

- // About menu

- NSMenuItem* aboutMenuItem = [[NSMenuItem alloc] initWithTitle: @"About Slimber" action: @selector(orderFrontStandardAboutPanel:) keyEquivalent: @""];

- [aboutMenuItem setTarget: [NSApplication sharedApplication]];

- [statusMenu addItem: aboutMenuItem];

- [statusMenu addItem: [NSMenuItem separatorItem]];

-

- // Exit item

- NSMenuItem* exitMenuItem = [[NSMenuItem alloc] initWithTitle: @"Exit" action: @selector(terminate:) keyEquivalent: @""];

- [exitMenuItem setTarget: [NSApplication sharedApplication]];

- [statusMenu addItem: exitMenuItem];

-}

-

-- (void) setXMPPStatus: (NSString*) status online: (BOOL) online {

- xmppStatus = status; // TODO: Should I retain status?

- xmppOnline = online;

- [self updateMenu];

-}

-

-- (void) setUserNames: (NSArray*) names {

- [names retain];

- [userNames release];

- userNames = names;

- [self updateMenu];

-}

-

-@end

-#pragma once

-

-#include <boost/shared_ptr.hpp>

-#include <boost/optional.hpp>

-

-#include "Slimber/ServerError.h"

-

-@class Menulet;

-namespace Swift {

- class Server;

- class VCardCollection;

- class LinkLocalServiceBrowser;

- class BonjourQuerier;

-}

-

-class Slimber {

- public:

- Slimber();

- ~Slimber();

-

- private:

- void handleSelfConnected(bool b);

- void handleServicesChanged();

- void handleServerStopped(boost::optional<Swift::ServerError> error);

-

- private:

- boost::shared_ptr<Swift::BonjourQuerier> dnsSDQuerier;

- Swift::LinkLocalServiceBrowser* linkLocalServiceBrowser;

- Swift::VCardCollection* vCardCollection;

- Swift::Server* server;

- Menulet* menulet;

-};

-#include "Slimber/Cocoa/Slimber.h"

-

-#include <boost/bind.hpp>

-

-#include "Swiften/Base/foreach.h"

-#include "Swiften/Application/Platform/PlatformApplication.h"

-#include "Swiften/LinkLocal/LinkLocalService.h"

-#include "Swiften/LinkLocal/LinkLocalServiceBrowser.h"

-#include "Swiften/LinkLocal/DNSSD/Bonjour/BonjourQuerier.h"

-#include "Slimber/Cocoa/Menulet.h"

-#include "Slimber/Server.h"

-#include "Slimber/FileVCardCollection.h"

-

-using namespace Swift;

-

-Slimber::Slimber() {

- dnsSDQuerier = boost::shared_ptr<BonjourQuerier>(new BonjourQuerier());

- dnsSDQuerier->start();

-

- linkLocalServiceBrowser = new LinkLocalServiceBrowser(dnsSDQuerier);

- linkLocalServiceBrowser->onServiceAdded.connect(

- boost::bind(&Slimber::handleServicesChanged, this));

- linkLocalServiceBrowser->onServiceRemoved.connect(

- boost::bind(&Slimber::handleServicesChanged, this));

- linkLocalServiceBrowser->onServiceChanged.connect(

- boost::bind(&Slimber::handleServicesChanged, this));

- linkLocalServiceBrowser->start();

-

- vCardCollection = new FileVCardCollection(

- PlatformApplication("Slimber").getSettingsDir());

-

- server = new Server(5222, 5562, linkLocalServiceBrowser, vCardCollection);

- server->onStopped.connect(

- boost::bind(&Slimber::handleServerStopped, this, _1));

- server->onSelfConnected.connect(

- boost::bind(&Slimber::handleSelfConnected, this, _1));

-

- menulet = [[Menulet alloc] init];

- handleSelfConnected(false);

- handleServicesChanged();

-

- server->start();

-}

-

-Slimber::~Slimber() {

- [menulet release];

- delete server;

- delete vCardCollection;

- linkLocalServiceBrowser->stop();

- delete linkLocalServiceBrowser;

- dnsSDQuerier->stop();

-}

-

-void Slimber::handleSelfConnected(bool b) {

- if (b) {

- [menulet setXMPPStatus: @"You are logged in" online: true];

- }

- else {

- [menulet setXMPPStatus: @"You are not logged in" online: false];

- }

-}

-

-void Slimber::handleServicesChanged() {

- NSMutableArray* names = [[NSMutableArray alloc] init];

- foreach(const LinkLocalService& service, linkLocalServiceBrowser->getServices()) {

- [names addObject: [NSString stringWithUTF8String: service.getDescription().getUTF8Data()]];

- }

-

- [menulet setUserNames: names];

- [names release];

-}

-

-void Slimber::handleServerStopped(boost::optional<ServerError> error) {

- if (error) {

- [menulet setXMPPStatus: @"XMPP Server Error." online: false];

- }

- else {

- [menulet setXMPPStatus: @"XMPP Server Not Runnning." online: false];

- }

-}